arXiv:2506.17466cs.LGcs.AI2025-06

联邦学习中用可解释模型提升隐私保护下的分析能力

FedNAMs: Performing Interpretability Analysis in Federated Learning Context

  • 将神经加性模型融入联邦学习,分头处理特征以增强可解释性
  • 在葡萄酒、心脏病等数据上保持高准确率且识别出关键预测特征
  • 适合金融医疗等需隐私与透明度的场景,便于理解模型决策依据

联邦学习持续发展,但仍面临可解释性挑战。本文提出联邦神经加性模型(FedNAMs),将神经加性模型(NAMs)与联邦学习结合,使各客户端专注特定输入特征,实现本地数据训练,保障隐私并提升模型鲁棒性与泛化能力。实验在OpenFetch ML Wine、UCI Heart Disease和Iris数据集上验证,结果显示FedNAMs在文本与图像分类任务中保持高准确率,同时显著提升可解释性。识别出关键预测特征:葡萄酒质量中挥发性酸度、硫酸盐和氯化物;心脏病中胸痛类型、最大心率和血管数量;鸢尾花分类中花瓣长度和宽度。该方法兼顾隐私保护、效率提升与可解释性,在多类数据上表现稳健,并能揭示高/低可解释特征的原因。

原文摘要 · Abstract (English)

Federated learning continues to evolve but faces challenges in interpretability and explainability. To address these challenges, we introduce a novel approach that employs Neural Additive Models (NAMs) within a federated learning framework. This new Federated Neural Additive Models (FedNAMs) approach merges the advantages of NAMs, where individual networks concentrate on specific input features, with the decentralized approach of federated learning, ultimately producing interpretable analysis results. This integration enhances privacy by training on local data across multiple devices, thereby minimizing the risks associated with data centralization and improving model robustness and generalizability. FedNAMs maintain detailed, feature-specific learning, making them especially valuable in sectors such as finance and healthcare. They facilitate the training of client-specific models to integrate local updates, preserve privacy, and mitigate concerns related to centralization. Our studies on various text and image classification tasks, using datasets such as OpenFetch ML Wine, UCI Heart Disease, and Iris, show that FedNAMs deliver strong interpretability with minimal accuracy loss compared to traditional Federated Deep Neural Networks (DNNs). The research involves notable findings, including the identification of critical predictive features at both client and global levels. Volatile acidity, sulfates, and chlorides for wine quality. Chest pain type, maximum heart rate, and number of vessels for heart disease. Petal length and width for iris classification. This approach strengthens privacy and model efficiency and improves interpretability and robustness across diverse datasets. Finally, FedNAMs generate insights on causes of highly and low interpretable features.

联邦学习可解释性神经加性模型隐私保护

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