Federated disentangled representation learning for unsupervised brain anomaly detection

Bercea C, Wiestler B, Rueckert D, Albarqouni S (2022)

Publication Type: Journal article

Publication year: 2022

Journal

Nature Machine Intelligence Nature Research

Book Volume: 4

Pages Range: 685-695

Journal Issue: 8

DOI: 10.1038/s42256-022-00515-2

Abstract

With the advent of deep learning and increasing use of brain MRIs, a great amount of interest has arisen in automated anomaly segmentation to improve clinical workflows; however, it is time-consuming and expensive to curate medical imaging. Moreover, data are often scattered across many institutions, with privacy regulations hampering its use. Here we present FedDis to collaboratively train an unsupervised deep convolutional autoencoder on 1,532 healthy magnetic resonance scans from four different institutions, and evaluate its performance in identifying pathologies such as multiple sclerosis, vascular lesions, and low- and high-grade tumours/glioblastoma on a total of 538 volumes from six different institutions. To mitigate the statistical heterogeneity among different institutions, we disentangle the parameter space into global (shape) and local (appearance). Four institutes jointly train shape parameters to model healthy brain anatomical structures. Every institute trains appearance parameters locally to allow for client-specific personalization of the global domain-invariant features. We have shown that our collaborative approach, FedDis, improves anomaly segmentation results by 99.74% for multiple sclerosis, 83.33% for vascular lesions and 40.45% for tumours over locally trained models without the need for annotations or sharing of private local data. We found out that FedDis is especially beneficial for institutes that share both healthy and anomaly data, improving their local model performance by up to 227% for multiple sclerosis lesions and 77% for brain tumours.

Involved external institutions

Helmholtz Zentrum München - Deutsches Forschungszentrum für Gesundheit und Umwelt (HMGU) / Helmholtz Munich DE Germany (DE) Technische Universität München (TUM) DE Germany (DE) Klinikum rechts der Isar DE Germany (DE)

How to cite

APA:

Bercea, C., Wiestler, B., Rueckert, D., & Albarqouni, S. (2022). Federated disentangled representation learning for unsupervised brain anomaly detection. Nature Machine Intelligence, 4(8), 685-695. https://doi.org/10.1038/s42256-022-00515-2

MLA:

Bercea, Cosmin, et al. "Federated disentangled representation learning for unsupervised brain anomaly detection." Nature Machine Intelligence 4.8 (2022): 685-695.

BibTeX: Download