Genetic algorithm for machine learning architecture selection for breast MRI classification

Peter S. Carras; Carina Pereira; Debosmita Biswas; Christoph Lee; Savannah Partridge; Adam M. Alessio

doi:10.1117/12.2547490

16 March 2020 Genetic algorithm for machine learning architecture selection for breast MRI classification

Peter S. Carras, Carina Pereira, Debosmita Biswas, Christoph Lee, Savannah Partridge, Adam M. Alessio

Proceedings Volume 11314, Medical Imaging 2020: Computer-Aided Diagnosis; 1131422 (2020) https://doi.org/10.1117/12.2547490
Event: SPIE Medical Imaging, 2020, Houston, Texas, United States

Abstract

Convolutional neural networks (CNN) are increasingly used for image classification tasks. In general, the architectures of these networks are set ad hoc with little justification for selecting various components, such as the number of layers, layer depth, and convolution settings. In this work, we develop a structured approach to explore and select architectures that provide optimal classification performance. This was developed with an IRB-approved data set with 9,216 2-D maximum intensity projection (MIP) MRI breast images, containing breast cancer malignant or benign classes. This data set was divided into 7,372 training, 922 validation, and 922 test images. The architecture search method employs a genetic algorithm to find optimal ResNet-based classification architectures through crossover and mutation. Each generation, new model architectures are created from the genetic algorithm and trained with supervised machine learning. This search method identifies the model with the highest area under the ROC curve (AUC). The genetic algorithm produced an optimal model architecture which classifies malignancy in images with 76% accuracy and achieves an AUC score of .83. This approach offers a rational framework for automatic architecture exploration, potentially leading to a set of more efficient and generalizable CNN-based classifiers.

Citation Download Citation

Peter S. Carras, Carina Pereira, Debosmita Biswas, Christoph Lee, Savannah Partridge, and Adam M. Alessio "Genetic algorithm for machine learning architecture selection for breast MRI classification", Proc. SPIE 11314, Medical Imaging 2020: Computer-Aided Diagnosis, 1131422 (16 March 2020); https://doi.org/10.1117/12.2547490

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