TY - GEN
T1 - Decentralized Gossip Mutual Learning (GML) for brain tumor segmentation on multi-parametric MRI
AU - Chen, Jingyun
AU - Yuan, Yading
N1 - Publisher Copyright:
© 2023 IEEE.
PY - 2023
Y1 - 2023
N2 - Federated Learning (FL) enables collaborative model training among medical centers without sharing private data. However, traditional FL risks on server failures and suboptimal performance on local data due to the nature of centralized model aggregation. To address these issues, we present Gossip Mutual Learning (GML), a decentralized framework that uses Gossip Protocol for direct peer-to-peer communication. In addition, GML encourages each site to optimize its local model through mutual learning to account for data variations among different sites. For the task of tumor segmentation using 146 cases from four clinical sites in BraTS 2021 dataset, we demonstrated GML outperformed local models and achieved similar performance as FedAvg with only 25% communication overhead.Clinical Relevance: Automatic segmentation of brain tumors is an essential but challenging step for extracting quantitative imaging biomarkers for accurate tumor detection, diagnosis, prognosis, treatment planning and assessment. We proposed a novel learning schedule to build an effective AI model by learning from data across different sources without sacrificing patient privacy, which can be generalized to a broader clinical research and applications.
AB - Federated Learning (FL) enables collaborative model training among medical centers without sharing private data. However, traditional FL risks on server failures and suboptimal performance on local data due to the nature of centralized model aggregation. To address these issues, we present Gossip Mutual Learning (GML), a decentralized framework that uses Gossip Protocol for direct peer-to-peer communication. In addition, GML encourages each site to optimize its local model through mutual learning to account for data variations among different sites. For the task of tumor segmentation using 146 cases from four clinical sites in BraTS 2021 dataset, we demonstrated GML outperformed local models and achieved similar performance as FedAvg with only 25% communication overhead.Clinical Relevance: Automatic segmentation of brain tumors is an essential but challenging step for extracting quantitative imaging biomarkers for accurate tumor detection, diagnosis, prognosis, treatment planning and assessment. We proposed a novel learning schedule to build an effective AI model by learning from data across different sources without sacrificing patient privacy, which can be generalized to a broader clinical research and applications.
UR - https://www.scopus.com/pages/publications/85185555487
U2 - 10.1109/IEEECONF58974.2023.10404978
DO - 10.1109/IEEECONF58974.2023.10404978
M3 - Conference contribution
AN - SCOPUS:85185555487
T3 - 2023 IEEE EMBS Special Topic Conference on Data Science and Engineering in Healthcare, Medicine and Biology, IEEECONF 2023
SP - 63
EP - 64
BT - 2023 IEEE EMBS Special Topic Conference on Data Science and Engineering in Healthcare, Medicine and Biology, IEEECONF 2023
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 2023 IEEE EMBS Special Topic Conference on Data Science and Engineering in Healthcare, Medicine and Biology, IEEECONF 2023
Y2 - 7 December 2023 through 9 December 2023
ER -