Kai Yi

@misc{yi2026jacquant,
  title = {JacQuant: STE-Free Quantization-Aware Training via Learned Jacobian Surrogates},
  author = {Yi, Kai and Vivekraja, Vignesh and Khaitan, Harshit and Li, Steven},
  year = {2026},
}

@inproceedings{li2026winq,
  title = {WinQ: Accelerating Quantization-Aware Training of Language Models Around Saddle Points},
  author = {Li, Dongyue and Liu, Zechun and Yi, Kai and Zhang, Zhenshuo and Zhao, Changsheng and Krishnamoorthi, Raghuraman and Khaitan, Harshit and Zhang, Hongyang R. and Li, Steven},
  booktitle = {International Conference on Machine Learning (ICML)},
  year = {2026},
}

@article{yi2025fedcomloc,
  title = {FedComLoc: Communication-Efficient Distributed Training of Sparse and Quantized Models},
  author = {Yi, Kai and Meinhardt, Georg and Condat, Laurent and Richtárik, Peter},
  journal = {Transactions on Machine Learning Research (TMLR)},
  year = {2025},
}

@inproceedings{yi2025symwanda,
  title = {Symmetric Pruning for Large Language Models},
  author = {Yi, Kai and Richtárik, Peter},
  booktitle = {ICLR Workshop on Sparsity in LLMs (SLLM)},
  year = {2025},
}

@inproceedings{malinovskii2024pvtuning,
  title = {PV-Tuning: Beyond Straight-Through Estimation for Extreme LLM Compression},
  author = {Malinovskii, Vladimir and Mazur, Denis and Ilin, Ivan and Kuznedelev, Denis and Burlachenko, Konstantin and Yi, Kai and Alistarh, Dan and Richtárik, Peter},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year = {2024},
  note = {Oral},
}

@inproceedings{yi2024fedp3,
  title = {FedP3: Federated Personalized and Privacy-friendly Network Pruning under Model Heterogeneity},
  author = {Yi, Kai and Gazagnadou, Nidham and Richtárik, Peter and Lyu, Lingjuan},
  booktitle = {International Conference on Learning Representations (ICLR)},
  year = {2024},
}

@inproceedings{yi2022hgrnet,
  title = {Exploring Hierarchical Graph Representation for Large-Scale Zero-Shot Image Classification},
  author = {Yi, Kai and Shen, Xiaoqian and Gou, Yunhao and Elhoseiny, Mohamed},
  booktitle = {European Conference on Computer Vision (ECCV)},
  year = {2022},
}

@inproceedings{condat2022efbv,
  title = {EF-BV: A Unified Theory of Error Feedback and Variance Reduction Mechanisms for Biased and Unbiased Compression in Distributed Optimization},
  author = {Condat, Laurent and Yi, Kai and Richtárik, Peter},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year = {2022},
}

@inproceedings{chen2022visualgpt,
  title = {VisualGPT: Data-efficient Image Captioning by Balancing Visual Input and Linguistic Knowledge from Pretraining},
  author = {Chen, Jun and Guo, Han and Yi, Kai and Li, Boyang and Elhoseiny, Mohamed},
  booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  year = {2022},
}