グラフ思考による報酬進化:強化学習のための二段階言語モデルフレームワーク
Reward Evolution with Graph-of-Thoughts: A Bi-Level Language Model Framework for Reinforcement Learning
LLMとVLMを組み合わせ、タスクをグラフ構造で分解して報酬関数を自動生成・改良する強化学習フレームワークを提案し、複数のロボット操作タスクで成功率を大幅に向上させた。
詳しい要約
1. どんなもの?
2. 先行研究と比べてどこがすごい?
3. 技術・手法の肝は?
4. どうやって有効だと検証した?
5. 議論はある?
6. 次に読むべき論文は?
※ AIが要旨から生成した要約です。正確性は原文をご確認ください。
著者: Changwei Yao, Xinzi Liu, Chen Li, Marios Savvides
分類: cs.RO
原文アブストラクト
Designing effective reward functions remains a major challenge in reinforcement learning (RL), often requiring considerable human expertise and iterative refinement. Recent advances leverage Large Language Models (LLMs) for automated reward design, but these approaches are limited by hallucinations, reliance on human feedback, and challenges with handling complex, multi-step tasks. In this work, we introduce Reward Evolution with Graph-of-Thoughts (RE-GoT), a novel bi-level framework that enhances LLMs with structured graph-based reasoning and integrates Visual Language Models (VLMs) for automated rollout evaluation. RE-GoT first decomposes tasks into text-attributed graphs, enabling comprehensive analysis and reward function generation, and then iteratively refines rewards using visual feedback from VLMs without human intervention. Extensive experiments on 10 RoboGen and 4 ManiSkill2 tasks demonstrate that RE-GoT consistently outperforms existing LLM-based baselines. On RoboGen, our method improves average task success rates by 32.25%, with notable gains on complex multi-step tasks. On ManiSkill2, RE-GoT achieves an average success rate of 93.73% across four diverse manipulation tasks, significantly surpassing prior LLM-based approaches and even exceeding expert-designed rewards. Our results indicate that combining LLMs and VLMs with graph-of-thoughts reasoning provides a scalable and effective solution for autonomous reward evolution in RL.
関連論文
- 進捗関係からロボットの報酬を学習するEvent2Reward強化学習/報酬設計
- ROOT: ユーザー指定の身体動作のための報酬発見フレームワーク強化学習/報酬設計
- 報酬設計エージェント:強化学習のための報酬設計強化学習/報酬設計
- 強化学習のための段階遷移型高密度報酬モデリング強化学習/報酬設計
- 生成的エピソードガイダンスによる二重粒度コントラスト報酬で効率的な身体性RL強化学習/報酬設計
- TimeRewarder:受動的動画からフレーム間時間距離で密報酬を学習強化学習/報酬設計
