実際に測定できる多様性:ロボットデータセットのための高速でモデルフリーな多様性指標
Diversity You Can Actually Measure: A Fast, Model-Free Diversity Metric for Robotics Datasets
模倣学習用ロボットデータセットの多様性を、シグネチャ変換に基づくエントロピーで定量化し、多様性を最大化するデータ選択アルゴリズムFAKTUALを提案した。
詳しい要約
1. どんなもの?
2. 先行研究と比べてどこがすごい?
3. 技術・手法の肝は?
4. どうやって有効だと検証した?
5. 議論はある?
6. 次に読むべき論文は?
※ AIが要旨から生成した要約です。正確性は原文をご確認ください。
著者: Sreevardhan Sirigiri, Nathan Samuel de Lara, Christopher Agia, Florian Shkurti, Fabio Ramos
分類: cs.RO
原文アブストラクト
Robotics datasets for imitation learning typically consist of long-horizon trajectories of different lengths over states, actions, and high-dimensional observations (e.g., RGB video), making it non-trivial to quantify diversity in a way that respects the underlying trajectory structure and geometry. We extend Shannon and von Neumann entropy to this setting by defining signature transform-based entropy on the Gram matrix of a signature kernel over demonstrations, yielding entropy and diversity metrics that operate directly on the demonstration dataset. Building on these metrics, we study how dataset diversity affects generalization performance in robot imitation learning and propose a simple, model-free way to curate diverse demonstrations. We introduce FAKTUAL (FAst trajectory Kernel enTropy cUration for imitation Learning), a data curation algorithm that selects a subset of demonstrations maximizing entropy given a subset-size budget. FAKTUAL is fully model-free, requires no access to the imitation policy or rollouts, and adds negligible overhead relative to policy training. We evaluate our approach on image and state-based RoboMimic and MetaWorld benchmarks, as well as four real-world manipulation tasks. Across tasks and architectures, diversity-aware curation with FAKTUAL consistently improves downstream success rates over random selection, while being substantially more computationally efficient compared to recent robot data curation methods. Our results suggest that the entropy of demonstration datasets is a practical tool for understanding and improving dataset diversity in robot imitation learning.
関連論文
- ロングテールな人間データキュレーションのための品質フィルタ監査データキュレーション
- ATHENA: ロボットデータキュレーションのための高速マルチタスク異種影響関数データキュレーション
- 量より質:データ中心ロボット学習のための影響関数によるデモンストレーションキュレーションデータキュレーション
- ROSER: ロボット学習のスケーラビリティを実現する数ショットロボットシーケンス検索データキュレーション
- Semantic-Drive: オープンボキャブラリ接地とニューロシンボリックVLM合意によるロングテールデータキュレーションの民主化データキュレーション
- OpenGVL:データキュレーションのための視覚的時間進捗ベンチマークデータキュレーション
