publications

2024

1. 

   Understanding convergence and generalization in federated learning through feature learning theory

   Wei Huang, Ye Shi, Zhongyi Cai, and 1 more author

   In The Twelfth International Conference on Learning Representations, 2024

   Bib PDF

   @inproceedings{huang2023understanding,
     title = {Understanding convergence and generalization in federated learning through feature learning theory},
     author = {Huang, Wei and Shi, Ye and Cai, Zhongyi and Suzuki, Taiji},
     booktitle = {The Twelfth International Conference on Learning Representations},
     year = {2024},
   }

2023

1. 

   Fed-CO2: Cooperation of Online and Offline Models for Severe Data Heterogeneity in Federated Learning

   Zhongyi Cai, Ye Shi, Wei Huang, and 1 more author

   Advances in Neural Information Processing Systems, 2023

   Bib PDF Code Poster Slides

   @article{cai2024fed,
     title = {Fed-CO2: Cooperation of Online and Offline Models for Severe Data Heterogeneity in Federated Learning},
     author = {Cai, Zhongyi and Shi, Ye and Huang, Wei and Wang, Jingya},
     journal = {Advances in Neural Information Processing Systems},
     volume = {36},
     year = {2023},
   }

2. 

   Fedtp: Federated learning by transformer personalization

   Hongxia Li#, Zhongyi Cai#, Jingya Wang, and 4 more authors

   IEEE transactions on neural networks and learning systems, 2023

   Bib PDF Code

   @article{li2023fedtp,
     title = {Fedtp: Federated learning by transformer personalization},
     author = {Li#, Hongxia and Cai#, Zhongyi and Wang, Jingya and Tang, Jiangnan and Ding, Weiping and Lin, Chin-Teng and Shi, Ye},
     journal = {IEEE transactions on neural networks and learning systems},
     year = {2023},
     publisher = {IEEE},
   }