Learning Generalizable Batch Active Learning Strategies via Deep Q-networks (Student Abstract). - Yi-Chen Li 001; Wen-Jie Shen; Boyu Zhang; Feng Mao; Zongzhang Zhang; Yang Yu 001 (AAAI, 2023)

Discovering Generalizable Multi-agent Coordination Skills from Multi-task Offline Data. - Fuxiang Zhang; Chengxing Jia; Yi-Chen Li 001; Lei Yuan 005; Yang Yu 001; Zongzhang Zhang (ICLR, 2023)

Policy Regularization with Dataset Constraint for Offline Reinforcement Learning. - Yuhang Ran; Yi-Chen Li 001; Fuxiang Zhang; Zongzhang Zhang; Yang Yu 001 (ICML, 2023)

Imitator Learning: Achieve Out-of-the-Box Imitation Ability in Variable Environments. - Xiong-Hui Chen; Junyin Ye; Hang Zhao; Yi-Chen Li 001; Haoran Shi; Yu-Yan Xu; Zhihao Ye; Si-Hang Yang; Anqi Huang; Kai Xu 004; Zongzhang Zhang; Yang Yu 001 (CoRR, 2023)

Cost-aware Offline Safe Meta Reinforcement Learning with Robust In-Distribution Online Task Adaptation. - Cong Guan; Ruiqi Xue; Ziqian Zhang; Lihe Li; Yi-Chen Li 001; Lei Yuan 005; Yang Yu 001 (AAMAS, 2024)

Disentangling Policy from Offline Task Representation Learning via Adversarial Data Augmentation. - Chengxing Jia; Fuxiang Zhang; Yi-Chen Li 001; Chenxiao Gao; Xu-Hui Liu; Lei Yuan 005; Zongzhang Zhang; Yang Yu 001 (AAMAS, 2024)

Deep Demonstration Tracing: Learning Generalizable Imitator Policy for Runtime Imitation from a Single Demonstration. - Xiong-Hui Chen; Junyin Ye; Hang Zhao; Yi-Chen Li 001; XuHui Liu; Haoran Shi; Yu-Yan Xu; Zhihao Ye; Si-Hang Yang; Yang Yu 001; Anqi Huang; Kai Xu 004; et al. (ICML, 2024)

Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics. - Xinyu Zhang; Wenjie Qiu 005; Yi-Chen Li 001; Lei Yuan 005; Chengxing Jia; Zongzhang Zhang; Yang Yu 001 (ICML, 2024)

Continual Multi-Objective Reinforcement Learning via Reward Model Rehearsal. - Lihe Li; Ruotong Chen; Ziqian Zhang; Zhichao Wu; Yi-Chen Li 001; Cong Guan; Yang Yu 001; Lei Yuan 005 (IJCAI, 2024)

Dynamics Adaptive Safe Reinforcement Learning with a Misspecified Simulator. - Ruiqi Xue; Ziqian Zhang; Lihe Li; Feng Chen 042; Yi-Chen Li 001; Yang Yu 001; Lei Yuan 005 (ECML/PKDD (7), 2024)

BWArea Model: Learning World Model, Inverse Dynamics, and Policy for Controllable Language Generation. - Chengxing Jia; Pengyuan Wang; Ziniu Li; Yi-Chen Li 001; Zhilong Zhang; Nan Tang; Yang Yu 001 (CoRR, 2024)

Q-Adapter: Training Your LLM Adapter as a Residual Q-Function. - Yi-Chen Li 001; Fuxiang Zhang; Wenjie Qiu 005; Lei Yuan 005; Chengxing Jia; Zongzhang Zhang; Yang Yu 001 (CoRR, 2024)

Improving Sample Efficiency of Reinforcement Learning with Background Knowledge from Large Language Models. - Fuxiang Zhang; Junyou Li; Yi-Chen Li 001; Zongzhang Zhang; Yang Yu 001; Deheng Ye (CoRR, 2024)

Q-Adapter: Customizing Pre-trained LLMs to New Preferences with Forgetting Mitigation. - Yi-Chen Li 001; Fuxiang Zhang; Wenjie Qiu 005; Lei Yuan 005; Chengxing Jia; Zongzhang Zhang; Yang Yu 001; Bo An 001 (ICLR, 2025)

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)

Constraining an Unconstrained Multi-agent Policy with offline data. - Cong Guan; Tao Jiang; Yi-Chen Li 001; Zongzhang Zhang; Lei Yuan 005; Yang Yu 001 (Neural Networks, 2025)

Sentence-level Reward Model can Generalize Better for Aligning LLM from Human Preference. - Wenjie Qiu 005; Yi-Chen Li 001; Xuqin Zhang; Tianyi Zhang; Yihang Zhang; Zongzhang Zhang; Yang Yu 001 (CoRR, 2025)

Controlling Large Language Model with Latent Actions. - Chengxing Jia; Ziniu Li; Pengyuan Wang; Yi-Chen Li 001; Zhenyu Hou; Yuxiao Dong; Yang Yu 001 (CoRR, 2025)

Generalist Reward Models: Found Inside Large Language Models. - Yi-Chen Li 001; Tian Xu 003; Yang Yu 001; Xuqin Zhang; Xiong-Hui Chen; Zhongxiang Ling; Ningjing Chao; Lei Yuan 005; Zhi-Hua Zhou (CoRR, 2025)

Generalizable Multi-Modal Adversarial Imitation Learning for Non-Stationary Dynamics. - Yi-Chen Li 001; Ningjing Chao; Zongzhang Zhang; Fuxiang Zhang; Lei Yuan 005; Yang Yu 001 (IEEE Trans. Pattern Anal. Mach. Intell., 2025)