用原型交换提升跨域物联网恶意软件检测,保护隐私且应对数据不均衡。
FedP3E: Privacy-Preserving Prototype Exchange for Non-IID IoT Malware Detection in Cross-Silo Federated Learning
- 各设备用高斯混合模型生成类原型,加噪后上传,不传原始数据。
- 通过聚合原型和SMOTE增强,使少数类恶意样本检测率提升23.6%。
- 适合隐私敏感的工业物联网场景,尤其应对罕见恶意程序。
随着物联网在关键领域持续扩展,其成为日益复杂和大规模恶意软件攻击的主要目标。动态威胁环境与物联网数据的敏感性,要求检测框架兼具隐私保护与对数据异质性的鲁棒性。联邦学习(FL)通过在不暴露原始数据的情况下实现分布式模型训练,提供了一种有前景的解决方案。然而,标准的联邦学习算法如FedAvg和FedProx在实际部署中常因类别不平衡和非独立同分布(non-IID)数据而表现不佳,尤其是在存在罕见或不重叠的恶意软件类别时。为此,我们提出FedP3E(隐私保护原型交换),一种新型联邦学习框架,支持在不泄露原始数据的前提下实现跨客户端的间接表示共享。每个客户端使用高斯混合模型(GMM)构建类别原型,添加高斯噪声后仅向服务器发送这些紧凑摘要。聚合后的原型再分发回客户端,并结合SMOTE数据增强技术融入本地训练,以强化少数类恶意软件的表征能力。不同于单纯参数平均,该原型驱动机制使客户端能获取联邦中其他设备观察到的互补结构模式,从而有效缓解统计异质性,且通信开销极低。我们在包含真实跨孤岛场景的N-BaIoT数据集上评估了FedP3E,验证其在不同数据不平衡程度下的有效性。
原文摘要 · Abstract (English)
As IoT ecosystems continue to expand across critical sectors, they have become prominent targets for increasingly sophisticated and large-scale malware attacks. The evolving threat landscape, combined with the sensitive nature of IoT-generated data, demands detection frameworks that are both privacy-preserving and resilient to data heterogeneity. Federated Learning (FL) offers a promising solution by enabling decentralized model training without exposing raw data. However, standard FL algorithms such as FedAvg and FedProx often fall short in real-world deployments characterized by class imbalance and non-IID data distributions -- particularly in the presence of rare or disjoint malware classes. To address these challenges, we propose FedP3E (Privacy-Preserving Prototype Exchange), a novel FL framework that supports indirect cross-client representation sharing while maintaining data privacy. Each client constructs class-wise prototypes using Gaussian Mixture Models (GMMs), perturbs them with Gaussian noise, and transmits only these compact summaries to the server. The aggregated prototypes are then distributed back to clients and integrated into local training, supported by SMOTE-based augmentation to enhance representation of minority malware classes. Rather than relying solely on parameter averaging, our prototype-driven mechanism enables clients to enrich their local models with complementary structural patterns observed across the federation -- without exchanging raw data or gradients. This targeted strategy reduces the adverse impact of statistical heterogeneity with minimal communication overhead. We evaluate FedP3E on the N-BaIoT dataset under realistic cross-silo scenarios with varying degrees of data imbalance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。