Transfer learning for small-scale data classification using CNN filter replacement

Ryo Muto; Noriko Yata; Yoshitsugu Manabe

doi:10.1117/12.2590965

13 March 2021 Transfer learning for small-scale data classification using CNN filter replacement

Ryo Muto, Noriko Yata, Yoshitsugu Manabe

Proceedings Volume 11766, International Workshop on Advanced Imaging Technology (IWAIT) 2021; 1176626 (2021) https://doi.org/10.1117/12.2590965
Event: International Workshop on Advanced Imaging Technology 2021 (IWAIT 2021), 2021, Online Only

Abstract

Recently, object recognition using CNN is widespread. Still, medical images do not have a sufficient number of images because they require the doctor’s findings in the training dataset. On such a small-scale dataset, there is a problem that CNN cannot realize enough high recognition accuracy. As a solution to this problem, there is a method called transfer learning that reuses the weights learned on a large dataset. In addition, there is research on a method of pruning parameters unimportant for the target task during transfer learning. In this study, after transfer learning is performed, the convolution filter is evaluated using pruning, and the low evaluation filter is replaced with the high evaluation filter. In order to confirm the usefulness of the proposed method in recognition accuracy, we compare it with the three methods, i.e., transfer learning only, pruning, and initializing the filter. As a result, we were able to obtain a high recognition accuracy compared to other methods. We confirmed that CNN might be affected by replacing the filter in object recognition of small-scale datasets.

Citation Download Citation

Ryo Muto, Noriko Yata, and Yoshitsugu Manabe "Transfer learning for small-scale data classification using CNN filter replacement", Proc. SPIE 11766, International Workshop on Advanced Imaging Technology (IWAIT) 2021, 1176626 (13 March 2021); https://doi.org/10.1117/12.2590965

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available