Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Shilin, Hu, Zican, Wu, Wenhao, Xie, Xinyi, Tang, Jianxiang, Chen, Chunlin, Dong, Daoyi, Cheng, Yu, Sun, Zhenhong, Wang, Zhi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918214085115904
author Zhang, Shilin
Hu, Zican
Wu, Wenhao
Xie, Xinyi
Tang, Jianxiang
Chen, Chunlin
Dong, Daoyi
Cheng, Yu
Sun, Zhenhong
Wang, Zhi
author_facet Zhang, Shilin
Hu, Zican
Wu, Wenhao
Xie, Xinyi
Tang, Jianxiang
Chen, Chunlin
Dong, Daoyi
Cheng, Yu
Sun, Zhenhong
Wang, Zhi
contents Offline meta-RL usually tackles generalization by inferring task beliefs from high-quality samples or warmup explorations. The restricted form limits their generality and usability since these supervision signals are expensive and even infeasible to acquire in advance for unseen tasks. Learning directly from the raw text about decision tasks is a promising alternative to leverage a much broader source of supervision. In the paper, we propose \textbf{T}ext-to-\textbf{D}ecision \textbf{A}gent (\textbf{T2DA}), a simple and scalable framework that supervises offline meta-RL with natural language. We first introduce a generalized world model to encode multi-task decision data into a dynamics-aware embedding space. Then, inspired by CLIP, we predict which textual description goes with which decision embedding, effectively bridging their semantic gap via contrastive language-decision pre-training and aligning the text embeddings to comprehend the environment dynamics. After training the text-conditioned generalist policy, the agent can directly realize zero-shot text-to-decision generation in response to language instructions. Comprehensive experiments on MuJoCo and Meta-World benchmarks show that T2DA facilitates high-capacity zero-shot generalization and outperforms various types of baselines. Our code is available at \textcolor{magenta}{\href{https://github.com/NJU-RL/T2DA}{https://github.com/NJU-RL/T2DA}}.
format Preprint
id arxiv_https___arxiv_org_abs_2504_15046
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision
Zhang, Shilin
Hu, Zican
Wu, Wenhao
Xie, Xinyi
Tang, Jianxiang
Chen, Chunlin
Dong, Daoyi
Cheng, Yu
Sun, Zhenhong
Wang, Zhi
Artificial Intelligence
Offline meta-RL usually tackles generalization by inferring task beliefs from high-quality samples or warmup explorations. The restricted form limits their generality and usability since these supervision signals are expensive and even infeasible to acquire in advance for unseen tasks. Learning directly from the raw text about decision tasks is a promising alternative to leverage a much broader source of supervision. In the paper, we propose \textbf{T}ext-to-\textbf{D}ecision \textbf{A}gent (\textbf{T2DA}), a simple and scalable framework that supervises offline meta-RL with natural language. We first introduce a generalized world model to encode multi-task decision data into a dynamics-aware embedding space. Then, inspired by CLIP, we predict which textual description goes with which decision embedding, effectively bridging their semantic gap via contrastive language-decision pre-training and aligning the text embeddings to comprehend the environment dynamics. After training the text-conditioned generalist policy, the agent can directly realize zero-shot text-to-decision generation in response to language instructions. Comprehensive experiments on MuJoCo and Meta-World benchmarks show that T2DA facilitates high-capacity zero-shot generalization and outperforms various types of baselines. Our code is available at \textcolor{magenta}{\href{https://github.com/NJU-RL/T2DA}{https://github.com/NJU-RL/T2DA}}.
title Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision
topic Artificial Intelligence
url https://arxiv.org/abs/2504.15046