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

结合重建与对比方法的多模态强化学习表示

Combining Reconstruction and Contrastive Methods for Multimodal Representations in RL

Karlsruhe Institute of Technology(卡尔斯鲁厄理工学院);FZI Research Center for Information Technology(FZI信息技术研究中心);Bosch Center for Artificial Intelligence(博世人工智能中心);University of Tübingen(蒂宾根大学)

Philipp Becker, Sebastian Mossburger, Fabian Otto, Gerhard Neumann

AI总结:

本文提出CoRAL框架,结合重建与对比损失为多模态RL各传感器选择合适自监督损失,提升干扰场景下的性能与样本效率,解决复杂任务。

摘要:

使用重建或对比损失学习自监督表示,可以提高基于图像和多模态强化学习(RL)的性能和样本复杂度。在此,不同的自监督损失函数根据底层传感器模态的信息密度具有不同的优势和局限性。重建提供强大的学习信号,但容易受到干扰和虚假信息的影响。虽然对比方法可以忽略这些,但它们可能无法捕获所有相关细节,并可能导致表示崩溃。对于多模态强化学习,这表明应根据信号中的干扰量对不同模态进行不同处理。我们提出了对比重建聚合表示学习(CoRAL),这是一个统一框架,使我们能够为每个传感器模态选择最合适的自监督损失,并使表示更好地关注相关方面。我们在包含干扰或遮挡的图像、新的运动套件以及具有视觉逼真干扰的具有挑战性的操作套件等一系列任务上评估了CoRAL的益处。我们的结果表明,通过结合对比和基于重建的损失来学习多模态表示,可以显著提高性能,并解决更朴素的表示学习方法和其他近期基线方法无法触及的任务。

英文摘要:

Learning self-supervised representations using reconstruction or contrastive losses improves performance and sample complexity of image-based and multimodal reinforcement learning (RL). Here, different self-supervised loss functions have distinct advantages and limitations depending on the information density of the underlying sensor modality. Reconstruction provides strong learning signals but is susceptible to distractions and spurious information. While contrastive approaches can ignore those, they may fail to capture all relevant details and can lead to representation collapse. For multimodal RL, this suggests that different modalities should be treated differently based on the amount of distractions in the signal. We propose Contrastive Reconstructive Aggregated representation Learning (CoRAL), a unified framework enabling us to choose the most appropriate self-supervised loss for each sensor modality and allowing the representation to better focus on relevant aspects. We evaluate CoRAL's benefits on a wide range of tasks with images containing distractions or occlusions, a new locomotion suite, and a challenging manipulation suite with visually realistic distractions. Our results show that learning a multimodal representation by combining contrastive and reconstruction-based losses can significantly improve performance and solve tasks that are out of reach for more naive representation learning approaches and other recent baselines.

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

↑