| 2026 | IWCMC | TPPPA: A Triangular Partition Path Planning Algorithm for UAV Coverage in Irregular Areas. | Ziyi Guo, Qian Dong, Gordon Owusu Boateng, Yihao Xue, Bintao Hu, Xingzhen Duan |
| 2025 | ICML | Representations Shape Weak-to-Strong Generalization: Theoretical Insights and Empirical Predictions. | Yihao Xue, Jiping Li, Baharan Mirzasoleiman |
| 2024 | ICLR | Investigating the Benefits of Projection Head for Representation Learning. | Yihao Xue, Eric Gan, Jiayi Ni, Siddharth Joshi, Baharan Mirzasoleiman |
| 2024 | ICLR | Understanding the Robustness of Multi-modal Contrastive Learning to Distribution Shift. | Yihao Xue, Siddharth Joshi, Dang Nguyen, Baharan Mirzasoleiman |
| 2024 | ICML | Few-shot Adaptation to Distribution Shifts By Mixing Source and Target Embeddings. | Yihao Xue, Ali Payani, Yu Yang, Baharan Mirzasoleiman |
| 2024 | UAI | Investigating the Impact of Model Width and Density on Generalization in Presence of Label Noise. | Yihao Xue, Kyle Whitecross, Baharan Mirzasoleiman |
| 2023 | ICML | Which Features are Learnt by Contrastive Learning? On the Role of Simplicity Bias in Class Collapse and Feature Suppression. | Yihao Xue, Siddharth Joshi, Eric Gan, Pin-Yu Chen, Baharan Mirzasoleiman |
| 2022 | ICML | Investigating Why Contrastive Learning Benefits Robustness against Label Noise. | Yihao Xue, Kyle Whitecross, Baharan Mirzasoleiman |
| 2021 | AAAI | Toward Understanding the Influence of Individual Clients in Federated Learning. | Yihao Xue, Chaoyue Niu, Zhenzhe Zheng, Shaojie Tang, Chengfei Lyu, Fan Wu, Guihai Chen |