arXiv:2607.20641cs.LG2026-07

提出只需1-2轮通信的联邦学习方法,解决医疗影像多标签任务中机构标注不全的问题。

One Round Is All You Need: Analytic Federated Learning for Task-Heterogeneous Multi-Label Medical Image Classification

论文配图:One Round Is All You Need: Analytic Federated Learning for Task-Heterogeneous Multi-Label Medical Image Classification
图 1 · 摘自论文原文
  • 用闭式公式替代迭代优化,避免梯度训练的收敛慢问题。
  • 在四个缺失标签配置下,准确率最高提升18.44点,通信量显著减少。
  • 适合医疗数据分散、标签不完整场景,尤其适用于多中心协作研究。

联邦学习(FL)使多个医疗机构可在不集中患者数据的前提下协同训练疾病分类器。但实践中,各机构仅标注其专业领域内的病灶,导致任务异构:每个客户端仅掌握目标疾病类别的严格子集,其余类别完全未被观测。现有基于梯度的联邦学习方法在此场景下失效,因其需数百轮通信才能收敛,且缺失标签引入系统性假阴性偏差,模型无法纠正。本文提出一种面向任务异构的多标签医疗图像分类分析型联邦学习框架。该方法将迭代梯度优化替换为三个闭式运算:平衡标签投影,通过归一化正负样本贡献以消除类别不平衡偏倚;每类绝对聚合律,从标注客户端上传的充分统计量中独立构建最优岭回归分类器;可选的分析伪标签精炼轮,将置信度过滤后的教师分类器知识传播至未标注客户端。整个流程最多只需两轮通信,与任务异构程度和参与客户端数量无关。在ChestXray14数据集上,四种逐步加剧的缺失类配置下,本方法相较最先进的联邦多标签方法FedMLP,最高提升18.44 BACC点和13.24 AUC点,同时大幅降低通信开销。

原文摘要 · Abstract (English)

Federated learning (FL) enables multiple clinical institutions to collaboratively train a shared disease classifier without centralizing patient data. In practice, however, each institution annotates only the pathologies within its area of expertise, so the federation operates under task heterogeneity: each client holds labels for a strict subset of the target disease categories while the remaining classes are entirely unobserved at that site. Existing gradient-based FL methods fail under this setting because they require hundreds of communication rounds to converge and because missing class labels introduce systematic false-negative bias that the model cannot correct without a principled mechanism. We propose an analytic federated learning framework for multi-label medical image classification under task heterogeneity. The proposed method replaces iterative gradient optimization with three closed-form operations: a balanced label projection that neutralizes class-imbalance bias by normalizing positive and negative contributions to equal total mass; a per-class absolute aggregation law that independently assembles the optimal ridge-regression classifier for each disease category from the sufficient statistics uploaded by its annotating clients; and an optional analytic pseudo-label refinement round that propagates missing-class knowledge from a confidence-filtered teacher classifier to non-annotating clients. The entire procedure requires at most two communication rounds, irrespective of the degree of task heterogeneity or the number of participating clients. Experiments on ChestXray14 under four progressively severe missing-class configurations demonstrate that the proposed method consistently outperforms the state-of-the-art federated multi-label method FedMLP by up to 18.44 BACC points and 13.24 AUC points, while reducing the communication.

联邦学习多标签分类医疗影像任务异构

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