| 2026 | ACL | WebSTAR: Scalable Data Synthesis for Computer Use Agents with Step-Level Filtering. | Yifei He, Pranit Chawla, Yaser Souri, Subhojit Som, Xia Song |
| 2026 | COMPSAC | CovCraft: LLM-Guided Intelligent Framework for Constraint-Based Testing of Deep Learning Compiler Pipelines. | Yifei He, Fangyuan Yang, Ting Su, Yueling Zhang, Geguang Pu |
| 2026 | IWCMC | Broadband Scanning-Free Rydberg Atomic Communications: Hybrid Noise Modeling and Detector Design. | Yifei He, Tianqi Mao, Minze Chen, Yang Zhao, Meng Hua, Dezhi Zheng |
| 2025 | ACL | Scaling Laws for Multilingual Language Models. | Yifei He, Alon Benhaim, Barun Patra, Praneetha Vaddamanu, Sanchit Ahuja, Parul Chopra, Vishrav Chaudhary, Han Zhao, Xia Song |
| 2025 | EMNLP | Efficiently Editing Mixture-of-Experts Models with Compressed Experts. | Yifei He, Yang Liu, Chen Liang, Hany Hassan Awadalla |
| 2025 | NAACL | Towards Understanding the Fragility of Multilingual LLMs against Fine-Tuning Attacks. | Samuele Poppi, Zheng-Xin Yong, Yifei He, Bobbie Chern, Han Zhao, Aobo Yang, Jianfeng Chi |
| 2024 | CLUSTER | High-Performance FFT Code Generation via MLIR Linalg Dialect and SIMD Micro-Kernels. | Yifei He, Stefano Markidis |
| 2024 | CogSci | Frequency interacts with lexicality during auditory lexical decision: Insights from diffusion drift modeling. | Yifei He, Anna Petukhova, Jinxing Yue |
| 2024 | EMNLP | Semi-Supervised Reward Modeling via Iterative Self-Training. | Yifei He, Haoxiang Wang, Ziyan Jiang, Alexandros Papangelis, Han Zhao |
| 2024 | ICML | Robust Multi-Task Learning with Excess Risks. | Yifei He, Shiji Zhou, Guojun Zhang, Hyokun Yun, Yi Xu, Belinda Zeng, Trishul Chilimbi, Han Zhao |
| 2023 | EuroPar | Leveraging MLIR for Loop Vectorization and GPU Porting of FFT Libraries. | Yifei He, Artur Podobas, Stefano Markidis |
| 2022 | EuroPar | FFTc: An MLIR Dialect for Developing HPC Fast Fourier Transform Libraries. | Yifei He, Artur Podobas, Mns I. Andersson, Stefano Markidis |
| 2022 | UAI | Greedy modality selection via approximate submodular maximization. | Runxiang Cheng, Gargi Balasubramaniam, Yifei He, Yao-Hung Hubert Tsai, Han Zhao |
| 2021 | AAAI | A Hierarchical Approach to Multi-Event Survival Analysis. | Donna Tjandra, Yifei He, Jenna Wiens |