Share Email Print

Proceedings Paper

Genetic algorithm for machine learning architecture selection for breast MRI classification
Format Member Price Non-Member Price
PDF $17.00 $21.00

Paper 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.

Paper Details

Date Published: 16 March 2020
PDF: 6 pages
Proc. SPIE 11314, Medical Imaging 2020: Computer-Aided Diagnosis, 1131422 (16 March 2020); doi: 10.1117/12.2547490
Show Author Affiliations
Peter S. Carras, Michigan State Univ. (United States)
Carina Pereira, Univ. of Washington (United States)
Debosmita Biswas, Univ. of Washington (United States)
Christoph Lee, Univ. of Washington (United States)
Savannah Partridge, Univ. of Washington (United States)
Adam M. Alessio, Michigan State Univ. (United States)

Published in SPIE Proceedings Vol. 11314:
Medical Imaging 2020: Computer-Aided Diagnosis
Horst K. Hahn; Maciej A. Mazurowski, Editor(s)

© SPIE. Terms of Use
Back to Top
Sign in to read the full article
Create a free SPIE account to get access to
premium articles and original research
Forgot your username?