AndyCao1125 / MambaDM

13 stars 1 forks source link


Mamba LogoMamba as Decision Maker:
Exploring Multi-scale Sequence Modelling in
Offline Reinforcement Learning

Jiahang Cao1 *, Qiang Zhang1 *, Ziqing Wang2, Jingkai Sun1, Jiaxu Wang1, Hao Cheng1,
Yecheng Shao4, Wen Zhao3, Gang Han3, Yijie Guo3, Renjing Xu1 †,

The Hong Kong University of Science and Technology (Guangzhou) 1
Northwestern University 2
Beijing Innovation Center of Humanoid Robotics 3
Center for X-Mechanics, Zhejiang University 4

Paper PDF

Codes are coming soon.