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...
Autores principales: | , , , , , , , , |
---|---|
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 |