Automatically Designing CNN Architectures for Medical Image Segmentation

by   Aliasghar Mortazi, et al.
University of Central Florida

Deep neural network architectures have traditionally been designed and explored with human expertise in a long-lasting trial-and-error process. This process requires huge amount of time, expertise, and resources. To address this tedious problem, we propose a novel algorithm to optimally find hyperparameters of a deep network architecture automatically. We specifically focus on designing neural architectures for medical image segmentation task. Our proposed method is based on a policy gradient reinforcement learning for which the reward function is assigned a segmentation evaluation utility (i.e., dice index). We show the efficacy of the proposed method with its low computational cost in comparison with the state-of-the-art medical image segmentation networks. We also present a new architecture design, a densely connected encoder-decoder CNN, as a strong baseline architecture to apply the proposed hyperparameter search algorithm. We apply the proposed algorithm to each layer of the baseline architectures. As an application, we train the proposed system on cine cardiac MR images from Automated Cardiac Diagnosis Challenge (ACDC) MICCAI 2017. Starting from a baseline segmentation architecture, the resulting network architecture obtains the state-of-the-art results in accuracy without performing any trial-and-error based architecture design approaches or close supervision of the hyperparameters changes.


page 1

page 2

page 3

page 4


Dual self-distillation of U-shaped networks for 3D medical image segmentation

U-shaped networks and its variants have demonstrated exceptional results...

SegNAS3D: Network Architecture Search with Derivative-Free Global Optimization for 3D Image Segmentation

Deep learning has largely reduced the need for manual feature selection ...

Selecting the Best Optimizers for Deep Learning based Medical Image Segmentation

The goal of this work is to identify the best optimizers for deep learni...

MixSearch: Searching for Domain Generalized Medical Image Segmentation Architectures

Considering the scarcity of medical data, most datasets in medical image...

Leveraging Image Complexity in Macro-Level Neural Network Design for Medical Image Segmentation

Recent progress in encoder-decoder neural network architecture design ha...

DLTK: State of the Art Reference Implementations for Deep Learning on Medical Images

We present DLTK, a toolkit providing baseline implementations for effici...

Please sign up or login with your details

Forgot password? Click here to reset