arXiv:2501.15038cs.LGcs.AI2025-01被引 17

动态调整参与联邦学习的客户端数量,提升网络异常检测精度与效率。

Adaptive Client Selection in Federated Learning: A Network Anomaly Detection Use Case

  • 根据模型表现和系统条件动态选择客户端,结合差分隐私保护数据
  • 相较FedL2P提升7%准确率,训练时间减少25%
  • 兼顾隐私与容错性,适合对安全与稳定性要求高的场景

联邦学习(FL)已成为在分布式数据上训练机器学习模型的主流方法,有效缓解了传统集中式方法带来的隐私风险。然而,其效率依赖于有效的客户端选择与强健的隐私保护机制。不当的客户端选择会导致模型性能下降,而隐私措施不足则可能泄露敏感信息。本文提出一种融合差分隐私与容错机制的自适应客户端选择框架,根据模型表现与系统约束动态调整参与客户端数量,并通过添加校准噪声保障隐私。该方法在使用UNSW-NB15和ROAD数据集的网络异常检测任务中进行评估,结果表明,相比FedL2P方法,准确率最高提升7%,训练时间减少25%。同时,研究揭示了隐私预算与模型性能之间的权衡:更高的隐私预算可减少噪声并提升准确率。尽管容错机制导致轻微性能下降,但显著增强了对客户端故障的鲁棒性。采用曼-惠特尼U检验进行统计验证,结果显著性水平低于0.05。

原文摘要 · Abstract (English)

Federated Learning (FL) has become a widely used approach for training machine learning models on decentralized data, addressing the significant privacy concerns associated with traditional centralized methods. However, the efficiency of FL relies on effective client selection and robust privacy preservation mechanisms. Ineffective client selection can result in suboptimal model performance, while inadequate privacy measures risk exposing sensitive data. This paper introduces a client selection framework for FL that incorporates differential privacy and fault tolerance. The proposed adaptive approach dynamically adjusts the number of selected clients based on model performance and system constraints, ensuring privacy through the addition of calibrated noise. The method is evaluated on a network anomaly detection use case using the UNSW-NB15 and ROAD datasets. Results demonstrate up to a 7% improvement in accuracy and a 25% reduction in training time compared to the FedL2P approach. Additionally, the study highlights trade-offs between privacy budgets and model performance, with higher privacy budgets leading to reduced noise and improved accuracy. While the fault tolerance mechanism introduces a slight performance decrease, it enhances robustness against client failures. Statistical validation using the Mann-Whitney U test confirms the significance of these improvements, with results achieving a p-value of less than 0.05.

联邦学习隐私保护异常检测客户端选择

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。