日本フィジカルAI新聞

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

オフライン強化学習arXiv:2611.04010

タスク認識表現を備えたDecision Transformerによるオフライン・メタ強化学習

Decision Transformer with Task-Aware Representations for Offline Meta-Reinforcement Learning

シェア:XThreadsFacebookLINEはてブBluesky

オフライン軌跡の対照学習や言語記述との整合など4要素を統合してタスク表現を学習し、それを条件とするDecision Transformer型メタ方策で未知タスクへのゼロショット・少数ショット適応を実現した研究。

詳しい要約

1. どんなもの?

- 本論文は CoTA-DT (Contrastive Task-Aware Decision Transformer) を提案する。 - DT (Decision Transformer) ベースの offline meta-RL フレームワーク。 - 4つの要素を統合して latent task representation を学習する。 - offline trajectory からの contrastive learning - natural-language task description との alignment - task-conditioned world model による one-step supervision - task-conditioned Q function による long-term supervision - 学習した表現を DT ベース policy の conditioning signal として用いる。 - zero-shot / few-shot 両設定での generalizatio…

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

- 従来の offline meta-RL では latent task representation の推論精度が generalization の鍵とされる。 - CoTA-DT は単一の表現学習ではなく4種の相補的 supervision を統合する点が特徴。 - contrastive learning、language alignment、world model、Q function を併用。 - pretrained LLM を language encoding、world modeling、Q-function estimation、policy learning の共有 backbone とする統合設計。 - parameter sharing により efficiency と generalization の向上を狙う。 - 実験では prior methods を一貫して上回り、特に zero-shot generalization で優位と報告。 - 具体的な比較対象名は要旨からは不明。

3. 技術・手法の肝は?

- 中核は task representation の学習と DT policy への conditioning。 - 4つの supervision を同時に統合する。 - offline trajectory からの contrastive learning - natural-language task description との alignment - task-conditioned world model による one-step supervision - task-conditioned Q function による long-term supervision - pretrained LLM を共有 backbone として用いる。 - language encoding、world modeling、Q-function estimation、policy learning を担う。 - zero-shot / few-shot 適応のため language-grounded で self-adaptive な戦略を設計。 - 各コンポーネント…

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

- 複数の diverse な offline meta-RL benchmarks で実験。 - CoTA-DT が prior methods を一貫して上回ると報告。 - 特に challenging な zero-shot generalization シナリオで優位。 - zero-shot と few-shot の両設定を対象に generalization を評価。 - 具体的な benchmark 名、baseline 名、指標、ablation の詳細は要旨からは不明。

5. 議論はある?

- 4つの supervision と LLM 共有 backbone の統合が generalization に寄与するという主張。 - parameter sharing による efficiency と generalization の両立を議論。 - zero-shot / few-shot 適応のための language-grounded self-adaptive 戦略の有効性に言及。 - 限界や failure case、計算コスト、ablation の詳細な議論は要旨からは不明。 - 各コンポーネントの寄与度の定量分析も要旨からは不明。

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

- 要旨で明示的に参照・比較されている個別研究名は不明。 - 関連手法として以下が挙げられる。 - Decision Transformer (DT) - offline meta-RL - contrastive learning による task representation 学習 - task-conditioned world model - task-conditioned Q function - pretrained LLM を backbone とする手法 - 同分野の定番として model-based offline RL や meta-RL の代表的手法を次に読む候補とする。

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

著者: Xinbo Zhao, Yuhang Liu, Xin Zhang, Lei Le, Zhaohan Xi, Yingxue Zhang

分類: cs.LG

原文アブストラクト

To enable effective generalization in offline meta-RL, it is essential to accurately infer the underlying task through latent task representations, which provide a compact and transferable summary of task-specific characteristics for guiding policy generation. In this paper, we propose Contrastive Task-Aware Decision Transformer (CoTA-DT), a DT-based offline meta-RL framework that learns task representations by jointly integrating four complementary components: (i) contrastive learning from offline trajectories, (ii) alignment with natural-language task descriptions, (iii) one-step supervision via a task-conditioned world model, and (iv) long-term supervision via a task-conditioned Q function. The learned task representations are used as conditioning signals for a DT-based policy, enabling task-aware meta-policy learning. To further support generalization in both zero-shot and few-shot settings, we design a language-grounded, self-adaptive strategy that enables effective and reliable adaptation to unseen tasks. Notably, in CoTA-DT, the pretrained LLM serves as a shared backbone for language encoding, world modeling, Q-function estimation, and policy learning. This unified design facilitates parameter sharing, improving both efficiency and generalization. Extensive experiments across diverse offline meta-RL benchmarks demonstrate that CoTA-DT consistently outperforms prior methods, particularly in challenging zero-shot generalization scenarios.

関連論文

  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・アフィリエイト広告

MINISFORUM(ミニPC)

MINISFORUM(ミニPC)の公式広告バナーMINISFORUM公式ストアを見る ↗
PR本紙発行元 EmplifAI