| 2024 | CVPR | DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models. | Muyang Li, Tianle Cai, Jiaxin Cao, Qinsheng Zhang, Han Cai, Junjie Bai, Yangqing Jia, Kai Li, Song Han |
| 2024 | CVPR | OmniMotionGPT: Animal Motion Generation with Limited Data. | Zhangsihao Yang, Mingyuan Zhou, Mengyi Shan, Bingbing Wen, Ziwei Xuan, Mitch Hill, Junjie Bai, Guo-Jun Qi, Yalin Wang |
| 2024 | OSDI | MonoNN: Enabling a New Monolithic Optimization Space for Neural Network Inference Tasks on Modern GPU-Centric Architectures. | Donglin Zhuang, Zhen Zheng, Haojun Xia, Xiafei Qiu, Junjie Bai, Wei Lin, Shuaiwen Leon Song |
| 2023 | ASPLOS | RECom: A Compiler Approach to Accelerating Recommendation Model Inference with Massive Embedding Columns. | Zaifeng Pan, Zhen Zheng, Feng Zhang, Ruofan Wu, Hao Liang, Dalin Wang, Xiafei Qiu, Junjie Bai, Wei Lin, Xiaoyong Du |
| 2022 | IJCAI | Parameter-Efficient Sparsity for Large Language Models Fine-Tuning. | Yuchao Li, Fuli Luo, Chuanqi Tan, Mengdi Wang, Songfang Huang, Shen Li, Junjie Bai |
| 2021 | EuroSys | DISC: A Dynamic Shape Compiler for Machine Learning Workloads. | Kai Zhu, Wenyi Zhao, Zhen Zheng, Tianyou Guo, Pengzhan Zhao, Junjie Bai, Jun Yang, Xiaoyong Liu, Lansong Diao, Wei Lin |
| 2015 | MICCAI | Multiple Surface Segmentation Using Truncated Convex Priors. | Abhay Shah, Junjie Bai, Zhihong Hu, Srinivas R. Sadda, Xiaodong Wu |
| 2014 | ACCV | Fully Automatic Segmentation of Hip CT Images via Random Forest Regression-Based Atlas Selection and Optimal Graph Search-Based Surface Detection. | Chengwen Chu, Junjie Bai, Li Liu, Xiaodong Wu, Guoyan Zheng |
| 2014 | CVPR | Error-Tolerant Scribbles Based Interactive Image Segmentation. | Junjie Bai, Xiaodong Wu |
| 2012 | CVPR | Fast dynamic programming for labeling problems with ordering constraints. | Junjie Bai, Qi Song, Olga Veksler, Xiaodong Wu |
| 2012 | MICCAI | 3D Lung Tumor Motion Model Extraction from 2D Projection Images of Mega-voltage Cone Beam CT via Optimal Graph Search. | Mingqing Chen, Junjie Bai, Yefeng Zheng, R. Alfredo C. Siochi |