A systematic study of transfer learning for colorectal cancer detection

in: Informatics in Medicine Unlocked (2023)
Luo, Ruihao; Bocklitz, Thomas W.
Background and objective: With the rapid development of data science methods like deep learning, these methods have already been used into the field of healthcare and medicine. However, due to regulations and ethical issues, it is practically difficult to obtain large amount of medical data for training deep learning models. Transfer learning is a powerful tool to reuse the knowledge gained from different domains, which makes it possible to retrain deep learning models only with small datasets in medical image processing. In this contribution, a systematic study of model transfer techniques like fine tuning parts of the network or adding additional layers, for medical image data was conducted. Methods: The study accomplished a binary classification task based on a colorectal cancer dataset, including microsatellite unstable or hypermutated (MSIMUT) and microsatellite stable (MSS) images. By using K-fold cross-validation, the performances of five pretrained models (DenseNet121, DenseNet201, InceptionV3, MobileNetV2 and ResNet50) were assessed according to balanced accuracy. As baseline methods, combinations of transfer learning as feature extractor and principal component analysis with linear discriminant analysis (PCA-LDA) or support vector machine (PCA-SVM) were utilised, and compared with the transfer learning counterparts. Conclusions: The results have shown that adding convolutional layers perform obviously better than simply using the original network or fine-tuning some last layers of the network. Furthermore, a proposed bagging method performed well on a testing dataset. This study reduces the workload for future transfer learning tasks in the biomedical domain and allows to test promising transfer learning strategies first.

Third party cookies & scripts

This site uses cookies. For optimal performance, smooth social media and promotional use, it is recommended that you agree to third party cookies and scripts. This may involve sharing information about your use of the third-party social media, advertising and analytics website.
For more information, see privacy policy and imprint.
Which cookies & scripts and the associated processing of your personal data do you agree with?

You can change your preferences anytime by visiting privacy policy.