Cheng, Yifei

4 publications

MLJ 2024 Communication-Efficient Clustered Federated Learning via Model Distance Mao Zhang, Tie Zhang, Yifei Cheng, Changcun Bao, Haoyu Cao, Deqiang Jiang, Linli Xu
ICLR 2023 DropIT: Dropping Intermediate Tensors for Memory-Efficient DNN Training Joya Chen, Kai Xu, Yuhui Wang, Yifei Cheng, Angela Yao
AAAI 2021 STL-SGD: Speeding up Local SGD with Stagewise Communication Period Shuheng Shen, Yifei Cheng, Jingchang Liu, Linli Xu
IJCAI 2019 Faster Distributed Deep Net Training: Computation and Communication Decoupled Stochastic Gradient Descent Shuheng Shen, Linli Xu, Jingchang Liu, Xianfeng Liang, Yifei Cheng