人工知能学会全国大会論文集
Online ISSN : 2758-7347
第38回 (2024)
セッションID: 2O6-OS-16a-04
会議情報

物体中心表現を用いた世界モデルの獲得
*中野 聡大鈴木 雅大松尾 豊
著者情報
会議録・要旨集 フリー

詳細
抄録

強化学習タスクにおいてモデルベース強化学習は,画像と行動情報のみから,想像によって世界モデルを獲得し,長期の予測を必要とするタスクにおいて複雑な行動学習が可能な手法である.特にTransformerを遷移モデルに適用した世界モデルは,長期依存性が捉えられるようになり,モデルの学習のサンプル効率性を大幅に改善した.しかし,世界モデルを用いた手法によって構成的なタスクを解くことは難しい.オブジェクトの相互作用を予測したり,オブジェクトを正確に追跡したりすることが主な問題である.物体中心学習は,シーンを構成的にdisentangleし,個々の物体を別々の潜在変数によって表すことができる学習手法であり,シーンの構成的な理解に加えて,未知物体・シーンへの汎化性能が上がることが知られている.本論文では,物体中心学習の優れた汎化性能と構成性と,Transformerベースの世界モデルのサンプル効率性と長期予測能力と組み合わせることを目的として,物体中心な表現を用いたTransformerベースの世界モデルを提案する.提案モデルの効果を検証するために,OCRLのベンチマークで実験を行った.

著者関連情報
© 2024 人工知能学会
前の記事 次の記事
feedback
Top