Variational Perturbation Personalized Federated Learning via Prior-Posterior Distance

Personalized Federated Learning (pFL) mitigates the impact of statistical heterogeneity on FL architecture to some extent by allowing participants to use personalized models based on local data distributions. The existing pFL methods optimize from the perspective of model structure, attempting to adopt strategies that maintain model processing or quickly adapt to local data distribution capabilities. Our proposed method draws inspiration from the concept of variational inference, guiding model updates by comparing prior and posterior data distributions, and innovatively applying model variational perturbations to improve robustness. Finally, we conducted multidimensional experiments and the results show that our method outperforms the current baseline. Code: https://github.com/RezinChow/VPFL.