日本フィジカルAI新聞

世界のフィジカルAIを、日本語で。

週刊ニュースレター購読
模倣学習arXiv:2609.15631

フローマッチング運動事前分布:模倣学習のためのオンライン最適輸送報酬

Flow-Matched Motion Priors: Online Optimal-Transport Rewards for Imitation Learning

シェア:XThreadsFacebookLINEはてブBluesky

現在のロールアウトと専門家運動の間の経路に沿ってフローマッチングで学習したスカラー報酬を提案し、Unitree G1の歩行タスクで従来のAMPや重心OT報酬より安定した歩行を実現した。

詳しい要約

1. どんなもの?

本論文は、模倣学習のための新しい報酬設計である Flow-Matched Motion Priors (FMP) を提案する。FMP は、現在のロールアウト履歴と専門家の動作バンクを結ぶ経路上で、オンラインでスカラー報酬を学習する。Entropic OT による結合と Flow Matching (FM) を用いてニューラルポテンシャルを訓練し、AMP と同様にアクターは物理観測のみを受け取り、報酬はスカラーである。Unitree G1 での実験により、FMP が安定した前進歩行を実現することが示された。

2. 先行研究と比べてどこがすごい?

先行研究の Adversarial Motion Priors (AMP) は、識別器を用いた敵対的報酬を提供するが、ポリシーと専門家のサポートが離れていると報酬が無情報になる問題がある。また、Optimal Transport (OT) を単純に使うと、マッチした専門家の後続状態を平均化する重心ターゲットとなり、歩行位相間の平均化が関節運動を弱める。FMP は、これらの問題を克服し、より一般化性能が高く、安定した歩行を実現する。

3. 技術・手法の肝は?

FMP の技術的核心は、オンラインでスカラー報酬を学習する点にある。Entropic OT が現在のロールアウトと専門家動作バンク間の結合を提供する。各ポリシー更新前に、ロールアウトから専門家への経路に沿った Flow Matching (FM)、端点勾配監督、相対価値校准を用いてニューラルポテンシャルを訓練する。アクターは物理観測のみを受け取り、報酬はスカラーとして与えられる。

4. どうやって有効だと検証した?

制御された報酬モデル実験により、FMP がフィッティングロールアウトを超えて大幅に優れた一般化を示すことが検証された。Unitree G1 では、5000万遷移のマッチした実験で、FMP を AMP、重心 OT 報酬、ネストされたアブレーションと比較した。FMP はデモンストレーションリセットから 0.727 m/s、固定デフォルトポーズから 0.338 m/s の安定した前進歩行を生成した。固定ポーズ条件では、FMP は 129 回の転倒に対し、端点のみの制御は 243 回であった。また、静的スコア勾配教師と比較して、動的 FM は補間分数 0.25 と 0.50 でスコア増分誤差を減少させ、オフライン適合時間を 29% 削減した。

5. 議論はある?

議論の詳細は要旨からは不明であるが、FMP の有効性と効率性が示されている。

6. 次に読むべき論文は?

次に読むべき論文は、要旨で参照されている Adversarial Motion Priors (AMP) や Optimal Transport (OT) 関連の研究、および Flow Matching (FM) の応用論文が考えられる。また、同分野の定番として、模倣学習における報酬設計や強化学習の手法に関する論文も参考になる。

※ AIが要旨から生成した要約です。正確性は原文をご確認ください。

著者: Yilin Zou, Chenghua Liu, Chenglong Wu, Fanghua Jiang

分類: cs.RO

原文アブストラクト

Learning a motion prior requires a reward that guides a policy from its current behavior toward demonstrated motion. Adversarial Motion Priors (AMP) provide such a reward with a discriminator. However, adversarial objectives can become uninformative when policy and expert supports are far apart. A naive use of optimal transport (OT) averages matched expert successors into a barycentric target. Averaging across gait phases can weaken the target's joint motion. We introduce Flow-Matched Motion Priors (FMP), an online scalar reward learned from paths connecting current rollout histories to an expert motion bank. Entropic OT supplies the coupling. Before each policy update, we train a neural potential with flow matching (FM) along the rollout-to-expert paths, endpoint-gradient supervision, and relative-value calibration. The actor receives only physical observations and the reward remains a scalar, as in AMP. Controlled reward-model experiments show substantially better generalization beyond the fitting rollout than value-only or endpoint-only fitting. On Unitree G1, matched 50-million-transition experiments compare FMP with AMP, a barycentric OT reward, and nested ablations under demonstration and fixed-pose initialization. FMP produces stable forward walking at 0.727 m/s from demonstration resets and 0.338 m/s from a fixed default pose. In the fixed-pose condition, it incurs 129 falls versus 243 for the endpoint-only control. Against a static score-gradient teacher, dynamic FM reduces score-increment error at interpolation fractions 0.25 and 0.50 while using 29% less offline fitting time.

関連論文