选择性修剪平均:一种在最优性近似上具有确定性保证的弹性联邦学习算法

Selective Trimmed Average: A Resilient Federated Learning Algorithm With Deterministic Guarantees on the Optimality Approximation

IEEE Transactions on Cybernetics · 2024
被引 15
ABS 3

中文导读

提出一种名为SETA的弹性联邦学习算法,通过过滤和组合参数来抵御恶意代理的攻击,并在服务器-工人和共享内存架构下均有效,理论证明其鲁棒性,在MNIST和天气数据集上验证。

Abstract

The federated learning (FL) paradigm aims to distribute the computational burden of the training process among several computation units, usually called agents or workers, while preserving private local training datasets. This is generally achieved by resorting to a server-worker architecture where agents iteratively update local models and communicate local parameters to a server that aggregates and returns them to the agents. However, the presence of adversarial agents, which may intentionally exchange malicious parameters or may have corrupted local datasets, can jeopardize the FL process. Therefore, we propose selective trimmed average (SETA), which is a resilient algorithm to cope with the undesirable effects of a number of misbehaving agents in the global model. SETA is based on properly filtering and combining the exchanged parameters. We mathematically prove that the proposed algorithm is resilient against data and local model poisoning attacks. Most resilient methods presented so far in the literature assume that a trusted server is in hand. In contrast, our algorithm works both in server-worker and shared memory architectures, where the latter excludes the necessity of a trusted server. The theoretical findings are corroborated through numerical results on MNIST dataset and on multiclass weather dataset (MWD).

联邦学习分布式计算机器学习网络安全