arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.01402cs.LGcs.AIcs.CL
收藏

预训练语言模型提升决策变换器的少样本提示能力

Pre-trained Language Models Improve the Few-shot Prompt Ability of Decision Transformer

Department of Electrical and Computer Engineering(电气与计算机工程系);Duke University(杜克大学);Department of Biostatistics and Bioinformatics(生物统计学与生物信息学系);Department of Computer Science(计算机科学系)

Yu Yang, Pan Xu

AI总结:

预训练语言模型提升决策变换器的少样本提示能力,通过初始化和提示正则化增强RL任务区分能力。

摘要:

决策变换器(DT)在离线强化学习(RL)任务中已崭露头角,利用预收集的数据集和Transformer对长序列建模的能力。近期研究表明,在DT中使用训练任务中的部分轨迹作为提示可以增强其在未见任务上的表现,从而产生Prompt-DT方法。然而,在许多场景中,从特定环境收集数据既昂贵又不安全,导致Transformer基于模型的数据渴求性造成性能不佳和少样本提示能力有限。此外,预训练中使用的有限数据集使得Prompt-DT类方法难以通过提示区分各种RL任务。为解决这些问题,我们引入了语言模型初始化的提示决策变换器(LPDT)框架,该框架利用预训练语言模型提供丰富的RL任务先验知识,并通过低秩适应(LoRA)对序列模型进行微调,用于元强化学习问题。我们进一步将提示正则化融入其中,以有效区分任务基于提示特征表示。全面的实证研究显示,在某些MuJoCo控制任务中,仅使用10%的数据初始化预训练语言模型即可提供先验知识,并实现与Prompt-DT相似的性能。我们还提供了详尽的消融研究,以验证每个组件的有效性,包括序列建模、语言模型、提示正则化和提示策略。

英文摘要:

Decision Transformer (DT) has emerged as a promising class of algorithms in offline reinforcement learning (RL) tasks, leveraging pre-collected datasets and Transformer's capability to model long sequences. Recent works have demonstrated that using parts of trajectories from training tasks as prompts in DT enhances its performance on unseen tasks, giving rise to Prompt-DT methods. However, collecting data from specific environments can be both costly and unsafe in many scenarios, leading to suboptimal performance and limited few-shot prompt abilities due to the data-hungry nature of Transformer-based models. Additionally, the limited datasets used in pre-training make it challenging for Prompt-DT type of methods to distinguish between various RL tasks through prompts alone. To address these challenges, we introduce the Language model-initialized Prompt Decision Transformer (LPDT) framework, which leverages pretrained language models providing rich prior knowledge for RL tasks and fine-tunes the sequence model using Low-rank Adaptation (LoRA) for meta-RL problems. We further incorporate prompt regularization to effectively differentiate between tasks based on prompt feature representations. Comprehensive empirical studies demonstrate that initializing with a pre-trained language model provides the prior knowledge and achieves a similar performance with Prompt-DT under only $10\%$ data in some MuJoCo control tasks. We also provide a thorough ablation study to validate the effectiveness of each component, including sequence modeling, language models, prompt regularizations, and prompt strategies.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑