EMONAS: efficient multiobjective neural architecture search framework for 3D medical image segmentation

Maria G. Baldeon Calisto; Susana K. Lai-Yuen

doi:10.1117/12.2577088

15 February 2021 EMONAS: efficient multiobjective neural architecture search framework for 3D medical image segmentation

Maria G. Baldeon Calisto, Susana K. Lai-Yuen

Author Affiliations +

Proceedings Volume 11596, Medical Imaging 2021: Image Processing; 1159607 (2021) https://doi.org/10.1117/12.2577088
Event: SPIE Medical Imaging, 2021, Online Only

Abstract

Deep learning plays a critical role in medical image segmentation. Nevertheless, manually designing a neural network for a specific segmentation problem is a very difficult and time-consuming task due to the massive hyperparameter search space, long training time and large volumetric data. Therefore, most designed networks are highly complex, task specific and over-parametrized. Recently, multiobjective neural architecture search (NAS) methods have been proposed to automate the design of accurate and efficient segmentation architectures. However, they only search for either the macro- or micro-structure of the architecture, do not use the information produced during the optimization process to increase the efficiency of the search, and do not consider the volumetric nature of medical images. In this work, we propose EMONAS, an Efficient MultiObjective Neural Architecture Search framework for 3D medical image segmentation. EMONAS is composed of a search space that considers both the macro- and micro-structure of the architecture, and a surrogate-assisted multiobjective evolutionary based algorithm that efficiently searches for the best hyperparameters using a Random Forest surrogate and guiding selection probabilities. EMONAS is evaluated on the task of cardiac segmentation from the ACDC MICCAI challenge. The architecture found is ranked within the top 10 submissions in all evaluation metrics, performing better or comparable to other approaches while reducing the search time by more than 50% and having considerably fewer number of parameters.

Conference Presentation

Citation Download Citation

Maria G. Baldeon Calisto and Susana K. Lai-Yuen "EMONAS: efficient multiobjective neural architecture search framework for 3D medical image segmentation", Proc. SPIE 11596, Medical Imaging 2021: Image Processing, 1159607 (15 February 2021); https://doi.org/10.1117/12.2577088

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