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.
Assess the quality of the AI-generated content by voting
Score: 0
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.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook 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.