基于神经架构搜索的色素性皮损图像分类方法-《计算机技术与发展》

文章信息/Info

Title:: Pigmented Skin Lesions Image Classification Based on Neural Architecture Search

作者:: 何晴¹ ; 杨铁军² ; 黄琳^1*; 1. 桂林理工大学信息科学与工程学院,广西桂林 541000;
2. 桂林医学院智能医学与生物技术学院,广西桂林 541000

Author(s):: HE Qing1 ; YANG Tie-jun2 ; HUANG Lin1*; 1. School of Information Science and Engineering,Guilin University of Technology,Guilin 541000,China;
2. School of Intelligent Medicine and Biotechnology,Guilin Medical University,Guilin 541000,China

Keywords:: neural architecture search; single - path activation; gradient descent; convolutional neural network; pigmented skin lesionimage classification

摘要:: 针对卷积神经网络设计高度依赖专家经验、需要大量参数调优和效率低的问题,提出了一种基于单路径激活搜索策略的神经架构搜索方法( SPA-NAS) ,并应用于色素性皮损图像分类。该方法将搜索空间构建为一个过参数化神经网络架构,该架构包含了所有的路径,并且每条路径都被分配一个架构参数以表示路径的占比强度。为了避免搜索所有路径,提出了一种单路径激活策略对构建的过参数化神经网络架构进行路径剪枝,以得到一个更加精简的子架构。搜索时,采用梯度下降法学习和优化架构参数,得到最佳子架构。最后,采用子架构堆叠方式构建色素性皮损图像分类卷积神经网络。实验表明,该方法自动构建的卷积神经网络取得了比 Dilated-VGG19 和 ARL-CNN 等 SOTA 方法更高的分类准确性,在 ISIC2017 和 HAM10000 数据集上的平均敏感度分别为 62. 4% 和 69. 8% 。

Abstract:: Because design of convolutional neural network ( CNN) requires abundant expert experience,considerable parameter tuning andlow efficiency,we propose a neural architecture search approach based on single path activation search strategy ( SPA-NAS) applied tothe image classification of pigmented skin lesions. This method constructs the search space as an hyperparameter neural networkarchitecture,which contains all paths,and each path is assigned an architecture parameter to represent the proportion of the path. In orderto avoid searching all paths,a single-path activation strategy is proposed to prune the constructed hyperparameter network architecture toobtain a more streamlined child architecture. When searching,the gradient descent method is used to learn and optimize the architecture parameters to search the best child architecture. Finally,a CNN for pigmented skin lesion image classification is constructed by usingchild architecture stacking. The test results showed that the CNN built with the proposed approach was more accurate than state-of the-art ( SOTA) approaches such as Dilated - VGG19 and ARL - CNN in terms of image classification. The average sensitivity of theproposed method on the ISIC2017 and HAM10000 datasets was 62. 4% and 69. 8% ,respectively.