arXiv:2507.19045cs.CVcs.DC2025-07中稿 · ECAI 2025被引 1

提出新框架,用特征生成和知识蒸馏实现医疗影像单轮联邦学习。

A New One-Shot Federated Learning Framework for Medical Imaging Classification with Feature-Guided Rectified Flow and Knowledge Distillation

  • 客户端用特征级生成模型加速建模并保护隐私
  • 单轮聚合中提升准确率最高达21.73%优于多轮方法
  • 适合数据异构的医疗多中心场景,隐私更安全

在多中心医疗场景中,单轮联邦学习(OSFL)因通信开销低而备受关注,仅需一轮传输。然而,现有基于生成模型的OSFL方法存在训练效率低、潜在隐私泄露问题,且在非独立同分布(non-IID)数据下难以收敛。为此,本文提出一种改进的OSFL框架,引入特征引导的修正流模型(FG-RF)与双层知识蒸馏(DLKD)聚合机制。客户端使用FG-RF生成特征级图像,既加速医学影像建模又保护隐私;为应对non-IID数据,DLKD在聚合时使全局学生模型同时模仿各客户端教师模型的输出逻辑和中间层特征。在三个non-IID医学影像数据集上的实验表明,本框架性能超越多轮联邦学习方法,最高提升21.73%,平均优于基线方法FedISCA达21.75%。此外,特征级合成图像显著降低隐私泄露风险。代码已开源:https://github.com/LMIAPC/one-shot-fl-medical。

原文摘要 · Abstract (English)

In multi-center scenarios, One-Shot Federated Learning (OSFL) has attracted increasing attention due to its low communication overhead, requiring only a single round of transmission. However, existing generative model-based OSFL methods suffer from low training efficiency and potential privacy leakage in the healthcare domain. Additionally, achieving convergence within a single round of model aggregation is challenging under non-Independent and Identically Distributed (non-IID) data. To address these challenges, in this paper a modified OSFL framework is proposed, in which a new Feature-Guided Rectified Flow Model (FG-RF) and Dual-Layer Knowledge Distillation (DLKD) aggregation method are developed. FG-RF on the client side accelerates generative modeling in medical imaging scenarios while preserving privacy by synthesizing feature-level images rather than pixel-level images. To handle non-IID distributions, DLKD enables the global student model to simultaneously mimic the output logits and align the intermediate-layer features of client-side teacher models during aggregation. Experimental results on three non-IID medical imaging datasets show that our new framework and method outperform multi-round federated learning approaches, achieving up to 21.73% improvement, and exceeds the baseline FedISCA by an average of 21.75%. Furthermore, our experiments demonstrate that feature-level synthetic images significantly reduce privacy leakage risks compared to pixel-level synthetic images. The code is available at https://github.com/LMIAPC/one-shot-fl-medical.

联邦学习医疗影像单轮学习知识蒸馏

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