Cargando…

How Well Do Self-Supervised Models Transfer to Medical Imaging?

Self-supervised learning approaches have seen success transferring between similar medical imaging datasets, however there has been no large scale attempt to compare the transferability of self-supervised models against each other on medical images. In this study, we compare the generalisability of...

Descripción completa

Detalles Bibliográficos
Autores principales: Anton, Jonah, Castelli, Liam, Chan, Mun Fai, Outters, Mathilde, Tang, Wan Hee, Cheung, Venus, Shukla, Pancham, Walambe, Rahee, Kotecha, Ketan
Formato: Online Artículo Texto
Lenguaje:English
Publicado: MDPI 2022
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9782186/
https://www.ncbi.nlm.nih.gov/pubmed/36547485
http://dx.doi.org/10.3390/jimaging8120320
_version_ 1784857281762951168
author Anton, Jonah
Castelli, Liam
Chan, Mun Fai
Outters, Mathilde
Tang, Wan Hee
Cheung, Venus
Shukla, Pancham
Walambe, Rahee
Kotecha, Ketan
author_facet Anton, Jonah
Castelli, Liam
Chan, Mun Fai
Outters, Mathilde
Tang, Wan Hee
Cheung, Venus
Shukla, Pancham
Walambe, Rahee
Kotecha, Ketan
author_sort Anton, Jonah
collection PubMed
description Self-supervised learning approaches have seen success transferring between similar medical imaging datasets, however there has been no large scale attempt to compare the transferability of self-supervised models against each other on medical images. In this study, we compare the generalisability of seven self-supervised models, two of which were trained in-domain, against supervised baselines across eight different medical datasets. We find that ImageNet pretrained self-supervised models are more generalisable than their supervised counterparts, scoring up to 10% better on medical classification tasks. The two in-domain pretrained models outperformed other models by over 20% on in-domain tasks, however they suffered significant loss of accuracy on all other tasks. Our investigation of the feature representations suggests that this trend may be due to the models learning to focus too heavily on specific areas.
format Online
Article
Text
id pubmed-9782186
institution National Center for Biotechnology Information
language English
publishDate 2022
publisher MDPI
record_format MEDLINE/PubMed
spelling pubmed-97821862022-12-24 How Well Do Self-Supervised Models Transfer to Medical Imaging? Anton, Jonah Castelli, Liam Chan, Mun Fai Outters, Mathilde Tang, Wan Hee Cheung, Venus Shukla, Pancham Walambe, Rahee Kotecha, Ketan J Imaging Article Self-supervised learning approaches have seen success transferring between similar medical imaging datasets, however there has been no large scale attempt to compare the transferability of self-supervised models against each other on medical images. In this study, we compare the generalisability of seven self-supervised models, two of which were trained in-domain, against supervised baselines across eight different medical datasets. We find that ImageNet pretrained self-supervised models are more generalisable than their supervised counterparts, scoring up to 10% better on medical classification tasks. The two in-domain pretrained models outperformed other models by over 20% on in-domain tasks, however they suffered significant loss of accuracy on all other tasks. Our investigation of the feature representations suggests that this trend may be due to the models learning to focus too heavily on specific areas. MDPI 2022-12-01 /pmc/articles/PMC9782186/ /pubmed/36547485 http://dx.doi.org/10.3390/jimaging8120320 Text en © 2022 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Anton, Jonah
Castelli, Liam
Chan, Mun Fai
Outters, Mathilde
Tang, Wan Hee
Cheung, Venus
Shukla, Pancham
Walambe, Rahee
Kotecha, Ketan
How Well Do Self-Supervised Models Transfer to Medical Imaging?
title How Well Do Self-Supervised Models Transfer to Medical Imaging?
title_full How Well Do Self-Supervised Models Transfer to Medical Imaging?
title_fullStr How Well Do Self-Supervised Models Transfer to Medical Imaging?
title_full_unstemmed How Well Do Self-Supervised Models Transfer to Medical Imaging?
title_short How Well Do Self-Supervised Models Transfer to Medical Imaging?
title_sort how well do self-supervised models transfer to medical imaging?
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9782186/
https://www.ncbi.nlm.nih.gov/pubmed/36547485
http://dx.doi.org/10.3390/jimaging8120320
work_keys_str_mv AT antonjonah howwelldoselfsupervisedmodelstransfertomedicalimaging
AT castelliliam howwelldoselfsupervisedmodelstransfertomedicalimaging
AT chanmunfai howwelldoselfsupervisedmodelstransfertomedicalimaging
AT outtersmathilde howwelldoselfsupervisedmodelstransfertomedicalimaging
AT tangwanhee howwelldoselfsupervisedmodelstransfertomedicalimaging
AT cheungvenus howwelldoselfsupervisedmodelstransfertomedicalimaging
AT shuklapancham howwelldoselfsupervisedmodelstransfertomedicalimaging
AT walamberahee howwelldoselfsupervisedmodelstransfertomedicalimaging
AT kotechaketan howwelldoselfsupervisedmodelstransfertomedicalimaging