Bai, He

8 publications

L4DC 2025 Exploiting Inter-Agent Coupling Information for Efficient Reinforcement Learning of Cooperative LQR Shahbaz P Qadri Syed, He Bai

AAAI 2025 Federated Recommendation with Explicitly Encoding Item Bias Zhihao Wang, He Bai, Wenke Huang, Duantengchuan Li, Jian Wang, Bing Li

L4DC 2025 Scalability Enhancement and Data-Heterogeneity Awareness in Gradient Tracking Based Decentralized Bayesian Learning Kinjal Bhar, He Bai, Jemin George, Carl Busart

ICLRW 2024 How Far Are We from Intelligent Visual Deductive Reasoning? Yizhe Zhang, He Bai, Ruixiang Zhang, Jiatao Gu, Shuangfei Zhai, Joshua M. Susskind, Navdeep Jaitly

ICLRW 2024 Rephrasing the Web: A Recipe for Compute and Data-Efficient Language Modeling Pratyush Maini, Skyler Seto, He Bai, David Grangier, Yizhe Zhang, Navdeep Jaitly

ICML 2022 A$^3$T: Alignment-Aware Acoustic and Text Pretraining for Speech Synthesis and Editing He Bai, Renjie Zheng, Junkun Chen, Mingbo Ma, Xintong Li, Liang Huang

AAAI 2021 Segatron: Segment-Aware Transformer for Language Modeling and Understanding He Bai, Peng Shi, Jimmy Lin, Yuqing Xie, Luchen Tan, Kun Xiong, Wen Gao, Ming Li

NeurIPS 2020 Decentralized Langevin Dynamics for Bayesian Learning Anjaly Parayil, He Bai, Jemin George, Prudhvi Gurram