Benchmarking Collaborative Learning Methods Cost-Effectiveness for Prostate Segmentation - Inria - Institut national de recherche en sciences et technologies du numérique
Communication Dans Un Congrès Année : 2023

Benchmarking Collaborative Learning Methods Cost-Effectiveness for Prostate Segmentation

Résumé

Healthcare data is often split into medium/small-sized collections across multiple hospitals and access to it is encumbered by privacy regulations. This brings difficulties to use them for the development of machine learning and deep learning models, which are known to be data-hungry. One way to overcome this limitation is to use collaborative learning (CL) methods, which allow hospitals to work collaboratively to solve a task, without the need to explicitly share local data. In this paper, we address a prostate segmentation problem from MRI in a collaborative scenario by comparing two different approaches: federated learning (FL) and consensus-based methods (CBM). To the best of our knowledge, this is the first work in which CBM, such as label fusion techniques, are used to solve a problem of collaborative learning. In this setting, CBM combine predictions from locally trained models to obtain a federated strong learner with ideally improved robustness and predictive variance properties. Our experiments show that, in the considered practical scenario, CBMs provide equal or better results than FL, while being highly cost-effective. Our results demonstrate that the consensus paradigm may represent a valid alternative to FL for typical training tasks in medical imaging.
Fichier principal
Vignette du fichier
ConsensusBenchmarkV2.pdf (636.48 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Licence

Dates et versions

hal-04357349 , version 1 (21-12-2023)

Licence

Identifiants

Citer

Lucia Innocenti, Michela Antonelli, Francesco Cremonesi, Kenaan Sarhan, Alejandro Granados, et al.. Benchmarking Collaborative Learning Methods Cost-Effectiveness for Prostate Segmentation. ECML - PharML - Applications of Machine Learning in Pharma and Healthcare (Workshop at ECML PKDD 2023), Sep 2023, Turin (IT), Italy. ⟨10.48550/arXiv.2309.17097⟩. ⟨hal-04357349⟩
43 Consultations
69 Téléchargements

Altmetric

Partager

More