Unified token representations for sequential decision models

Zhuojing Tian, Yushu Chen

Transformers have demonstrated strong potential in offline reinforcement learning (RL) by modeling trajectories as sequences of return-to-go, states, and actions. However, existing…