Sorry, you need to enable JavaScript to visit this website.

FANTOM: Federated Adversarial Network for Training Multi-sequence Magnetic Resonance Imaging in Semantic Segmentation

Citation Author(s):
Apoorva Srivastava, Avik Kar, Dipayan Dewan, Debdoot Sheet
Submitted by:
Anupam Borthakur
Last updated:
19 November 2024 - 7:20am
Document Type:
Presentation Slides
Document Year:
2024
Event:
Presenters:
Anupam Borthakur
Paper Code:
2707
 

Ischemic stroke lesions (ISL) segmentation aids clinicians in the diagnosis of stroke in acute care units. But, a generalized segmentation model requires data from various patients. However considering the data privacy, the patient's data is not available for centralized training. The Federated Learning (FL) framework overcomes this, but in FL, semantic segmentation is challenging due to the complex model, adversarial training, and non-independent and identically distributed dataset. In this work, we address these drawbacks for the segmentation of ISL into core and penumbra using multi-sequence magnetic resonance imaging data. Instead of applying cordinate-wise weight aggregation which is normally followed in vanilla FL aggregation strategies, we have applied the concept of neural matching in the proposed method named FANTOM. It helped in faster convergence of the training when performed with different non-IID data across clients. We have adversarially trained and tested our segmentation model comprises of generator and discriminator for the ISLES-2015 dataset. It is observed that keeping the discriminator model locally and aggregating only the generator model not only performs well but also lowers the communication burden in the FL framework.

up
0 users have voted: