CNN-based transfer learning with 10-fold cross-validation: a novel approach for customized education of mammography training

Xuetong Tao; Ziba Gandomkar; Tong Li; Jessica Yi; Patrick C. Brennan; Warren M. Reed

doi:10.1117/12.3006659

29 March 2024 CNN-based transfer learning with 10-fold cross-validation: a novel approach for customized education of mammography training

Xuetong Tao, Ziba Gandomkar, Tong Li, Jessica Yi, Patrick C. Brennan, Warren M. Reed

Author Affiliations +

Proceedings Volume 12929, Medical Imaging 2024: Image Perception, Observer Performance, and Technology Assessment; 1292904 (2024) https://doi.org/10.1117/12.3006659
Event: SPIE Medical Imaging, 2024, San Diego, California, United States

Abstract

Early detection of breast cancer through screening mammography is crucial. However, the interpretation of mammograms is prone to high error rates, and radiologists often exhibit common errors specific to their practice regions. It is essential to identify prevalent errors and offer tailored mammography training to address these region-specific challenges. This study investigated the feasibility of leveraging Convolutional Neural Networks (CNN) with transfer learning to identify areas in screening mammograms that may contribute to a high proportion of false positive diagnoses by radiologists from the same geographical region. We collected mammography test sets evaluated by a cohort of Australian radiologists and segmented error-related patches based on their assessments. Each patch was labeled as “easy” or “difficult”, and subsequently, we proposed a patch-wise ResNet model to predict the difficulty level of each patch. Specifically, we employed the pre-trained ResNet-18, ResNet-50, and ResNet-101 as feature extractors. During training, we modified and fine-tuned the fully connected layers for our target task while keeping the convolutional layers frozen. The model’s performance was evaluated using 10-fold cross-validation, and the transferred ResNet-50 obtained the highest performance, achieving Receiver Operating Characteristics Area Under the Curve (AUC) values of 0.975 (±0.011) on the validation sets. In conclusion, our study demonstrated the feasibility of employing CNN-based transfer learning to identify the prevalent errors in specific radiology communities. This approach shows promise in automating the customization of mammography training materials to mitigate errors among radiologists in a region.

Conference Presentation

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Xuetong Tao, Ziba Gandomkar, Tong Li, Jessica Yi, Patrick C. Brennan, and Warren M. Reed "CNN-based transfer learning with 10-fold cross-validation: a novel approach for customized education of mammography training", Proc. SPIE 12929, Medical Imaging 2024: Image Perception, Observer Performance, and Technology Assessment, 1292904 (29 March 2024); https://doi.org/10.1117/12.3006659

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