arXiv:2512.04264cs.LGcs.CV2025-12

对比10种激活函数,研究非独立同分布数据下模型鲁棒性提升方法。

Studying Various Activation Functions and Non-IID Data for Machine Learning Model Robustness

  • 提出融合架构调整与软标签的对抗训练新方法。
  • 中心化环境下鲁棒准确率达67.96%(CIFAR-10,FGSM攻击)。
  • 非独立同分布数据中引入数据共享,显著提升鲁棒性。

对抗训练是提升机器学习模型鲁棒性的有效方法。现有研究多采用ReLU激活函数和集中式训练环境。本文在集中式环境与联邦学习环境中,通过对抗训练研究十种不同激活函数对模型鲁棒性的影响。在集中式环境下,提出一种改进的对抗训练方法,结合模型架构调整、软标签、简化数据增强与可变学习率,实验对比了包括ReLU在内的十种激活函数。结果表明,ReLU通常表现最佳。在联邦学习环境中,尤其在非独立同分布(non-IID)数据下,鲁棒性显著下降。为此引入数据共享机制,在40%数据共享时,自然准确率与鲁棒准确率分别达70.09%与54.79%,优于CalFAT算法。结果表明,适度数据共享可显著提升模型鲁棒性,适用于真实场景。

原文摘要 · Abstract (English)

Adversarial training is an effective method to improve the machine learning (ML) model robustness. Most existing studies typically consider the Rectified linear unit (ReLU) activation function and centralized training environments. In this paper, we study the ML model robustness using ten different activation functions through adversarial training in centralized environments and explore the ML model robustness in federal learning environments. In the centralized environment, we first propose an advanced adversarial training approach to improving the ML model robustness by incorporating model architecture change, soft labeling, simplified data augmentation, and varying learning rates. Then, we conduct extensive experiments on ten well-known activation functions in addition to ReLU to better understand how they impact the ML model robustness. Furthermore, we extend the proposed adversarial training approach to the federal learning environment, where both independent and identically distributed (IID) and non-IID data settings are considered. Our proposed centralized adversarial training approach achieves a natural and robust accuracy of 77.08% and 67.96%, respectively on CIFAR-10 against the fast gradient sign attacks. Experiments on ten activation functions reveal ReLU usually performs best. In the federated learning environment, however, the robust accuracy decreases significantly, especially on non-IID data. To address the significant performance drop in the non-IID data case, we introduce data sharing and achieve the natural and robust accuracy of 70.09% and 54.79%, respectively, surpassing the CalFAT algorithm, when 40% data sharing is used. That is, a proper percentage of data sharing can significantly improve the ML model robustness, which is useful to some real-world applications.

对抗训练联邦学习非IID激活函数

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