Resource Optimized Neural Architecture Search for 3D Medical Image Segmentation

09/02/2019
by   Woong Bae, et al.
0

Neural Architecture Search (NAS), a framework which automates the task of designing neural networks, has recently been actively studied in the field of deep learning. However, there are only a few NAS methods suitable for 3D medical image segmentation. Medical 3D images are generally very large; thus it is difficult to apply previous NAS methods due to their GPU computational burden and long training time. We propose the resource-optimized neural architecture search method which can be applied to 3D medical segmentation tasks in a short training time (1.39 days for 1GB dataset) using a small amount of computation power (one RTX 2080Ti, 10.8GB GPU memory). Excellent performance can also be achieved without retraining(fine-tuning) which is essential in most NAS methods. These advantages can be achieved by using a reinforcement learning-based controller with parameter sharing and focusing on the optimal search space configuration of macro search rather than micro search. Our experiments demonstrate that the proposed NAS method outperforms manually designed networks with state-of-the-art performance in 3D medical image segmentation.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
06/13/2019

Scalable Neural Architecture Search for 3D Medical Image Segmentation

In this paper, a neural architecture search (NAS) framework is proposed ...
research
12/20/2019

C2FNAS: Coarse-to-Fine Neural Architecture Search for 3D Medical Image Segmentation

3D convolution neural networks (CNN) have been proved very successful in...
research
06/06/2019

V-NAS: Neural Architecture Search for Volumetric Medical Image Segmentation

Deep learning algorithms, in particular 2D and 3D fully convolutional ne...
research
07/28/2020

Efficient OCT Image Segmentation Using Neural Architecture Search

In this work, we propose a Neural Architecture Search (NAS) for retinal ...
research
03/29/2021

DiNTS: Differentiable Neural Network Topology Search for 3D Medical Image Segmentation

Recently, neural architecture search (NAS) has been applied to automatic...
research
05/07/2019

Efficient Neural Architecture Search on Low-Dimensional Data for OCT Image Segmentation

Typically, deep learning architectures are handcrafted for their respect...
research
12/17/2020

On the performance of deep learning for numerical optimization: an application to protein structure prediction

Deep neural networks have recently drawn considerable attention to build...

Please sign up or login with your details

Forgot password? Click here to reset