arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2410.00064cs.LGcs.AIcs.CVcs.RO

M2Distill:面向终身模仿学习的多模态蒸馏

M2Distill: Multi-Modal Distillation for Lifelong Imitation Learning

CSIRO(联邦科学与工业研究组织);DATA61;Queensland University of Technology(昆士兰科技大学)

Kaushik Roy, Akila Dissanayake, Brendan Tidd, Peyman Moghadam

AI总结:

针对终身模仿学习中的分布偏移与灾难性遗忘问题,提出多模态蒸馏方法M2Distill,通过跨模态潜在表示对齐与GMM策略差异最小化,在LIBERO基准上全面超越现有方法。

摘要:

由于增量学习步骤中出现的分布偏移,面向操作任务的终身模仿学习提出了重大挑战。现有方法通常专注于无监督技能发现以构建不断增长的技能库,或从多个策略进行蒸馏,随着多样化操作任务被不断引入,这可能导致可扩展性问题,并且可能无法在整个学习过程中确保一致的潜在空间,从而导致对先前学习技能的灾难性遗忘。在本文中,我们介绍了M2Distill,一种用于终身模仿学习的基于多模态蒸馏的方法,专注于在整个学习过程中保持跨视觉、语言和动作分布的一致潜在空间。通过调节从先前步骤到当前步骤跨不同模态的潜在表示偏移,并减少连续学习步骤之间高斯混合模型(GMM)策略的差异,我们确保学习到的策略在无缝集成新技能的同时,保留其执行先前学习任务的能力。在LIBERO终身模仿学习基准套件(包括LIBERO-OBJECT、LIBERO-GOAL和LIBERO-SPATIAL)上的广泛评估表明,我们的方法在所有评估指标上始终优于先前最先进的方法。

英文摘要:

Lifelong imitation learning for manipulation tasks poses significant challenges due to distribution shifts that occur in incremental learning steps. Existing methods often focus on unsupervised skill discovery to construct an ever-growing skill library or distillation from multiple policies, which can lead to scalability issues as diverse manipulation tasks are continually introduced and may fail to ensure a consistent latent space throughout the learning process, leading to catastrophic forgetting of previously learned skills. In this paper, we introduce M2Distill, a multi-modal distillation-based method for lifelong imitation learning focusing on preserving consistent latent space across vision, language, and action distributions throughout the learning process. By regulating the shifts in latent representations across different modalities from previous to current steps, and reducing discrepancies in Gaussian Mixture Model (GMM) policies between consecutive learning steps, we ensure that the learned policy retains its ability to perform previously learned tasks while seamlessly integrating new skills. Extensive evaluations on the LIBERO lifelong imitation learning benchmark suites, including LIBERO-OBJECT, LIBERO-GOAL, and LIBERO-SPATIAL, demonstrate that our method consistently outperforms prior state-of-the-art methods across all evaluated metrics.

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

↑