- 
             
            Yihan Wang*, Yiwei Lu*, Xiao-Shan Gao, Gautam Kamath, Yaoliang Yu. 
            Proceedings of the 39th Conference on Neural Information Processing Systems (NeurIPS 2025). 
- 
            
            William Xu, Yiwei Lu,  Yihan Wang , Matthew Y.R. Yang, Zuoqiu Liu, Gautam Kamath, Yaoliang Yu. 
- 
            
            Yihan Wang*, Yiwei Lu*, Guojun Zhang, Franziska Boenisch, Adam Dziedzic, Yaoliang Yu, Xiao-Shan Gao. Transactions on Machine Learning Research (TMLR), 2025 
- 
            
            Shuang Liu, Yihan Wang, Xiao-Shan Gao.
            Proceedings of the 13th International Conference on Learning Representations (ICLR 2025) 
- 
            
            Yihan Wang, Shuang Liu, Xiao-Shan Gao.
            Neural Networks, Volume 183, 2025 
- 
             
            Yihan Wang, Yifan Zhu, Xiao-Shan Gao. 
            Proceedings of the 38th Conference on Neural Information Processing Systems (NeurIPS 2024). 
- 
            
            Yiwei Lu, Yihan Wang, Guojun Zhang, Yaoliang Yu. ICML 2024 Next Generation of AI Safety Workshop. 
- 
            
            Shuang Liu, Yihan Wang, Xiao-Shan Gao.
            Proceedings of the 38th Annual AAAI Conference on Artificial Intelligence (AAAI 2024). 
- 
            
            Lijia Yu, Yihan Wang, Xiao-Shan Gao.  Proceedings of the 40th International Conference on Machine Learning (ICML 2023). 
- 
            
            Yihan Wang, Lijia Yu, Xiao-Shan Gao.
            Proceedings of the 30th International Conference on Neural Information Processing (ICONIP 2023).