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

通过序列建模解决离线强化学习中的数据损坏问题

Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modeling

The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);City University of Hong Kong(香港城市大学);Rice University(莱斯大学);University of Liverpool(利物浦大学);Tencent Robotics X(腾讯Robotics X实验室)

Jiawei Xu, Rui Yang, Shuang Qiu, Feng Luo, Meng Fang, Baoxiang Wang, Lei Han

AI总结:

针对离线强化学习中的数据损坏问题,提出结合嵌入dropout、高斯加权学习和迭代数据校正的Robust Decision Transformer(RDT),显著提升了序列建模在噪声数据下的鲁棒性和性能。

摘要:

通过离线强化学习(RL)从离线数据集中学习策略,在扩展数据驱动决策的同时避免不安全且昂贵的在线交互,具有广阔前景。然而,从传感器或人类收集的真实数据通常包含噪声和错误,这对现有的离线RL方法构成了重大挑战,尤其是在真实数据有限的情况下。我们的研究表明,以往专注于适配基于时序差分学习的主流离线RL方法的研究,在数据集有限时,面对数据损坏仍然存在不足。相比之下,我们发现普通的序列建模方法(如Decision Transformer)即使没有经过专门修改,也表现出对数据损坏的鲁棒性。为了充分释放序列建模的潜力,我们提出了Robust Decision Rransformer(RDT),结合了三种简单而有效的鲁棒技术:嵌入dropout以提高模型对错误输入的鲁棒性,高斯加权学习以减轻损坏标签的影响,以及迭代数据校正以从源头上消除损坏数据。在MuJoCo、Kitchen和Adroit任务上的大量实验表明,与以往方法相比,RDT在各种数据损坏场景下表现出优越的性能。此外,在结合训练时数据损坏和测试时观测扰动的更具挑战性的环境中,RDT也展现出显著的鲁棒性。这些结果突显了序列建模在从噪声或损坏的离线数据集中学习的潜力,从而促进了离线RL在真实场景中的可靠应用。我们的代码可在https://github.com/jiawei415/RobustDecisionTransformer获取。

英文摘要:

Learning policy from offline datasets through offline reinforcement learning (RL) holds promise for scaling data-driven decision-making while avoiding unsafe and costly online interactions. However, real-world data collected from sensors or humans often contains noise and errors, posing a significant challenge for existing offline RL methods, particularly when the real-world data is limited. Our study reveals that prior research focusing on adapting predominant offline RL methods based on temporal difference learning still falls short under data corruption when the dataset is limited. In contrast, we discover that vanilla sequence modeling methods, such as Decision Transformer, exhibit robustness against data corruption, even without specialized modifications. To unlock the full potential of sequence modeling, we propose Robust Decision Rransformer (RDT) by incorporating three simple yet effective robust techniques: embedding dropout to improve the model's robustness against erroneous inputs, Gaussian weighted learning to mitigate the effects of corrupted labels, and iterative data correction to eliminate corrupted data from the source. Extensive experiments on MuJoCo, Kitchen, and Adroit tasks demonstrate RDT's superior performance under various data corruption scenarios compared to prior methods. Furthermore, RDT exhibits remarkable robustness in a more challenging setting that combines training-time data corruption with test-time observation perturbations. These results highlight the potential of sequence modeling for learning from noisy or corrupted offline datasets, thereby promoting the reliable application of offline RL in real-world scenarios. Our code is available at https://github.com/jiawei415/RobustDecisionTransformer.

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

↑