Yi, Kai

17 publications

ICML 2025 All-Atom Inverse Protein Folding Through Discrete Flow Matching Kai Yi, Kiarash Jamali, Sjors Hw Scheres

TMLR 2025 Explicit Personalization and Local Training: Double Communication Acceleration in Federated Learning Kai Yi, Laurent Condat, Peter Richtárik

TMLR 2025 FedComLoc: Communication-Efficient Distributed Training of Sparse and Quantized Models Kai Yi, Georg Meinhardt, Laurent Condat, Peter Richtárik

NeurIPS 2025 How Particle System Theory Enhances Hypergraph Message Passing Yixuan Ma, Kai Yi, Pietro Lio, Shi Jin, Yu Guang Wang

CVPR 2025 MANTA: A Large-Scale Multi-View and Visual-Text Anomaly Detection Dataset for Tiny Objects Lei Fan, Dongdong Fan, Zhiguang Hu, Yiwen Ding, Donglin Di, Kai Yi, Maurice Pagnucco, Yang Song

ICLRW 2025 Symmetric Pruning for Large Language Models Kai Yi, Peter Richtárik

NeurIPSW 2024 Cohort Squeeze: Beyond a Single Communication Round per Cohort in Cross-Device Federated Learning Kai Yi, Timur Kharisov, Igor Sokolov, Peter Richtárik

ICLR 2024 FedP3: Federated Personalized and Privacy-Friendly Network Pruning Under Model Heterogeneity Kai Yi, Nidham Gazagnadou, Peter Richtárik, Lingjuan Lyu

NeurIPS 2024 PV-Tuning: Beyond Straight-Through Estimation for Extreme LLM Compression Vladimir Malinovskii, Denis Mazur, Ivan Ilin, Denis Kuznedelev, Konstantin Burlachenko, Kai Yi, Dan Alistarh, Peter Richtarik

ICLR 2023 ACMP: Allen-Cahn Message Passing with Attractive and Repulsive Forces for Graph Neural Networks Yuelin Wang, Kai Yi, Xinliang Liu, Yu Guang Wang, Shi Jin

ICCV 2023 Continual Zero-Shot Learning Through Semantically Guided Generative Random Walks Wenxuan Zhang, Paul Janson, Kai Yi, Ivan Skorokhodov, Mohamed Elhoseiny

NeurIPS 2023 Graph Denoising Diffusion for Inverse Protein Folding Kai Yi, Bingxin Zhou, Yiqing Shen, Pietro Lió, Yuguang Wang

NeurIPSW 2022 ACMP: Allen-Cahn Message Passing with Attractive and Repulsive Forces for Graph Neural Networks Yuelin Wang, Kai Yi, Xinliang Liu, Yu Guang Wang, Shi Jin

NeurIPS 2022 EF-BV: A Unified Theory of Error Feedback and Variance Reduction Mechanisms for Biased and Unbiased Compression in Distributed Optimization Laurent Condat, Kai Yi, Peter Richtarik

ECCV 2022 Exploring Hierarchical Graph Representation for Large-Scale Zero-Shot Image Classification Kai Yi, Xiaoqian Shen, Yunhao Gou, Mohamed Elhoseiny

NeurIPS 2022 Variance Reduced ProxSkip: Algorithm, Theory and Application to Federated Learning Grigory Malinovsky, Kai Yi, Peter Richtarik

CVPR 2022 VisualGPT: Data-Efficient Adaptation of Pretrained Language Models for Image Captioning Jun Chen, Han Guo, Kai Yi, Boyang Li, Mohamed Elhoseiny