arXiv:2410.12316cs.LGcs.DC2024-10被引 1

用主观逻辑提升联邦学习的可信度,兼顾隐私与可靠性。

TPFL: A Trustworthy Personalized Federated Learning Framework via Subjective Logic

  • 引入主观逻辑构建概率决策并评估不确定性
  • 通过可训练异质性先验缓解非独立同分布数据影响
  • 在攻击、数据漂移下仍保持稳定,适合高风险场景

联邦学习(FL)可在保护数据隐私的前提下实现分布式客户端协同训练。尽管广泛应用,现有方法多仅关注隐私保护,在需要可信性的场景中仍显不足,缺乏对安全训练、可靠决策、抗干扰能力及非独立同分布(Non-IID)数据性能的综合保障。为此,本文提出可信个性化联邦学习(TPFL)框架,面向分类任务,基于主观逻辑构建模型。该框架采用主观逻辑生成带有不确定性评估的概率输出,而非简单概率分配;通过在本地训练前引入可训练的异质性先验,有效缓解数据异构性带来的负面影响;进一步利用模型不确定性和实例不确定性,确保训练与推理阶段的安全可靠。在多个主流联邦学习基准上的大量实验表明,TPFL不仅性能媲美先进方法,还在面对常见恶意攻击、领域偏移时展现出强鲁棒性,并在高风险场景中表现高度可信。

原文摘要 · Abstract (English)

Federated learning (FL) enables collaborative model training across distributed clients while preserving data privacy. Despite its widespread adoption, most FL approaches focusing solely on privacy protection fall short in scenarios where trustworthiness is crucial, necessitating advancements in secure training, dependable decision-making mechanisms, robustness on corruptions, and enhanced performance with Non-IID data. To bridge this gap, we introduce Trustworthy Personalized Federated Learning (TPFL) framework designed for classification tasks via subjective logic in this paper. Specifically, TPFL adopts a unique approach by employing subjective logic to construct federated models, providing probabilistic decisions coupled with an assessment of uncertainty rather than mere probability assignments. By incorporating a trainable heterogeneity prior to the local training phase, TPFL effectively mitigates the adverse effects of data heterogeneity. Model uncertainty and instance uncertainty are further utilized to ensure the safety and reliability of the training and inference stages. Through extensive experiments on widely recognized federated learning benchmarks, we demonstrate that TPFL not only achieves competitive performance compared with advanced methods but also exhibits resilience against prevalent malicious attacks, robustness on domain shifts, and reliability in high-stake scenarios.

联邦学习可信计算不确定性建模

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