Collaborative Computing: Networking, Applications and Worksharing. 15th EAI International Conference, CollaborateCom 2019, London, UK, August 19-22, 2019, Proceedings

Research Article

CNASV: A Convolutional Neural Architecture Search-Train Prototype for Computer Vision Task

Download
18 downloads
  • @INPROCEEDINGS{10.1007/978-3-030-30146-0_26,
        author={Tianbao Zhou and Yu Weng and Guosheng Yang},
        title={CNASV: A Convolutional Neural Architecture Search-Train Prototype for Computer Vision Task},
        proceedings={Collaborative Computing: Networking, Applications and Worksharing. 15th EAI International Conference, CollaborateCom 2019, London, UK, August 19-22, 2019, Proceedings},
        proceedings_a={COLLABORATECOM},
        year={2019},
        month={8},
        keywords={CNASV Image classification Image segmentation},
        doi={10.1007/978-3-030-30146-0_26}
    }
    
  • Tianbao Zhou
    Yu Weng
    Guosheng Yang
    Year: 2019
    CNASV: A Convolutional Neural Architecture Search-Train Prototype for Computer Vision Task
    COLLABORATECOM
    Springer
    DOI: 10.1007/978-3-030-30146-0_26
Tianbao Zhou1,*, Yu Weng1, Guosheng Yang1
  • 1: Minzu University of China
*Contact email: tianbaochou@163.com

Abstract

Neural Architecture Search (NAS) has become more and more prevalent in the field of deep learning in the past two years. Existing works often focus on image classification, and few works recently extend NAS to another computer vision task, such as semantic image segmentation. The semantic image segmentation is essentially a dense prediction for each pixel on whole image. Therefore, we choose the same basic primitive operations to build the search space for the two computer vision task respectively. Searching good neural network architectures and then training them from scratch is a regular procedure for NAS. In this paper, we design a prototype system that deploy search module and train module to collaborate with each other. Follow the former research, we initialize over-parameterized cells architecture and then transform to the continuous relaxation of the architecture to derive the good subnetwork by gradient descent. Our system can support any differential search algorithm, such as one-shot, DARTS or ProxylessNAS. We illustrate the effectiveness of our chosen primitive operations in the image classification and ability to transfer these operations to build search space for semantic image segmentation.