arXiv:2511.18847cs.CVcs.AI2025-11

跨客户端共享特征+边界聚焦,提升医学图像肿瘤分割精度

Personalized Federated Segmentation with Shared Feature Aggregation and Boundary-Focused Calibration

  • 用解耦交叉注意力捕捉不同器官间的长程特征依赖
  • 引入扰动边界损失,使分割边界更精确,一致性更强
  • 适合医疗影像领域需保护隐私的个性化分割场景

个性化联邦学习(PFL)在保护客户端数据隐私的同时,有效应对非独立同分布(Non-IID)数据带来的异质性问题,已在医学图像分割等领域广泛应用。然而现有方法普遍忽视了跨客户端共享特征的潜力,尤其在包含不同器官分割数据的场景中。本文提出一种面向器官无关肿瘤分割的新型个性化联邦方法FedOAP,利用交叉注意力建模不同客户端间共享特征的长程依赖,并设计边界感知损失提升分割一致性。FedOAP采用解耦交叉注意力(DCA),使每个客户端保留本地查询,同时关注全局聚合的键值对,从而捕获跨器官的长程特征关系。此外,引入扰动边界损失(PBL),聚焦于各客户端预测掩码边界的不一致问题,强制模型更精准地定位边缘。在多种器官的肿瘤分割任务上进行评估,实验表明FedOAP持续优于现有的先进联邦与个性化分割方法。

原文摘要 · Abstract (English)

Personalized federated learning (PFL) possesses the unique capability of preserving data confidentiality among clients while tackling the data heterogeneity problem of non-independent and identically distributed (Non-IID) data. Its advantages have led to widespread adoption in domains such as medical image segmentation. However, the existing approaches mostly overlook the potential benefits of leveraging shared features across clients, where each client contains segmentation data of different organs. In this work, we introduce a novel personalized federated approach for organ agnostic tumor segmentation (FedOAP), that utilizes cross-attention to model long-range dependencies among the shared features of different clients and a boundary-aware loss to improve segmentation consistency. FedOAP employs a decoupled cross-attention (DCA), which enables each client to retain local queries while attending to globally shared key-value pairs aggregated from all clients, thereby capturing long-range inter-organ feature dependencies. Additionally, we introduce perturbed boundary loss (PBL) which focuses on the inconsistencies of the predicted mask's boundary for each client, forcing the model to localize the margins more precisely. We evaluate FedOAP on diverse tumor segmentation tasks spanning different organs. Extensive experiments demonstrate that FedOAP consistently outperforms existing state-of-the-art federated and personalized segmentation methods.

联邦学习医学图像分割隐私保护

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