May, 2023

联邦变分推断:实现更好的个性化和泛化

TL;DR本文提出了基于贝叶斯推断的层次生成模型和 FedVI 算法,该算法通过变分推断训练模型并结合 PAC-Bayes 分析提供了更好的泛化性能。作者还在 FEMNIST 和 CIFAR-100 数据集上进行了实验验证,结果显示 FedVI 在图像分类任务上的表现超过了现有算法。