Hyper-Decision Transformer for Efficient Online Policy Adaptation
AI-generated Key Points
- The Hyper-Decision Transformer (HDT) is a new framework proposed to improve the generalization capability of large transformer-based agents with limited online interactions and computation budgets.
- Existing large transformer-based agents may have sub-optimal performances when directly deployed in novel tasks.
- HDT can generalize to novel unseen tasks from a handful of demonstrations in a data- and parameter-efficient manner.
- The base Decision Transformer (DT) is augmented with an adaptation module, whose parameters are initialized by a hyper-network.
- HDT adapts faster to unseen tasks than fine-tuning the whole DT model with only 0.5% of DT parameters fine-tuned after a single expert demonstration.
- HDT outperforms state-of-the-art baselines in terms of task success rates by a large margin even when expert actions are not available.
- Efficient adaptations of large transformer-based agents can be achieved through meta RL techniques such as few-shot learning or transfer learning.
- The Hyper Decision Transformer framework provides an effective solution for adapting quickly to novel unseen tasks while maintaining data and parameter efficiency.
Authors: Mengdi Xu, Yuchen Lu, Yikang Shen, Shun Zhang, Ding Zhao, Chuang Gan
Abstract: Decision Transformers (DT) have demonstrated strong performances in offline reinforcement learning settings, but quickly adapting to unseen novel tasks remains challenging. To address this challenge, we propose a new framework, called Hyper-Decision Transformer (HDT), that can generalize to novel tasks from a handful of demonstrations in a data- and parameter-efficient manner. To achieve such a goal, we propose to augment the base DT with an adaptation module, whose parameters are initialized by a hyper-network. When encountering unseen tasks, the hyper-network takes a handful of demonstrations as inputs and initializes the adaptation module accordingly. This initialization enables HDT to efficiently adapt to novel tasks by only fine-tuning the adaptation module. We validate HDT's generalization capability on object manipulation tasks. We find that with a single expert demonstration and fine-tuning only 0.5% of DT parameters, HDT adapts faster to unseen tasks than fine-tuning the whole DT model. Finally, we explore a more challenging setting where expert actions are not available, and we show that HDT outperforms state-of-the-art baselines in terms of task success rates by a large margin.
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
Welcome to our AI assistant! Here are some important things to keep in mind:
- The assistant will only answer questions related to this specific paper.
- Please note that this is not a bot for casual chatting.
- If you want the answer in a language other than the language you chose for navigating the website, simply add "TRANSLATE IN LANGUAGE L" at the end of your query (replace "LANGUAGE L" with the language of your choice).
- For example, you could ask "Can you extract the most important aspect of the paper? TRANSLATE IN SPANISH".
- If you want to keep the history of your questions/answers you should create an account.
Assess the quality of the AI-generated content by voting
Why do we need votes?
Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.
Similar papers summarized with our AI tools
Navigate through even more similar papers through atree representation
Look for similar papers (in beta version)
By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.