AutoSegNet: An Automated Neural Network for Image Segmentation

Abstract

Neural Architecture Search (NAS) has drawn significant attention as a tool for automatically constructing deep neural networks. The generated neural networks are mainly applied for image classification, and natural language processing. However, there are increasing demands for image segmentation in various areas, such as medical image processing, satellite image object location, and autopilot technology. We propose a NAS method called Automated Segmentation Network (AutoSegNet), targeting industrial and medical image segmentation. The search architectures are constructed by stacking the downsampling layer, the bridge layer, and the upsampling layer, which are explored by a recurrent neural network. Compared with other related methods for image segmentation, the proposed method has a small search space but can explore most of the-state-of-the-art supervised image segmentation models. We perform verification on two datasets, and the results show that AutoSegNet achieves superior segmentation results with clear and continuous segmented edges, as well as better image details.