| 2026 | HPDC | Full-Core Fluid-Structure-Interaction Simulation of Nuclear Reactor on CPU+GPU Hybrid Clusters. | Xue Miao, Jue Wang, Qida Lin, Shufei Zhang, Rongqiang Cao, Chunbao Zhou, Ningming Nie, He Bai, Yangang Wang |
| 2025 | AAAI | Federated Recommendation with Explicitly Encoding Item Bias. | Zhihao Wang, He Bai, Wenke Huang, Duantengchuan Li, Jian Wang, Bing Li |
| 2025 | ICASSP | MetaCon: Revitalizing Internet Congestion Control with Meta-Reinforcement Learning. | He Bai, Hui Li, Jianming Que, Minglong Zhang, Peter Han Joo Chong, Kalupahana Liyanage Kushan Sudheera, Xinyuan Pei |
| 2025 | ICASSP | Exploring Prediction Targets in Masked Pre-Training for Speech Foundation Models. | Li-Wei Chen, Takuya Higuchi, He Bai, Ahmed Hussen Abdelaziz, Shinji Watanabe, Alexander Rudnicky, Tatiana Likhomanenko, Barry-John Theobald, Zakaria Aldeneh |
| 2025 | ICPP | Pisces: Towards Adaptive and Fair Congestion Control via Multi-Agent Meta-Reinforcement Learning. | He Bai, Hui Li, Jianming Que, Minglong Zhang, Zhiqiang Hu, Ximing Xu, Bing Lin, Runhuai Huang, Junyang Qiu, Shaowen Deng |
| 2025 | SC | MISA-AKMC : Achieve Kinetic Monte Carlo Simulation of 20 Quadrillion Atoms on GPU Clusters. | Shunde Li, Zhijie Pan, Ningming Nie, Jue Wang, He Bai, Genshen Chu, Yan Zeng, Xinfu He, Yangang Wang, Changjun Hu, Xuebin Chi |
| 2024 | COLING | Construction of Paired Knowledge Graph - Text Datasets Informed by Cyclic Evaluation. | Ali Mousavi, Xin Zhan, He Bai, Peng Shi, Theodoros Rekatsinas, Benjamin Han, Yunyao Li, Jeffrey Pound, Joshua M. Susskind, Natalie Schluter, Ihab F. Ilyas, Navdeep Jaitly |
| 2024 | CSCWD | A Scalable and Resilient Protocol for Synchronous Collaboration. | Zhengqi Wu, Hui Li, He Bai, Xinyuan Pei |
| 2024 | IROS | An LSTM-based Model to Recognize Driving Style and Predict Acceleration. | Jiaxing Lu, Sanzida Hossain, Weihua Sheng, He Bai |
| 2024 | IROS | Real-time Dexterous Telemanipulation with an End-Effect-Oriented Learning-based Approach. | Haoyang Wang, He Bai, Xiaoli Zhang, Yunsik Jung, Michel Bowman, Lingfeng Tao |
| 2023 | HCI | A Study on Workload Assessment and Usability of Wind-Aware User Interface for Small Unmanned Aircraft System Remote Operations. | Asma Tabassum, He Bai, Nicoletta Fala |
| 2023 | HPDC | Efficient Algorithm Design of Optimizing SpMV on GPU. | Genshen Chu, Yuanjie He, Lingyu Dong, Zhezhao Ding, Dandan Chen, He Bai, Xuesong Wang, Changjun Hu |
| 2023 | ICRA | Cooperative Driving in Mixed Traffic of Manned and Unmanned Vehicles based on Human Driving Behavior Understanding. | Jiaxing Lu, Sanzida Hossain, Weihua Sheng, He Bai |
| 2023 | IROS | Incorporating Stochastic Human Driving States in Cooperative Driving Between a Human-Driven Vehicle and an Autonomous Vehicle. | Sanzida Hossain, Jiaxing Lu, He Bai, Weihua Sheng |
| 2022 | ACL | Better Language Model with Hypernym Class Prediction. | He Bai, Tong Wang, Alessandro Sordoni, Peng Shi |
| 2022 | EMNLP | Cross-lingual Text-to-SQL Semantic Parsing with Representation Mixup. | Peng Shi, Linfeng Song, Lifeng Jin, Haitao Mi, He Bai, Jimmy Lin, Dong Yu |
| 2022 | EMNLP | XRICL: Cross-lingual Retrieval-Augmented In-Context Learning for Cross-lingual Text-to-SQL Semantic Parsing. | Peng Shi, Rui Zhang, He Bai, Jimmy Lin |
| 2022 | HPCC | MISA-MD: A New Design of Molecular Dynamics Software for GPU Architecture. | Genshen Chu, Tianyi Chen, Dandan Chen, He Bai, Guannan Li, Shuai Ren, Yang Li |
| 2022 | ICML | A | He Bai, Renjie Zheng, Jun-Kun Chen, Mingbo Ma, Xintong Li, Liang Huang |
| 2022 | IROS | Development of a Research Testbed for Cooperative Driving in Mixed Traffic of Human-driven and Autonomous Vehicles. | Jiaxing Lu, Ryan Stracener, Weihua Sheng, He Bai, Sanzida Hossain |
| 2022 | WCNC | DSCCP: A Differentiated Service-based Congestion Control Protocol for Information-Centric Networking. | He Bai, Hui Li, Jianming Que, Minglong Zhang, Peter Han Joo Chong |
| 2021 | AAAI | 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 |
| 2021 | ACL | Semantics of the Unwritten: The Effect of End of Paragraph and Sequence Tokens on Text Generation with GPT2. | He Bai, Peng Shi, Jimmy Lin, Luchen Tan, Kun Xiong, Wen Gao, Jie Liu, Ming Li |
| 2020 | EMNLP | Cross-Lingual Training of Neural Models for Document Ranking. | Peng Shi, He Bai, Jimmy Lin |
| 2020 | FUSION | A Variational Bayesian Approach for Estimating System Parameters and Process Noise. | He Bai, Clark N. Taylor |
| 2019 | ACL | Memory Consolidation for Contextual Spoken Language Understanding with Dialogue Logistic Inference. | He Bai, Yu Zhou, Jiajun Zhang, Chengqing Zong |
| 2018 | COLING | Source Critical Reinforcement Learning for Transferring Spoken Language Understanding to a New Language. | He Bai, Yu Zhou, Jiajun Zhang, Liang Zhao, Mei-Yuh Hwang, Chengqing Zong |
| 2014 | ICARCV | Uncertainty estimation for random sample consensus. | Huili Yu, Shalini Keshavamurthy, He Bai, Sameer Sheorey, Hieu T. Nguyen, Clark N. Taylor |
| 2011 | SIGIR | UPS: efficient privacy protection in personalized web search. | Gang Chen, He Bai, Lidan Shou, Ke Chen, Yunjun Gao |
| 2007 | ACC | Group Coordination when the Reference Velocity is Available Only to the Leader: An Adaptive Design. | He Bai, Murat Arcak, John T. Wen |