Robust and data-efficient generalization of self-supervised machine learning for diagnostic imaging.
Nat Biomed Eng
; 7(6): 756-779, 2023 06.
Article
em En
| MEDLINE
| ID: mdl-37291435
ABSTRACT
Machine-learning models for medical tasks can match or surpass the performance of clinical experts. However, in settings differing from those of the training dataset, the performance of a model can deteriorate substantially. Here we report a representation-learning strategy for machine-learning models applied to medical-imaging tasks that mitigates such 'out of distribution' performance problem and that improves model robustness and training efficiency. The strategy, which we named REMEDIS (for 'Robust and Efficient Medical Imaging with Self-supervision'), combines large-scale supervised transfer learning on natural images and intermediate contrastive self-supervised learning on medical images and requires minimal task-specific customization. We show the utility of REMEDIS in a range of diagnostic-imaging tasks covering six imaging domains and 15 test datasets, and by simulating three realistic out-of-distribution scenarios. REMEDIS improved in-distribution diagnostic accuracies up to 11.5% with respect to strong supervised baseline models, and in out-of-distribution settings required only 1-33% of the data for retraining to match the performance of supervised models retrained using all available data. REMEDIS may accelerate the development lifecycle of machine-learning models for medical imaging.
Texto completo:
1
Bases de dados:
MEDLINE
Assunto principal:
Aprendizado de Máquina
/
Aprendizado de Máquina Supervisionado
Tipo de estudo:
Diagnostic_studies
Idioma:
En
Revista:
Nat Biomed Eng
Ano de publicação:
2023
Tipo de documento:
Article
País de afiliação:
Estados Unidos