Yihao Sun

PhD Student, Mila & University of Montreal
Supervisor: Prof. Pierre-Luc Bacon
Email: yihao.sun@mila.quebec

[ Google Scholar ] [ Github ]

Biography

I’m a first-year PhD student at Mila & University of Montreal. Before this, I just had a three-year memorable graduate study in LAMDA Group, Nanjing University.

Research Interests

My research interest is reinforcement learning (RL). Currently, I focus on offline reinforcement learning (Offline RL) and model-based reinforcement learning (MBRL).

News

Sep 26, 2024 Our work Provably and practically efficient adversarial imitation learning with general function approximation is accepted by NeurIPS 2024!
May 3, 2024 Our work Policy Representation Can be Utilized for More Generalizable Offline Dynamics Model Learning is accepted by ICML 2024!
Jan 16, 2024 Our work Flow to better: Offline preference-based reinforcement learning via preferred trajectory generation is accepted by ICLR 2024!
Dec 9, 2023 Our work Episodic return decomposition by difference of implicitly assigned sub-trajectory reward is accepted by AAAI 2024!

Selected Publications

* indicates equal contribution
  1. ICML
    Model-Bellman inconsistency for model-based offline reinforcement learning | [ Link Code ]
    Yihao Sun* , Jiaji Zhang*, Chengxing Jia, Haoxin Lin, Junyin Ye, and Yang Yu.
    In Proceedings of the 40th International Conference on Machine Learning (ICML’23). 2023.
  2. ICLR
    Flow to better: Offline preference-based reinforcement learning via preferred trajectory generation | [ Link Code ]
    Zhilong Zhang*, Yihao Sun* , Junyin Ye, Tianshuo Liu, Jiaji Zhang, and Yang Yu.
    In Proceedings of the 12th International Conference on Learning Representations (ICLR’24). 2024.


    Correspondence


    Email: yihao.sun@mila.quebec

    Laboratory: 6666, St-Urbain, #200 Montreal, Quebec, H2S 3H1