解决医疗影像联邦学习中模态缺失与个性化需求难题
Federated Modality-specific Encoders and Partially Personalized Fusion Decoder for Multimodal Brain Tumor Segmentation
- 为每种影像模态设计专用编码器,应对跨机构模态差异
- 融合解码器部分个性化,通过参数差异动态决定定制程度
- 用全局锚点引导缺失模态重建,提升不完整数据参与效果
现有联邦学习方法多关注同模态异质性,难以应对多模态医疗影像中的跨模态异质性问题。实际中部分参与者仅拥有部分影像模态,导致全局模型训练困难。同时,各参与方期望模型适配本地数据特征。本文提出FedMEPD框架,包含联邦式模态专用编码器和部分个性化融合解码器。每个模态使用专属编码器处理跨模态异质性;解码器则基于全局与本地参数更新的差异,动态决定个性化程度。服务器利用全模态数据训练融合解码器,通过反向传播优化编码器,并将多组融合表示锚点分发至客户端。缺少模态的客户端通过缩放点积交叉注意力对缺失模态表示进行校准,弥补信息缺失。在BraTS 2018和2020多模态脑肿瘤分割基准上验证,该方法优于多种先进多模态个性化联邦学习方法,其创新设计有效。
原文摘要 · Abstract (English)
Most existing federated learning (FL) methods for medical image analysis only considered intramodal heterogeneity, limiting their applicability to multimodal imaging applications. In practice, some FL participants may possess only a subset of the complete imaging modalities, posing intermodal heterogeneity as a challenge to effectively training a global model on all participants' data. Meanwhile, each participant expects a personalized model tailored to its local data characteristics in FL. This work proposes a new FL framework with federated modality-specific encoders and partially personalized multimodal fusion decoders (FedMEPD) to address the two concurrent issues. Specifically, FedMEPD employs an exclusive encoder for each modality to account for the intermodal heterogeneity. While these encoders are fully federated, the decoders are partially personalized to meet individual needs -- using the discrepancy between global and local parameter updates to dynamically determine which decoder filters are personalized. Implementation-wise, a server with full-modal data employs a fusion decoder to fuse representations from all modality-specific encoders, thus bridging the modalities to optimize the encoders via backpropagation. Moreover, multiple anchors are extracted from the fused multimodal representations and distributed to the clients in addition to the model parameters. Conversely, the clients with incomplete modalities calibrate their missing-modal representations toward the global full-modal anchors via scaled dot-product cross-attention, making up for the information loss due to absent modalities. FedMEPD is validated on the BraTS 2018 and 2020 multimodal brain tumor segmentation benchmarks. Results show that it outperforms various up-to-date methods for multimodal and personalized FL, and its novel designs are effective.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。