Search Results for author: Jake Grigsby

Found 10 papers, 7 papers with code

AMAGO: Scalable In-Context Reinforcement Learning for Adaptive Agents

1 code implementation15 Oct 2023 Jake Grigsby, Linxi Fan, Yuke Zhu

We introduce AMAGO, an in-context Reinforcement Learning (RL) agent that uses sequence models to tackle the challenges of generalization, long-term memory, and meta-learning.

In-Context Learning Meta-Learning +2

PGrad: Learning Principal Gradients For Domain Generalization

1 code implementation2 May 2023 Zhe Wang, Jake Grigsby, Yanjun Qi

In this work, we develop a novel DG training strategy, we call PGrad, to learn a robust gradient direction, improving models' generalization ability on unseen domains.

Domain Generalization

Launchpad: Learning to Schedule Using Offline and Online RL Methods

no code implementations1 Dec 2022 Vanamala Venkataswamy, Jake Grigsby, Andrew Grimshaw, Yanjun Qi

We utilize Offline RL as a launchpad to learn effective scheduling policies from prior experience collected using Oracle or heuristic policies.

Offline RL reinforcement-learning +2

RARE: Renewable Energy Aware Resource Management in Datacenters

no code implementations10 Nov 2022 Vanamala Venkataswamy, Jake Grigsby, Andrew Grimshaw, Yanjun Qi

Finally, we demonstrate that the DRL scheduler can learn from and improve upon existing heuristic policies using Offline Learning.

Management Scheduling

A Closer Look at Advantage-Filtered Behavioral Cloning in High-Noise Datasets

3 code implementations10 Oct 2021 Jake Grigsby, Yanjun Qi

A thorough investigation on a custom benchmark helps identify several key challenges involved in learning from high-noise datasets.

Decision Making reinforcement-learning +1

ST-MAML: A Stochastic-Task based Method for Task-Heterogeneous Meta-Learning

no code implementations27 Sep 2021 Zhe Wang, Jake Grigsby, Arshdeep Sekhon, Yanjun Qi

This paper proposes a novel method, ST-MAML, that empowers model-agnostic meta-learning (MAML) to learn from multiple task distributions.

Few-Shot Image Classification Meta-Learning

Long-Range Transformers for Dynamic Spatiotemporal Forecasting

2 code implementations24 Sep 2021 Jake Grigsby, Zhe Wang, Nam Nguyen, Yanjun Qi

Multivariate time series forecasting focuses on predicting future values based on historical context.

Multivariate Time Series Forecasting Time Series

Towards Automatic Actor-Critic Solutions to Continuous Control

1 code implementation16 Jun 2021 Jake Grigsby, Jin Yong Yoo, Yanjun Qi

Model-free off-policy actor-critic methods are an efficient solution to complex continuous control tasks.

Continuous Control

Measuring Visual Generalization in Continuous Control from Pixels

2 code implementations13 Oct 2020 Jake Grigsby, Yanjun Qi

Self-supervised learning and data augmentation have significantly reduced the performance gap between state and image-based reinforcement learning agents in continuous control tasks.

Continuous Control Data Augmentation +2

TextAttack: A Framework for Adversarial Attacks, Data Augmentation, and Adversarial Training in NLP

2 code implementations EMNLP 2020 John X. Morris, Eli Lifland, Jin Yong Yoo, Jake Grigsby, Di Jin, Yanjun Qi

TextAttack also includes data augmentation and adversarial training modules for using components of adversarial attacks to improve model accuracy and robustness.

Adversarial Text Data Augmentation +3

Cannot find the paper you are looking for? You can Submit a new open access paper.