Advertisement
Published in:
2020 | OriginalPaper | Chapter
Mid-Level Feature Extractor for Transfer Learning to Small-Scale Dataset of Medical Images
Authors : Dong-ho Lee, Yeon Lee, Byeong-seok Shin
Published in: Advances in Computer Science and Ubiquitous Computing
Publisher: Springer Singapore
Activate our intelligent search to find suitable subject content or patents.
Select sections of text to find matching patents with Artificial Intelligence. powered by
Select sections of text to find additional relevant content using AI-assisted search. powered by
Abstract
In fine-tuning-based transfer learning, the size of the dataset may affect the learning accuracy. When a dataset scale is small, fine-tuning-based transfer learning methods use high computing costs, similar to a large-scale dataset. we propose a mid-level feature extractor that only retrains the mid-level convolutional layers, resulting in increased efficiency and reduced computing costs. This mid-level feature extractor is likely to provide an effective alternative in training a small-scale medical image dataset. The performance of the mid-level feature extractor is compared with performance of low- and high-level feature extractors, as well as the fine-tuning method. The mid-level feature extractor takes shorter time to converge than other methods, and it shows good accuracy, obtaining an area under the ROC curve (AUC) of 0.87 in untrained test dataset that is very different from training dataset.