| 2025 | IJCNN | MRAFF: Multi-resolution Attention Feature Fusion Method for Multi-label Fundus Image Ocular Disease Screening. | Chucheng Chen, Zheng Gong, Weihao Gao, Zhuo Deng, Lei Shao, Li Dong, Wenbin Wei, Lan Ma |
| 2025 | IJCNN | UMGen: Multi-scale and Multi-region Joint Prediction Pipeline for Uveal Melanoma Prognosis-related Gene Subtyping. | Zhuo Deng, Zheng Gong, Ruiheng Zhang, Li Dong, Weihao Gao, Lei Shao, Wenbin Wei, Lan Ma |
| 2024 | ICANN | ProTeM: Unifying Protein Function Prediction via Text Matching. | Ming Qin, Xun Li, Yuhao Wang, Zhenping Li, Hongbin Ye, Zongbing Wang, Weihao Gao, Shangsong Liang, Qiang Zhang, Keyan Ding |
| 2024 | ICANN | Enhancing LM's Task Adaptability: Powerful Post-training Framework with Reinforcement Learning from Model Feedback. | Fuju Rong, Weihao Gao, Zhuo Deng, Zheng Gong, Chucheng Chen, Wenze Zhang, Zhiyuan Niu, Fang Li, Lan Ma |
| 2024 | ICANN | Predicting City Origin-Destination Flow with Generative Pre-training. | Mingwei Zhang, Lizhong Gao, Qiao Wang, Weihao Gao |
| 2023 | ICML | Machine Learning Force Fields with Data Cost Aware Training. | Alexander Bukharin, Tianyi Liu, Shengjie Wang, Simiao Zuo, Weihao Gao, Wen Yan, Tuo Zhao |
| 2022 | ICLR | Label Leakage and Protection in Two-party Split Learning. | Oscar Li, Jiankai Sun, Xin Yang, Weihao Gao, Hongyi Zhang, Junyuan Xie, Virginia Smith, Chong Wang |
| 2022 | UAI | PathFlow: A normalizing flow generator that finds transition paths. | Tianyi Liu, Weihao Gao, Zhirui Wang, Chong Wang |
| 2021 | CIKM | Learning An End-to-End Structure for Retrieval in Large-Scale Recommendations. | Weihao Gao, Xiangjun Fan, Chong Wang, Jiankai Sun, Kai Jia, Wenzhi Xiao, Ruofan Ding, Xingyan Bin, Hui Yang, Xiaobing Liu |
| 2020 | AAAI | Information-Theoretic Understanding of Population Risk Improvement with Model Compression. | Yuheng Bu, Weihao Gao, Shaofeng Zou, Venugopal V. Veeravalli |
| 2019 | AISTATS | Learning One-hidden-layer Neural Networks under General Input Distributions. | Weihao Gao, Ashok Vardhan Makkuva, Sewoong Oh, Pramod Viswanath |
| 2019 | ICML | Rate Distortion For Model Compression: From Theory To Practice. | Weihao Gao, Yu-Han Liu, Chong Wang, Sewoong Oh |
| 2017 | ISIT | Demystifying fixed k-nearest neighbor information estimators. | Weihao Gao, Sewoong Oh, Pramod Viswanath |
| 2017 | ISIT | Density functional estimators with k-nearest neighbor bandwidths. | Weihao Gao, Sewoong Oh, Pramod Viswanath |
| 2016 | ICML | Conditional Dependence via Shannon Capacity: Axioms, Estimators and Applications. | Weihao Gao, Sreeram Kannan, Sewoong Oh, Pramod Viswanath |
| 2014 | CISS | On the bit error rate of repeated error-correcting codes. | Weihao Gao, Yury Polyanskiy |