Publications

  1. Pre-trained Large Language Models Learn Hidden Markov Models In-context
    Dai, Yijia,  Gao, Zhaolin, Sattar, Yahya, Dean, Sarah, and Sun, Jennifer J.
    Preprint
  2. Accelerating RL for LLM Reasoning with Optimal Advantage Regression
    Brantley, Kianté, Chen, Mingyu,  Gao, Zhaolin, Lee, Jason D., Sun, Wen, Zhan, Wenhao, and Zhang, Xuezhou (alphabetical order)
    Preprint
  3. Value-Guided Search for Efficient Chain-of-Thought Reasoning
    Wang, Kaiwen, Zhou, Jin Peng, Chang, Jonathan D.,  Gao, Zhaolin, Kallus, Nathan, Brantley, Kianté, and Sun, Wen
    Preprint
  4. Q#: Provably Optimal Distributional RL for LLM Post-Training
    Zhou, Jin Peng*, Wang, Kaiwen*, Chang, Jonathan D.,  Gao, Zhaolin, Kallus, Nathan, Weinberger, Kilian Q., Brantley, Kianté, and Sun, Wen
    Preprint
  5. Regressing the Relative Future: Efficient Policy Optimization for Multi-turn RLHF
    Gao, Zhaolin, Zhan, Wenhao, Chang, Jonathan D., Swamy, Gokul, Brantley, Kianté, Lee, Jason D., and Sun, Wen
    ICLR 2025
  6. End-to-end Training for Recommendation with Language-based User Profiles
    Gao, Zhaolin, Zhou, Joyce, Dai, Yijia, and Joachims, Thorsten
    CIKM 2025
  7. REBEL: Reinforcement Learning via Regressing Relative Rewards
    Gao, Zhaolin, Chang, Jonathan D., Zhan, Wenhao, Oertell, Owen, Swamy, Gokul, Brantley, Kianté, Joachims, Thorsten, Bagnell, J. Andrew, Lee, Jason D., and Sun, Wen
    NeurIPS 2024
  8. Reviewer2: Optimizing Review Generation Through Prompt Generation
    Gao, Zhaolin, Brantley, Kianté, and Joachims, Thorsten
    Preprint
  9. Session-based Recommendation With Transformers
    Lu, Yichao,  Gao, Zhaolin*, Cheng, Zhaoyue*, Sun, Jianing*, Brown, Bradley, Yu, Guangwei, Wong, Anson, Perez, Felipe, and Volkovs, Maksims
    RecSys Challenge 2022
  10. Mitigating the Filter Bubble while Maintaining Relevance: Targeted Diversification with VAE-based Recommender Systems
    Gao, Zhaolin, Shen, Tianshu, Mai, Zheda, Bouadjenek, Mohamed Reda, Waller, Isaac, Anderson, Ashton, Bodkin, Ron, and Sanner, Scott
    SIGIR 2022
  11. MCL: Mixed-Centric Loss for Collaborative Filtering
    Gao, Zhaolin*, Cheng, Zhaoyue*, Perez, Felipe, Sun, Jianing, and Volkovs, Maksims
    WWW 2022
  12. Shoestring: Graph-Based Semi-Supervised Classification With Severely Limited Labeled Data
    Lin, Wanyu,  Gao, Zhaolin, and Li, Baochun
    CVPR 2020
  13. Guardian: Evaluating Trust in Online Social Networks with Graph Convolutional Networks
    Lin, Wanyu,  Gao, Zhaolin, and Li, Baochun
    INFOCOM 2020