BlendFL融合横向与纵向联邦学习,解决多模态数据异构难题。
BlendFL: Blended Federated Learning for Handling Multimodal Data Heterogeneity
- 混合使用水平与垂直联邦学习,按客户端数据灵活适配
- 在真实医疗与多模态数据集上,分类准确率优于现有方法
- 支持去中心化推理,降低延迟,适合医疗金融等隐私敏感场景
真实场景中协同机器学习面临多模态数据异构的核心挑战。尽管联邦学习(FL)可在不共享数据的情况下训练模型,但现有框架如水平和垂直联邦学习仅在理想假设下有效,难以应对客户端既不全具备所有模态也不全覆盖所有样本的情况。为此,我们提出BlendFL,一种新型联邦学习框架,以同步且无约束的方式融合水平与垂直联邦学习的原理,适应客户端间的不对称性。任一客户端可根据自身数据,选择使用其中一种或同时采用两种策略。此外,BlendFL具备去中心化推理机制,使客户端可利用本地数据运行协同训练的本地模型,从而降低延迟并减少对中心服务器的依赖。我们还引入BlendAvg,一种基于客户端性能自适应的全局模型聚合策略。在大规模真实世界多模态医学数据集和主流多模态基准上,对三种分类任务进行训练与评估。结果表明,BlendFL在多模态和单模态分类任务中均表现更优。消融实验显示其收敛速度显著快于传统方法,加速了协同学习进程。研究表明,BlendFL在数据隐私至关重要的现实场景(如医疗、金融)中具有处理多模态数据异构的潜力。
原文摘要 · Abstract (English)
One of the key challenges of collaborative machine learning, without data sharing, is multimodal data heterogeneity in real-world settings. While Federated Learning (FL) enables model training across multiple clients, existing frameworks, such as horizontal and vertical FL, are only effective in `ideal' settings that meet specific assumptions. Hence, they struggle to address scenarios where neither all modalities nor all samples are represented across the participating clients. To address this gap, we propose BlendFL, a novel FL framework that seamlessly blends the principles of horizontal and vertical FL in a synchronized and non-restrictive fashion despite the asymmetry across clients. Specifically, any client within BlendFL can benefit from either of the approaches, or both simultaneously, according to its available dataset. In addition, BlendFL features a decentralized inference mechanism, empowering clients to run collaboratively trained local models using available local data, thereby reducing latency and reliance on central servers for inference. We also introduce BlendAvg, an adaptive global model aggregation strategy that prioritizes collaborative model updates based on each client's performance. We trained and evaluated BlendFL and other state-of-the-art baselines on three classification tasks using a large-scale real-world multimodal medical dataset and a popular multimodal benchmark. Our results highlight BlendFL's superior performance for both multimodal and unimodal classification. Ablation studies demonstrate BlendFL's faster convergence compared to traditional approaches, accelerating collaborative learning. Overall, in our study we highlight the potential of BlendFL for handling multimodal data heterogeneity for collaborative learning in real-world settings where data privacy is crucial, such as in healthcare and finance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。