1
Return

TADynFed: Dynamic modality-adaptive federated learning with tissue-aware disentanglement for cross-disease analysis

delete2026-05-01
delete0
PRE
AI
S
Saeed Iqbal *
X
Xiaopin Zhong
K
Khan, Muhammad Attique
Z
Zongze Wu
N
Nouf Abdullah Almujally
W
Weixiang Liu
A
Amir Hussain *
DOI:10.1016/j.artmed.2026.103378delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Federated learning (FL) enables collaborative medical image analysis across decentralized institutions while preserving data privacy. However, real-world deployment faces critical challenges: modality heterogeneity, where clients possess incomplete or varying medical image modalities, and cross-disease generalization, requiring models to adapt to unseen pathologies with anatomical consistency. Existing methods like FedAvg, FedProx, IOP-FL, and PntTranForFL assume uniform modality availability and static client participation, leading to poor performance under realistic clinical constraints. We propose TADynFed, a novel framework for the Heterogeneous Federated Learning (HFL) paradigm, which addresses both data and modality heterogeneity through, a tissue-aware disentanglement strategy that decouples modality-tailored and modality-shared features, a dynamic prototype memory bank for missing modality compensation, an adaptive aggregation mechanism that accounts for client reliability and tenure. We evaluate TADynFed using multi-disease MRI datasets including BraTS21 along with cross-domain imaging data from CheXpert (chest X-ray) and Hep-2 (microscopy), simulating a 13-client FL environment. The proposed framework achieves an average mDice score of 66.03%, significantly outperforming baseline methods such as PointTransformerFL at 58.60% and FedAvg at 53.06%. It also attains the lowest boundary alignment error with an ASD of 1.85 mm and HD95 of 8.70 mm, surpassing existing approaches by notable margins. In terms of calibration stability, TADynFed records an ECE of 0.09, indicating superior confidence reliability. Furthermore, it demonstrates high communication efficiency with only 76 MB of data exchanged per round, compared to 95-105 MB in other frameworks. These results validate TADynFed's ability to maintain high segmentation accuracy, boundary precision, and calibration stability while minimizing bandwidth usage. Outperforming existing frameworks by significant margins, TADynFed demonstrates robust boundary alignment, superior calibration, and efficient communication. It also exhibits strong cross-disease transferability without retraining. By integrating structured representation decomposition, prototype-guided fusion, and client-adaptive learning, TADynFed establishes a new benchmark in realistic, heterogeneous, and mix-modal federated medical imaging systems. Code supporting this study TADynFed.
Keywords:
Dynamic modality-adaptive federated learning
Tissue-aware disentanglement
Cross-disease segmentation
Prototype memory bank
Unseen domain generalization
Client-reliable aggregation

Journal

Artificial Intelligence in Medicine cover
Artificial Intelligence in Medicine
IF:
6.2
Papers:
2.5K
Citations:
7.8K

Organization

E
edinburgh napier university
Scholars:
377
Papers: 239
Citations: 0
P
Princess Nourah bint Abdulrahman University
Scholars:
7.4K
Papers: 9.0K
Citations: 10
S
shenzhen university
Scholars:
4.4W
Papers: 3.4W
Citations: 72
Prince Mohammad bin Fahd University cover
Prince Mohammad bin Fahd University
Scholars:
854
Papers: 1.3K
Citations: 1.5K
Cited Papers

Cited Papers

Citing Papers

Citing Papers