通过减少轨迹拼接中的插值段,提升模仿学习数据生成质量。
MinInter: Minimizing Trajectory Interpolation During Data Augmentation for Imitation Learning

- 选择插值最少的示范轨迹拼接,生成更真实的数据。
- 在12个任务上显著提高数据生成成功率和策略成功率。
- 适合接触多、时长久、变化大的复杂操作场景使用。
模仿学习使机器人能从示范中习得复杂操作技能,但高质量数据采集成本高。轨迹级数据增强通过在不同初始状态下重组专家示范来缓解此问题。然而,现有方法常在不连续部分插入非专家过渡段,降低生成数据质量。本文提出MinInter,一种轨迹选择方法:对每个采样初始状态,选择所需插值最少的源示范以构成完整轨迹。通过显式最小化插值,MinInter在保持与现有框架兼容的同时,生成更高质量的合成示范。在MimicGen基准的12个操纵任务(共26种变体)上实验表明,MinInter稳定提升数据生成成功率和策略成功率,尤其在接触密集、长时程和高方差场景下增益最大。相比近期的SkillGen框架,尽管概念更简单,但其策略成功率更高,凸显了插值最小化在数据增强中的价值。
原文摘要 · Abstract (English)
Imitation learning enables robots to acquire complex manipulation skills from demonstrations, but its effectiveness is limited by the cost of collecting high-quality data. Trajectory-level data augmentation methods alleviate this challenge by recombining expert demonstrations under varied initial states. However, such methods typically insert interpolations or other non-expert transition segments between disjoint parts, and such non-expert segments could reduce the quality of the generated data. This paper introduces Minimizing Interpolation (MinInter), an effective trajectory selection method that, for each sampled initial configuration, chooses the source demonstration requiring the least interpolation to form a complete trajectory. By explicitly minimizing interpolations during data generation, MinInter produces higher-quality synthetic demonstrations while remaining compatible with existing data generation frameworks. Experiments on 12 manipulation tasks with 26 variants from the MimicGen benchmark show that MinInter consistently improves both data generation success rates and policy success rates, with the largest gains on contact-rich, long-horizon and high-variance settings. Compared to the recent SkillGen framework, MinInter achieves higher policy success rates despite its conceptual simplicity, underscoring the value of interpolation minimization for data augmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。