In this paper, a neural architecture search (NAS) framework is proposed for
3D medical image segmentation, to automatically optimize a neural architecture
from a large design space. Our NAS framework searches the structure of each
layer including neural connectivities and operation types in both of the
encoder and decoder. Since optimizing over a large discrete architecture space
is difficult due to high-resolution 3D medical images, a novel stochastic
sampling algorithm based on a continuous relaxation is also proposed for
scalable gradient based optimization. On the 3D medical image segmentation
tasks with a benchmark dataset, an automatically designed architecture by the
proposed NAS framework outperforms the human-designed 3D U-Net, and moreover
this optimized architecture is well suited to be transferred for different
tasks.

本文提出了一种神经体系结构搜索 (NAS) 框架，用于自动优化大规模设计空间中的神经体系结构，支持 3D 医学图像分割任务，其优化准确度在基准测试数据集上优于人类设计的 3D U-Net，经过优化的体系结构适用于不同的任务。