日本フィジカルAI新聞

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

模倣学習/逆強化学習arXiv:2605.00623

拡散ベースのポリシーにおける隠れた報酬の回復

Recovering Hidden Reward in Diffusion-Based Policies

シェア:XThreadsFacebookLINEはてブBluesky

拡散モデルと逆強化学習を統合し、スコア関数が専門家のソフトQ関数の勾配を回復することを利用して、敵対的学習なしで報酬を抽出するフレームワークを提案した。

詳しい要約

1. どんなもの?

本論文は EnergyFlow を提案する。これは generative action modeling と inverse reinforcement learning (IRL) を統合するフレームワークである。 - スカラー energy function をパラメータ化し、その勾配を denoising field とする。 - maximum-entropy optimality の下で、denoising score matching で学習した score function が expert の soft Q-function の勾配を回復することを示す。 - これにより adversarial training なしで reward extraction が可能となる。 - 操作タスクでの imitation 性能と下流 RL の reward signal で有効性を示す。

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

先行研究と比べて以下の点がすごい。 - adversarial IRL 手法や likelihood-based 代替手法を上回る reward signal を提供。 - adversarial training を必要とせずに reward extraction を実現。 - 学習した field を conservative に制約することで仮説複雑度を減らし、out-of-distribution generalization bounds を tight にすることを証明。 - 回復された reward の identifiability を特徴づけ、score estimation errors が action preferences に伝播する様子を bound。 - 有効な reward extraction に必要な構造制約が policy generalization の有益な inductive biases としても機能することを示す。

3. 技術・手法の肝は?

技術や手法の肝は以下。 - スカラー energy function をパラメータ化し、その勾配を denoising field として扱う。 - maximum-entropy optimality の下で、denoising score matching により学習された score function が expert の soft Q-function の勾配を回復する。 - 学習された field を conservative に制約することで、仮説複雑度を低減し、out-of-distribution generalization bounds を tight にする。 - reward の identifiability を特徴づけ、score estimation errors の action preferences への伝播を bound する。

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

有効性の検証は以下による。 - 様々な manipulation tasks において state-of-the-art の imitation performance を達成。 - 下流の reinforcement learning に対して有効な reward signal を提供し、adversarial IRL 手法や likelihood-based 代替手法を上回る。 - 理論的には、conservative 制約による仮説複雑度の低減と generalization bounds の tight 化、reward の identifiability、score estimation errors の伝播の bound を証明。

5. 議論はある?

議論は以下。 - 有効な reward extraction に必要な構造制約が、policy generalization のための有益な inductive biases として同時に機能することを示唆。 - 回復された reward の identifiability と score estimation errors の action preferences への伝播を bound することで、理論的保証を提供。 - ただし、要旨からは具体的な限界や失敗ケースについての議論は不明。

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

次に読むべき論文は以下。 - adversarial IRL 手法(要旨で比較されているが具体名は不明)。 - likelihood-based 代替手法(要旨で比較されているが具体名は不明)。 - denoising score matching や diffusion-based policies に関する研究。 - maximum-entropy inverse reinforcement learning の定番研究。 - soft Q-function や energy-based models に関する研究。

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

著者: Yanbiao Ji, Qiuchang Li, Yuting Hu, Shaokai Wu, Wenyuan Xie, Guodong Zhang, Qicheng He, Deyi Ji, Yue Ding, Hongtao Lu

分類: cs.RO

原文アブストラクト

This paper introduces EnergyFlow, a framework that unifies generative action modeling with inverse reinforcement learning by parameterizing a scalar energy function whose gradient is the denoising field. We establish that under maximum-entropy optimality, the score function learned via denoising score matching recovers the gradient of the expert's soft Q-function, enabling reward extraction without adversarial training. Formally, we prove that constraining the learned field to be conservative reduces hypothesis complexity and tightens out-of-distribution generalization bounds. We further characterize the identifiability of recovered rewards and bound how score estimation errors propagate to action preferences. Empirically, EnergyFlow achieves state-of-the-art imitation performance on various manipulation tasks while providing an effective reward signal for downstream reinforcement learning that outperforms both adversarial IRL methods and likelihood-based alternatives. These results show that the structural constraints required for valid reward extraction simultaneously serve as beneficial inductive biases for policy generalization. The code is available at https://github.com/sotaagi/EnergyFlow.

関連論文

  1. 1Boston Dynamics、車輪付きAtlasの研究試作機を紹介研究・技術
  2. 2GXOとApptronik、Apolloの物流現場試験を拡大 最大1,000台の展開を構想量産・展開
  3. 3newmoとKDDIが資本業務提携 2028年に堺市で自動運転タクシー商用化目指す提携・買収
  4. 4NVIDIA、CoRLでGR00T 2とIsaac Lab 3.0を発表新製品・発表
  5. 5UC BerkeleyとNVIDIAの研究者、AIエージェントでロボット制御コードを仮想環境で改善研究・技術

PR・アフィリエイト広告

FRONTIER(GPU搭載BTOパソコン)

FRONTIER(GPU搭載BTOパソコン)の公式広告バナーFRONTIERでGPU搭載PCを見る ↗
PR本紙発行元 EmplifAI