Federated PAC-Bayesian Learning on Non-IID Data

Z Zhao, Y Liu, W Ding, XP Zhang - ICASSP 2024-2024 IEEE …, 2024 - ieeexplore.ieee.org
ICASSP 2024-2024 IEEE International Conference on Acoustics …, 2024ieeexplore.ieee.org
Existing research has either adapted the Probably Approximately Correct (PAC) Bayesian
framework for federated learning (FL) or used information-theoretic PAC-Bayesian bounds
while introducing their theorems, but few consider the non-IID challenges in FL. Our work
presents the first non-vacuous federated PAC-Bayesian bound tailored for non-IID local
data. This bound assumes unique prior knowledge for each client and variable aggregation
weights. We also introduce an objective function and an innovative Gibbs-based algorithm …
Existing research has either adapted the Probably Approximately Correct (PAC) Bayesian framework for federated learning (FL) or used information-theoretic PAC-Bayesian bounds while introducing their theorems, but few consider the non-IID challenges in FL. Our work presents the first non-vacuous federated PAC-Bayesian bound tailored for non-IID local data. This bound assumes unique prior knowledge for each client and variable aggregation weights. We also introduce an objective function and an innovative Gibbs-based algorithm for the optimization of the derived bound. The results are validated on real-world datasets.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果