| 2026 | ACL | Rethinking Parameter Sharing for LLM Fine-Tuning with Multiple LoRAs. | Hao Ban, Kaiyi Ji |
| 2025 | AAAI | First-Order Federated Bilevel Learning. | Yifan Yang, Peiyao Xiao, Shiqian Ma, Kaiyi Ji |
| 2025 | ICCV | SAMO: A Lightweight Sharpness-Aware Approach for Multi-Task Optimization with Joint Global-Local Perturbation. | Hao Ban, Gokul Ram Subramani, Kaiyi Ji |
| 2025 | ICLR | Tuning-Free Bilevel Optimization: New Algorithms and Convergence Analysis. | Yifan Yang, Hao Ban, Minhui Huang, Shiqian Ma, Kaiyi Ji |
| 2025 | ICLR | MGDA Converges under Generalized Smoothness, Provably. | Qi Zhang, Peiyao Xiao, Shaofeng Zou, Kaiyi Ji |
| 2024 | ICLR | AUC-CL: A Batchsize-Robust Framework for Self-Supervised Contrastive Representation Learning. | Rohan Sharma, Kaiyi Ji, Zhiqiang Xu, Changyou Chen |
| 2024 | ICML | Fair Resource Allocation in Multi-Task Learning. | Hao Ban, Kaiyi Ji |
| 2024 | ICML | Understanding Forgetting in Continual Learning with Linear Regression. | Meng Ding, Kaiyi Ji, Di Wang, Jinhui Xu |
| 2023 | ICML | Achieving Linear Speedup in Non-IID Federated Bilevel Learning. | Minhui Huang, Dewei Zhang, Kaiyi Ji |
| 2023 | ICML | Communication-Efficient Federated Hypergradient Computation via Aggregated Iterative Differentiation. | Peiyao Xiao, Kaiyi Ji |
| 2023 | MOBIHOC | Network Utility Maximization with Unknown Utility Functions: A Distributed, Data-Driven Bilevel Optimization Approach. | Kaiyi Ji, Lei Ying |
| 2022 | UAI | Data sampling affects the complexity of online SGD over dependent data. | Shaocong Ma, Ziyi Chen, Yi Zhou, Kaiyi Ji, Yingbin Liang |
| 2021 | ICML | Bilevel Optimization: Convergence Analysis and Enhanced Design. | Kaiyi Ji, Junjie Yang, Yingbin Liang |
| 2020 | AAAI | Robust Stochastic Bandit Algorithms under Probabilistic Unbounded Adversarial Attack. | Ziwei Guan, Kaiyi Ji, Donald J. Bucci Jr., Timothy Y. Hu, Joseph Palombo, Michael Liston, Yingbin Liang |
| 2020 | ICML | History-Gradient Aided Batch Size Adaptation for Variance Reduced Algorithms. | Kaiyi Ji, Zhe Wang, Bowen Weng, Yi Zhou, Wei Zhang, Yingbin Liang |
| 2020 | IJCAI | Proximal Gradient Algorithm with Momentum and Flexible Parameter Restart for Nonconvex Optimization. | Yi Zhou, Zhe Wang, Kaiyi Ji, Yingbin Liang, Vahid Tarokh |
| 2019 | ICML | Improved Zeroth-Order Variance Reduced Algorithms and Analysis for Nonconvex Optimization. | Kaiyi Ji, Zhe Wang, Yi Zhou, Yingbin Liang |
| 2018 | INFOCOM | Asymptotic Miss Ratio of LRU Caching with Consistent Hashing. | Kaiyi Ji, Guocong Quan, Jian Tan |
| 2018 | INFOCOM | LRU Caching with Dependent Competing Requests. | Guocong Quan, Kaiyi Ji, Jian Tan |
| 2018 | SIGMETRICS | On Resource Pooling and Separation for LRU Caching. | Jian Tan, Guocong Quan, Kaiyi Ji, Ness B. Shroff |