Knowledge transfer evolutionary search for lightweight neural architecture with dynamic inference

被引:5
作者
Qian, Xiaoxue [1 ,2 ,3 ,4 ]
Liu, Fang [1 ,2 ,3 ,4 ]
Jiao, Licheng [1 ,2 ,3 ,4 ]
Zhang, Xiangrong [1 ,2 ,3 ,4 ]
Huang, Xinyan [1 ,2 ,3 ,4 ]
Li, Shuo [1 ,2 ,3 ,4 ]
Chen, Puhua [1 ,2 ,3 ,4 ]
Liu, Xu [1 ,2 ,3 ,4 ]
机构
[1] Xidian Univ, Sch Artificial Intelligence, Xian 710071, Peoples R China
[2] Key Lab Intelligent Percept & Image Understanding, Minist Educ, Xian, Peoples R China
[3] Int Res Ctr Intelligent Percept & Computat, Xian, Peoples R China
[4] Joint Int Res Lab Intelligent Percept & Computat, Beijing, Peoples R China
基金
中国国家自然科学基金;
关键词
Neural architecture search (NAS); Knowledge transfer; Dynamic inference; Image classification; NETWORK; DESIGN;
D O I
10.1016/j.patcog.2023.109790
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Relying on the availability of massive labeled samples, most neural architecture search (NAS) methods focus on searching large and complex models; and adopt fixed structures and parameters at the infer-ence stage. Few approaches automatically design lightweight networks for label-limited tasks and fur-ther consider the inference differences between inputs. To address these issues, we introduce evolution-ary computation (EC) and attention mechanism and propose a knowledge transfer evolutionary search for lightweight neural architecture with dynamic inference, then verify it using synthetic aperture radar (SAR) images. SAR image classification is a typical label-limited task due to the inherent imaging mecha-nism of SAR. We design the EC-based architecture search and attention-based dynamic inference for SAR image scene classification. Specifically, we build a SAR-tailored search space, explore topology pruning -based mutation operators to search lightweight architectures, and further design a dynamic Ridgelet con-volution capable of adaptive reasoning to enhance the representation ability of searched lightweight net-works. Moreover, we propose a knowledge transfer training strategy and hybrid evaluation criteria to ensure searching quickly and robustly. Experimental results show that the proposed method can search for superior neural architectures, thus improving the classification performance of SAR images.& COPY; 2023 Elsevier Ltd. All rights reserved.
引用
收藏
页数:12
相关论文
共 49 条
  • [1] [Anonymous], 2017, P 5 INT C LEARN REPR
  • [2] AdaResU-Net: Multiobjective adaptive convolutional neural network for medical image segmentation
    Baldeon-Calisto, Maria
    Lai-Yuen, Susana K.
    [J]. NEUROCOMPUTING, 2020, 392 : 325 - 340
  • [3] Bender G, 2018, PR MACH LEARN RES, V80
  • [4] Chenxi Liu, 2020, Computer Vision - ECCV 2020. 16th European Conference. Proceedings. Lecture Notes in Computer Science (LNCS 12349), P798, DOI 10.1007/978-3-030-58548-8_46
  • [5] Automatic Design of CNNs via Differentiable Neural Architecture Search for PolSAR Image Classification
    Dong, Hongwei
    Zou, Bin
    Zhang, Lamei
    Zhang, Siyu
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (09): : 6362 - 6375
  • [6] SAR Image segmentation based on convolutional-wavelet neural network and markov random field
    Duan, Yiping
    Liu, Fang
    Jiao, Licheng
    Zhao, Peng
    Zhang, Lu
    [J]. PATTERN RECOGNITION, 2017, 64 : 255 - 267
  • [7] Elsken T., 2018, P INT C LEARNING REP
  • [8] Elsken T, 2019, P INT C LEARNING REP
  • [9] Differentiable neural architecture learning for efficient neural networks
    Guo, Qingbei
    Wu, Xiao-Jun
    Kittler, Josef
    Feng, Zhiquan
    [J]. PATTERN RECOGNITION, 2022, 126
  • [10] Dynamic Neural Networks: A Survey
    Han, Yizeng
    Huang, Gao
    Song, Shiji
    Yang, Le
    Wang, Honghui
    Wang, Yulin
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (11) : 7436 - 7456