分享自:

从时空角度防御联邦学习中的投毒攻击

期刊:2023 42nd International Symposium on Reliable Distributed Systems (SRDS)DOI:10.1109/SRDS60354.2023.00013

这篇文档属于类型a,即报告了一项原创性研究。以下是针对该研究的学术报告:

主要作者及机构

该研究的主要作者包括Zhipin Gu、Jiangyong Shi、Yuexiang Yang,他们来自国防科技大学计算机科学与技术学院,另一位作者Liangzhong He来自中国移动(苏州)软件技术有限公司。该研究发表在2023年的42nd International Symposium on Reliable Distributed Systems (SRDS)上。

学术背景

该研究的科学领域是联邦学习(Federated Learning, FL),旨在解决联邦学习中的投毒攻击(Poisoning Attacks)问题。联邦学习是一种分布式神经网络训练方法,允许多个远程客户端在不共享私有数据的情况下协作训练共享模型。然而,由于其分布式特性,联邦学习容易受到投毒攻击的威胁,尤其是拜占庭攻击(Byzantine Attack)和后门攻击(Backdoor Attack)。现有的防御方法主要从空间角度出发,难以有效应对这些攻击。因此,本研究提出了FedMVAE框架,从时空角度出发,旨在减轻投毒攻击的影响,并提高联邦学习的鲁棒性。

研究流程

该研究分为多个步骤,详细流程如下:

  1. 问题定义与框架设计
    研究首先定义了联邦学习中的投毒攻击问题,特别是拜占庭攻击和后门攻击。拜占庭攻击通过上传任意本地模型更新来干扰训练过程,而后门攻击则通过操纵本地数据集或模型更新在全局模型中留下后门。现有的防御方法主要基于异常检测、鲁棒聚合和扰动机制,但这些方法在非独立同分布(non-IID)数据环境下表现不佳。因此,研究提出了FedMVAE框架,结合了多个变分自编码器(Variational Autoencoder, VAE)模型和基于动量的更新过滤器,从时空角度进行防御。

  2. FedMVAE框架的构建
    FedMVAE框架包括三个主要模块:代理生成模块、异常检测模块和基于动量的更新过滤器。

    • 代理生成模块:该模块通过降维方法处理本地模型更新的每一层梯度,生成低维代理。降维过程包括将层参数分割为低维向量、应用最小-最大缩放和主成分分析(PCA),以进一步减少冗余特征。
    • 异常检测模块:该模块使用多个VAE模型来学习不同层的模式,并根据重构误差对本地模型更新进行聚类。每个VAE模型训练在本地模型更新的不同层上,生成的重构误差用于构建特征,并通过简单的软投票方法将本地模型更新分为两类。最后,通过不确定性评估进一步移除可疑的模型更新。
    • 基于动量的更新过滤器:该模块根据历史全局更新生成推测值,并通过比较推测值与历史更新来调整学习率,从而生成鲁棒的全局模型更新。
  3. 实验设计与数据集
    研究在MNIST、MNIST-D和FEMNIST三个基准数据集上进行了广泛的实验,评估了FedMVAE在拜占庭攻击和后门攻击下的鲁棒性。实验中,恶意客户端的比例分别设置为20%、30%和40%,以测试FedMVAE在不同攻击强度下的表现。实验还比较了FedMVAE与现有聚合方法(如FedAvg、Krum、Spectral等)的性能。

  4. 实验结果与分析

    • 拜占庭攻击下的表现:FedMVAE在对抗拜占庭攻击时表现出色,特别是在加性噪声攻击、符号翻转攻击和高斯攻击中,FedMVAE的主任务准确率(MTA)均高于现有方法。实验结果表明,FedMVAE能够有效检测并排除恶意模型更新,从而保证全局模型的鲁棒性。
    • 后门攻击下的表现:FedMVAE在后门攻击下也表现出色,特别是在周期性攻击、PGD攻击和DBA攻击中,FedMVAE的攻击成功率(ASR)显著低于现有方法。这表明FedMVAE能够有效防御后门攻击,同时保持较高的主任务准确率。
    • 恶意客户端比例的影响:实验还评估了不同比例恶意客户端对FedMVAE性能的影响。结果表明,FedMVAE在恶意客户端比例不超过30%时表现稳定,但随着恶意客户端比例的增加,其主任务准确率逐渐下降。

主要结果

研究的主要结果包括: 1. FedMVAE在拜占庭攻击下的鲁棒性:在加性噪声攻击、符号翻转攻击和高斯攻击中,FedMVAE的主任务准确率分别为93.7%、93.4%和92.9%,均高于现有方法。 2. FedMVAE在后门攻击下的鲁棒性:在周期性攻击、PGD攻击和DBA攻击中,FedMVAE的攻击成功率分别为0.010、0.074和0.008,显著低于现有方法。 3. 恶意客户端比例的影响:当恶意客户端比例为30%时,FedMVAE的主任务准确率保持在较高水平,但随着恶意客户端比例的增加,其性能逐渐下降。

结论

该研究提出了FedMVAE框架,通过结合多个VAE模型和基于动量的更新过滤器,从时空角度有效防御了联邦学习中的投毒攻击。实验结果表明,FedMVAE在拜占庭攻击和后门攻击下均表现出色,特别是在非IID数据环境下,其鲁棒性显著优于现有方法。FedMVAE的提出为联邦学习的实际部署提供了重要的理论和技术支持,具有广泛的应用前景。

研究亮点

  1. 创新性方法:FedMVAE首次从时空角度出发,结合了多个VAE模型和基于动量的更新过滤器,提出了全新的防御框架。
  2. 广泛的应用场景:FedMVAE在IID和非IID数据环境下均表现出色,能够应对多种投毒攻击,具有广泛的应用价值。
  3. 实验验证:研究在多个基准数据集上进行了广泛的实验,验证了FedMVAE的有效性和鲁棒性。

其他有价值的内容

研究还探讨了恶意客户端比例对FedMVAE性能的影响,并提出了进一步优化FedMVAE框架的可能性,例如通过引入更多的历史信息或改进VAE模型的结构来提高防御效果。

上述解读依据用户上传的学术文献,如有不准确或可能侵权之处请联系本站站长:admin@fmread.com