Model-Based Reinforcement Learning with Multi-Step Plan Value Estimation. - Haoxin Lin; Yihao Sun; Jiaji Zhang; Yang Yu 001 (ECAI, 2023)

Model-Bellman Inconsistency for Model-based Offline Reinforcement Learning. - Yihao Sun; Jiaji Zhang; Chengxing Jia; Haoxin Lin; Junyin Ye; Yang Yu 001 (ICML, 2023)

Episodic Return Decomposition by Difference of Implicitly Assigned Sub-trajectory Reward. - Haoxin Lin; Hongqiu Wu; Jiaji Zhang; Yihao Sun; Junyin Ye; Yang Yu 001 (AAAI, 2024)

WHALE: Towards Generalizable and Scalable World Models for Embodied Decision-making. - Zhilong Zhang; Ruifeng Chen 003; Junyin Ye; Yihao Sun; Pengyuan Wang; Jingcheng Pang; Kaiyuan Li; Tianshuo Liu; Haoxin Lin; Yang Yu 001; Zhi-Hua Zhou (CoRR, 2024)

Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning. - Haoxin Lin; Yu-Yan Xu; Yihao Sun; Zhilong Zhang; Yi-Chen Li 001; Chengxing Jia; Junyin Ye; Jiaji Zhang; Yang Yu 001 (ICLR, 2025)