Hyper-Decision Transformer for Efficient Online Policy Adaptation

AI-generated keywords: Hyper-Decision Transformer Adaptation Generalization Meta RL Few-shot Learning

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.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Mengdi Xu, Yuchen Lu, Yikang Shen, Shun Zhang, Ding Zhao, Chuang Gan

ICLR 2023. Project page: https://sites.google.com/view/hdtforiclr2023/home
License: CC ZERO 1.0

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.

Submitted to arXiv on 17 Apr. 2023

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2304.08487v1

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. While existing large transformer-based agents have demonstrated strong performances in training tasks, they may have sub-optimal performances when directly deployed in novel tasks. To address this challenge, 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. 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. The authors validate HDT's generalization capability on object manipulation tasks and 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. Furthermore, they explore a more challenging setting where expert actions are not available, and show that HDT outperforms state-of-the-art baselines in terms of task success rates by a large margin. The problem formulation for efficient adaptation from observations is discussed in detail, highlighting how existing large transformer based agents may have suboptimal performance when directly deployed in novel tasks due to discrepancies between training and testing tasks. The authors propose that efficient adaptations of large transformer based agents can be achieved through meta RL techniques such as few shot learning or transfer learning. Overall, the Hyper Decision Transformer framework provides an effective solution for adapting quickly to novel unseen tasks while maintaining data and parameter efficiency. It has promising applications in various domains where rapid adaptation is crucial for successful task completion.
Created on 18 Apr. 2023

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 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.