An ENAS Based Approach for Constructing Deep Learning Models for Breast Cancer Recognition from Ultrasound Images

Mohammed Ahmed, Hongbo Du, Alaa AlZoubi

Abstract: Deep Convolutional Neural Networks (CNN) provides an "end-to-end" solution for image pattern recognition with impressive performance in many areas of application including medical imaging. Most CNN models of high performance use hand-crafted network architectures that require expertise in CNNs to utilise their potentials. In this paper, we applied the Efficient Neural Architecture Search (ENAS) method to find optimal CNN architectures for classifying breast lesions from ultrasound (US) images. Our empirical study with a dataset of 524 US images shows that the optimal models generated by ENAS achieve an average accuracy of 89.3%, surpassing other hand-crafted alternatives. Furthermore, the models are simpler in complexity and more efficient. Our study demonstrates that the ENAS approach to CNN model design is a promising direction for classifying ultrasound images of breast lesions.

Paper Type: well-validated application

Track: short paper

Keywords: Efficient Neural Architecture Search, Ultrasound Image, Breast Cancer, Deep Learning

Presentation Upload Agreement: I agree that my presentation material (videos and slides) will be made public.