Prompting Decision Transformer for Few-Shot Policy Generalization

27 June 2022

Ding Zhao

Chuang Gan

Papers citing "Prompting Decision Transformer for Few-Shot Policy Generalization"

50 / 96 papers shown

Title
Scaling Algorithm Distillation for Continuous Control with Mamba Samuel Beaussant Mehdi Mounsif 25 0 0 16 Jun 2025
Can In-Context Reinforcement Learning Recover From Reward Poisoning Attacks? Paulius Sasnauskas Yiğit Yalın Goran Radanović 15 0 0 07 Jun 2025
Mastering Massive Multi-Task Reinforcement Learning via Mixture-of-Expert Decision Transformer Yilun Kong Guozheng Ma Qi Zhao Haoyu Wang Li Shen Xueqian Wang Dacheng Tao MoE OffRL 31 1 0 30 May 2025
Prompting Decision Transformers for Zero-Shot Reach-Avoid Policies Kevin Li Marinka Zitnik OffRL 23 0 0 25 May 2025
TAGS: A Test-Time Generalist-Specialist Framework with Retrieval-Augmented Reasoning and Verification Jianghao Wu Feilong Tang Yulong Li Ming Hu Haochen Xue Shoaib Jameel Yutong Xie Imran Razzak LRM 46 0 0 23 May 2025
Implicit Neural-Representation Learning for Elastic Deformable-Object Manipulations Minseok Song JeongHo Ha Bonggyeong Park Daehyung Park 449 0 0 01 May 2025
Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision Shilin Zhang Zican Hu Wenhao Wu Xinyi Xie Jianxiang Tang Chunlin Chen Daoyi Dong Yu Cheng Zhenhong Sun Zhi Wang OffRL 438 0 0 21 Apr 2025
Target Return Optimizer for Multi-Game Decision Transformer Kensuke Tatematsu Akifumi Wachi OffRL 123 0 0 04 Mar 2025
An Adaptable Budget Planner for Enhancing Budget-Constrained Auto-Bidding in Online Advertising Zhijian Duan Yusen Huo Tianyu Wang Zhilin Zhang Yuezun Li Chuan Yu Jian Xu Jian Xu Xiaotie Deng 68 0 0 26 Jan 2025
Hierarchical Prompt Decision Transformer: Improving Few-Shot Policy Generalization with Global and Adaptive Guidance Zhe Wang Haozhu Wang Yanjun Qi OffRL 118 0 0 01 Dec 2024
Few-Shot Task Learning through Inverse Generative Modeling Aviv Netanyahu Yilun Du Antonia Bronars Jyothish Pari J. Tenenbaum Tianmin Shu Pulkit Agrawal 131 4 0 07 Nov 2024
Vision Language Models are In-Context Value Learners Yecheng Jason Ma Joey Hejna Ayzaan Wahid Chuyuan Fu Dhruv Shah ... Dinesh Jayaraman Wenhao Yu Tingnan Zhang Dorsa Sadigh Fei Xia 99 10 0 07 Nov 2024
Task-Aware Harmony Multi-Task Decision Transformer for Offline Reinforcement Learning Ziqing Fan Shengchao Hu Yuhang Zhou Li Shen Ya Zhang Yanfeng Wang Dacheng Tao OffRL 117 0 0 02 Nov 2024
Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglement Zhi Wang Li Zhang Wenhao Wu Yuanheng Zhu Dongbin Zhao C. L. Philip Chen OffRL 100 9 0 15 Oct 2024
Solving Multi-Goal Robotic Tasks with Decision Transformer Paul Gajewski Dominik Zurek Marcin Pietroñ Kamil Faber OffRL 61 1 0 08 Oct 2024
Predictive Coding for Decision Transformer Tung M. Luu Donghoon Lee Chang D. Yoo OffRL 129 2 0 04 Oct 2024
Scaling Offline Model-Based RL via Jointly-Optimized World-Action Model Pretraining Jie Cheng Ruixi Qiao Gang Xiong Binhua Li Yingwei Ma Binhua Li Yongbin Li Yisheng Lv OffRL OnRL LM&Ro 133 4 0 01 Oct 2024
Task-agnostic Pre-training and Task-guided Fine-tuning for Versatile Diffusion Planner Chenyou Fan Chenjia Bai Zhao Shan Haoran He Yang Zhang Zhen Wang 100 3 0 30 Sep 2024
An Offline Adaptation Framework for Constrained Multi-Objective Reinforcement Learning Qian Lin Zongkai Liu Danying Mo Chao Yu OffRL 143 1 0 16 Sep 2024
Online Decision MetaMorphFormer: A Casual Transformer-Based Reinforcement Learning Framework of Universal Embodied Intelligence Luo Ji Runji Lin OffRL AI4CE LM&Ro 76 0 0 11 Sep 2024
Unveiling Markov Heads in Pretrained Language Models for Offline Reinforcement Learning Wenhao Zhao Qiushui Xu Linjie Xu Lei Song Jinyu Wang Chunlai Zhou Jiang Bian 87 0 0 11 Sep 2024
In-Context Imitation Learning via Next-Token Prediction Letian Fu Huang Huang Gaurav Datta Lawrence Yunliang Chen William Chung-Ho Panitch Fangchen Liu Hui Li Ken Goldberg LM&Ro 84 23 0 28 Aug 2024
Pre-trained Language Models Improve the Few-shot Prompt Ability of Decision Transformer Yu Yang Pan Xu VLM OffRL 85 2 0 02 Aug 2024
Empowering Clinicians with Medical Decision Transformers: A Framework for Sepsis Treatment A. Rahman Pranav Agarwal R. Noumeir P. Jouvet Vincent Michalski Samira Ebrahimi Kahou OffRL 96 0 0 28 Jul 2024
Domain Adaptation of Visual Policies with a Single Demonstration Weiyao Wang Gregory D. Hager 89 0 0 23 Jul 2024
Multi-State-Action Tokenisation in Decision Transformers for Multi-Discrete Action Spaces Perusha Moodley Pramod S. Kaushik Dhillu Thambi Mark Trovinger Praveen Paruchuri Xia Hong Benjamin Rosman 139 0 0 01 Jul 2024
Residual-MPPI: Online Policy Customization for Continuous Control Pengcheng Wang Chenran Li Catherine Weaver Kenta Kawamoto Masayoshi Tomizuka Chen Tang Wei Zhan OffRL 151 3 0 01 Jul 2024
Mental Modeling of Reinforcement Learning Agents by Language Models Wenhao Lu Xufeng Zhao Josua Spisak Jae Hee Lee Stefan Wermter LLMAG LRM LM&Ro 68 3 0 26 Jun 2024
Accelerating Matrix Diagonalization through Decision Transformers with Epsilon-Greedy Optimization Kshitij Bhatta Geigh Zollicoffer Manish Bhattarai Phil Romero C. Negre Anders M. N. Niklasson A. Adedoyin 58 0 0 23 Jun 2024
SAM-E: Leveraging Visual Foundation Model with Sequence Imitation for Embodied Manipulation Junjie Zhang Chenjia Bai Haoran He Wenke Xia Zhigang Wang Bin Zhao Xiu Li Xuelong Li 120 13 0 30 May 2024
Preferred-Action-Optimized Diffusion Policies for Offline Reinforcement Learning Tianle Zhang Jiayi Guan Lin Zhao Yihang Li Dongjiang Li ... Lei Sun Yue Chen Xuelong Wei Lusong Li Xiaodong He 92 2 0 29 May 2024
HarmoDT: Harmony Multi-Task Decision Transformer for Offline Reinforcement Learning Shengchao Hu Ziqing Fan Li Shen Ya Zhang Yanfeng Wang Dacheng Tao OffRL 97 11 0 28 May 2024
Position: Foundation Agents as the Paradigm Shift for Decision Making Xiaoqian Liu Xingzhou Lou Jianbin Jiao Junge Zhang OffRL LLMAG 105 7 0 27 May 2024
One-Shot Imitation Learning with Invariance Matching for Robotic Manipulation Xinyu Zhang Abdeslam Boularias 120 10 0 21 May 2024
Prompt Learning for Generalized Vehicle Routing Fei Liu Xi Lin Weiduo Liao Zhenkun Wang Qingfu Zhang Xialiang Tong Mingxuan Yuan VLM 86 3 0 20 May 2024
Scrutinize What We Ignore: Reining In Task Representation Shift Of Context-Based Offline Meta Reinforcement Learning Hai Zhang Boyuan Zheng Anqi Guo Tianying Ji Anqi Guo Junqiao Zhao Lanqing Li OffRL 119 0 0 20 May 2024
A Minimalist Prompt for Zero-Shot Policy Learning Meng Song Xuezhi Wang Tanay Biradar Yao Qin Manmohan Chandraker OffRL 66 1 0 09 May 2024
X-Light: Cross-City Traffic Signal Control Using Transformer on Transformer as Meta Multi-Agent Reinforcement Learner Haoyuan Jiang Ziyue Li Hua Wei Xuantang Xiong Jingqing Ruan Jiaming Lu Hangyu Mao Rui Zhao 83 10 0 18 Apr 2024
Self-adaptive PSRO: Towards an Automatic Population-based Game Solver Pengdeng Li Shuxin Li Chang Yang Xinrun Wang Xiao Huang Hau Chan Bo An 71 1 0 17 Apr 2024
Decision Transformer as a Foundation Model for Partially Observable Continuous Control Xiangyuan Zhang Weichao Mao Haoran Qiu Tamer Basar OffRL AI4CE 96 6 0 03 Apr 2024
Can large language models explore in-context? Akshay Krishnamurthy Keegan Harris Dylan J. Foster Cyril Zhang Aleksandrs Slivkins LM&Ro LLMAG LRM 280 29 0 22 Mar 2024
Disentangling Policy from Offline Task Representation Learning via Adversarial Data Augmentation Chengxing Jia Fuxiang Zhang Yi-Chen Li Chenxiao Gao Xu-Hui Liu Lei Yuan Zongzhang Zhang Yang Yu AAML 83 4 0 12 Mar 2024
Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics Xinyu Zhang Wenjie Qiu Yi-Chen Li Lei Yuan Chengxing Jia Zongzhang Zhang Yang Yu OffRL 121 1 0 17 Feb 2024
Federated Prompt-based Decision Transformer for Customized VR Services in Mobile Edge Computing System Tailin Zhou Jiadong Yu Jun Zhang Danny H. K. Tsang 48 1 0 15 Feb 2024
SemTra: A Semantic Skill Translator for Cross-Domain Zero-Shot Policy Adaptation Sangwoo Shin Minjong Yoo Jeongwoo Lee Honguk Woo 94 4 0 12 Feb 2024
Towards an Information Theoretic Framework of Context-Based Offline Meta-Reinforcement Learning Lanqing Li Hai Zhang Xinyu Zhang Shatong Zhu Junqiao Zhao Junqiao Zhao Pheng-Ann Heng OffRL 106 11 0 04 Feb 2024
Context-Former: Stitching via Latent Conditioned Sequence Modeling Ziqi Zhang Jingzehua Xu Jinxin Liu Zifeng Zhuang Donglin Wang Miao Liu Shuai Zhang OffRL 104 4 0 29 Jan 2024
Solving Continual Offline Reinforcement Learning with Decision Transformer Kaixin Huang Li Shen Chen Zhao Chun Yuan Dacheng Tao CLL OffRL 84 5 0 16 Jan 2024
Self-supervised Pretraining for Decision Foundation Model: Formulation, Pipeline and Challenges Xiaoqian Liu Jianbin Jiao Junge Zhang OffRL LRM 100 2 0 29 Dec 2023
In-Context Reinforcement Learning for Variable Action Spaces Viacheslav Sinii Alexander Nikulin Vladislav Kurenkov Ilya Zisman Sergey Kolesnikov 137 17 0 20 Dec 2023