A specificity-preserving generative model for federated MRI translation

Date

2022-10-07

Authors

Dalmaz, Onat
Mirza, Usama
Elmas, Gökberk
Özbey, Muzaffer
Dar, Salman U. H
Çukur, Tolga

Advisor

Supervisor

Co-Advisor

Co-Supervisor

Instructor

Source Title

Distributed, Collaborative, and Federated Learning, and Affordable AI and Healthcare for Resource Diverse Global Health

Print ISSN

Electronic ISSN

Publisher

Springer Cham

Volume

13573

Issue

Pages

79 - 88

Language

English

Type

Conference Paper

Journal Title

Journal ISSN

Volume Title

Series

Lecture Notes in Computer Science;

Abstract

MRI translation models learn a mapping from an acquired source contrast to an unavailable target contrast. Collaboration between institutes is essential to train translation models that can generalize across diverse datasets. That said, aggregating all imaging data and training a centralized model poses privacy problems. Recently, federated learning (FL) has emerged as a collaboration framework that enables decentralized training to avoid sharing of imaging data. However, FL-trained translation models can deteriorate by the inherent heterogeneity in the distribution of MRI data. To improve reliability against domain shifts, here we introduce a novel specificity-preserving FL method for MRI contrast translation. The proposed approach is based on an adversarial model that adaptively normalizes the feature maps across the generator based on site-specific latent variables. Comprehensive FL experiments were conducted on multi-site datasets to show the effectiveness of the proposed approach against prior federated methods in MRI contrast translation.

Course

Other identifiers

Book Title

Keywords

Federated learning, Heterogeneity, MRI, Site-specificity, Translation

Citation