Neural architecture search algorithm based on voting scheme

被引:0
|
作者
Yang J. [1 ,2 ]
Zhang J. [1 ]
机构
[1] School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou
[2] Faculty of Geomatics, Lanzhou Jiaotong University, Lanzhou
关键词
3D model classification; group sparsity regularization; neural architecture search; performance estimator; weighted voting scheme;
D O I
10.37188/OPE.20223017.2119
中图分类号
学科分类号
摘要
A neural-architecture search algorithm based on a voting scheme was proposed to address the difference between network architectures that are automatically searched by existing algorithms and those that were evaluated by the algorithm. First, to solve the problem that uniform sampling ignores the importance of each network architecture, the training losses tested on small batch training data were used as performance estimators to sample candidate networks, thus concentrating computing resources on high-performance candidate network architectures. Second, a group sparsity regularization strategy was adopted to rank all candidate operations to solve the problem of selecting candidate operations in each node. This strategy could screen suitable candidate operations and further enhance the precision of path selection in the cell structure. Finally, by integrating the differentiable architecture search, noise and sparse regularization strategies, the optimal cell structure was selected using a weighted voting scheme, and the network architecture for 3D model recognition and classification was constructed. Experimental results indicate that the classification accuracy of the constructed network for 3D models reaches 93.9% on the ModelNet40 dataset, which is higher than that of current mainstream algorithms. The proposed algorithm effectively narrows the gap between the network architecture during the search and evaluation phases, thereby resolving the problem of inefficient network training caused by uniform sampling in previous neural-architecture search methods. © 2022 Guangxue Jingmi Gongcheng/Optics and Precision Engineering. All rights reserved.
引用
收藏
页码:2119 / 2132
页数:13
相关论文
共 30 条
  • [1] GUO Y, CHEN J, DU Q, Multi-way backpropagation for training compact deep neural networks[J], Neural Networks, 126, pp. 250-261, (2020)
  • [2] SIMONYAN K, ZISSERMAN A, Very deep convolutional networks for large-scale image recognition[EB/OL], (2014)
  • [3] HE K M, ZHANG X Y, REN S Q, Et al., Deep residual learning for image recognition[C], 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 770-778, (2016)
  • [4] CHARLES R Q, HAO S, MO K C, Et al., PointNet: deep learning on point sets for 3D classification and segmentation[C], 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 652-660, (2017)
  • [5] ZOPH B, VASUDEVAN V, SHLENS J, Learning transferable architectures for scalable image recognition[C], 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 8697-8710, (2018)
  • [6] LIU H X, SIMONYAN K, YANG Y M, DARTS: differentiable architecture search[EB/OL], (2018)
  • [7] CHEN X, XIE L X, WU J, Progressive differentiable architecture search: bridging the depth gap between search and evaluation[C], 2019 IEEE/CVF International Conference on Computer Vision (ICCV), pp. 1294-1303, (2019)
  • [8] DONG X Y, YANG Y, Searching for a robust neural architecture in four GPU hours[C], 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1761-1770, (2019)
  • [9] ZOPH B, LE Q V, Neural architecture search with reinforcement learning[EB/OL], (2016)
  • [10] ZHONG Z, YAN J J, WU W, Practical block-wise neural network architecture generation[C], 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 2423-2432, (2018)