Implicit Task-Driven Probability Discrepancy Measure for Unsupervised Domain Adaptation. - Mao Li; Kaiqi Jiang; Xinhua Zhang (NeurIPS, 2021)

How Does Adaptive Optimization Impact Local Neural Network Geometry? - Kaiqi Jiang; Dhruv Malik; Yuanzhi Li (NeurIPS, 2023)

Fairness Risks for Group-Conditionally Missing Demographics. - Kaiqi Jiang; Wenzhe Fan; Mao Li; Xinhua Zhang (AISTATS, 2025)

Understanding the Evolution of the Neural Tangent Kernel at the Edge of Stability. - Kaiqi Jiang; Jeremy Cohen; Yuanzhi Li (CoRR, 2025)