ST-FL: style transfer preprocessing in federated learning for COVID-19 segmentation

Antonios Georgiadis; Varun Babbar; Fran Silavong; Sean Moran; Rob Otter

doi:10.1117/12.2611096

4 April 2022 ST-FL: style transfer preprocessing in federated learning for COVID-19 segmentation

Antonios Georgiadis, Varun Babbar, Fran Silavong, Sean Moran, Rob Otter

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

Abstract

Chest Computational Tomography (CT) scans present low cost, speed and objectivity for COVID-19 diagnosis and deep learning methods have shown great promise in assisting the analysis and interpretation of these images. Most hospitals or countries can train their own models using in-house data, however empirical evidence shows that those models perform poorly when tested on new unseen cases, surfacing the need for coordinated global collaboration. Due to privacy regulations, medical data sharing between hospitals and nations is extremely difficult. We propose a GAN-augmented federated learning model, dubbed ST-FL (Style Transfer Federated Learning), for COVID-19 image segmentation. Federated learning (FL) permits a centralised model to be learned in a secure manner from heterogeneous datasets located in disparate private data silos. We demonstrate that the widely varying data quality on FL client nodes leads to a sub-optimal centralised FL model for COVID-19 chest CT image segmentation. ST-FL is a novel FL framework that is robust in the face of highly variable data quality at client nodes. The robustness is achieved by a denoising CycleGAN model at each client of the federation that maps arbitrary quality images into the same target quality, counteracting the severe data variability evident in real-world FL use-cases. Each client is provided with the target style, which is the same for all clients, and trains their own denoiser. Our qualitative and quantitative results suggest that this FL model performs comparably to, and in some cases better than, a model that has centralised access to all the training data.

Conference Presentation

Citation Download Citation

Antonios Georgiadis, Varun Babbar, Fran Silavong, Sean Moran, and Rob Otter "ST-FL: style transfer preprocessing in federated learning for COVID-19 segmentation", Proc. SPIE 12037, Medical Imaging 2022: Imaging Informatics for Healthcare, Research, and Applications, 1203704 (4 April 2022); https://doi.org/10.1117/12.2611096

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