A comparative analysis of the impact of data distribution on distributed learning with a traveling model for brain age prediction

Raissa Souza; Agampreet Aulakh; Pauline Mouches; Anup Tuladhar; Matthias Wilms; Sönke Langner; Nils D. Forkert

doi:10.1117/12.2612728

4 April 2022 A comparative analysis of the impact of data distribution on distributed learning with a traveling model for brain age prediction

Raissa Souza, Agampreet Aulakh, Pauline Mouches, Anup Tuladhar, Matthias Wilms, Sönke Langner, Nils D. Forkert

Author Affiliations +

Proceedings Volume 12037, Medical Imaging 2022: Imaging Informatics for Healthcare, Research, and Applications; 1203702 (2022) https://doi.org/10.1117/12.2612728
Event: SPIE Medical Imaging, 2022, San Diego, California, United States

Abstract

Limited access to medical datasets, due to regulations that protect patient data, is a major hinderance to the development of machine learning models for computer-aided diagnosis tools using medical images. Distributed learning is an alternative to training machine learning models on centrally collected data that solves data sharing issues. The main idea of distributed learning is to train models remotely at each medical center rather than collecting the data in a central database, thereby avoiding sharing data between centers and model developers. In this work, we propose a travelling model that performs distributed learning for biological brain age prediction using morphological measurements of different brain structures. We specifically investigate the impact of nonidentically distributed data between collaborators on the performance of the travelling model. Our results, based on a large dataset of 2058 magnetic resonance imaging scans, demonstrate that transferring the model weights between the centers more frequently achieves results (mean age prediction error = 5.89 years) comparable to central learning implementations (mean age prediction error = 5.93 years), which were trained using the data from all sites hosted together at a central location. Moreover, we show that our model does not suffer from catastrophic forgetting, and that data distribution is less important than the number of times that the model travels between collaborators.

Conference Presentation

Citation Download Citation

Raissa Souza, Agampreet Aulakh, Pauline Mouches, Anup Tuladhar, Matthias Wilms, Sönke Langner, and Nils D. Forkert "A comparative analysis of the impact of data distribution on distributed learning with a traveling model for brain age prediction", Proc. SPIE 12037, Medical Imaging 2022: Imaging Informatics for Healthcare, Research, and Applications, 1203702 (4 April 2022); https://doi.org/10.1117/12.2612728

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available