[发明专利]基于结构熵采样的多目标架构搜索骨质疏松图像识别方法

说明书

本发明公开了一种基于结构熵采样的多目标架构搜索骨质疏松图像识别方法，包括：1收集含骨密度信息的腰椎X光图像数据并进行预处理，划分为训练集和验证集；2初始化神经网络结构种群；3将种群中所有个体解码成对应的神经网络架构，并使用X光图片训练集对每个神经网络进行训练，将不同类别识别的准确率和神经网络的参数量作为适应度值；4根据个体的适应度值进行交配池选择、基于结构熵采样的交叉变异和环境选择；5重复上述过程进行启发式网络结构搜索，达到预设定条件结束进化，并输出最优的网络结构；6利用训练集重新训练所获得的最优网络结构。本发明能自动搜索适应腰椎X光图像数据集的卷积神经网络架构，实现对腰椎X光图像的分类。

技术领域

本发明属于计算机/机器学习中的自动化机器学习和神经网络架构搜索领域，具体的说是一种基于结构熵采样的多目标架构搜索骨质疏松图像识别方法。

背景技术

医疗图像识别领域利用影像测试的形式，诸如X射线、超声波、CT和核磁共振，来辅助医生。目前随着深度学习的发展，卷积神经网络(CNNs)已经成为图像识别领域的主流技术，同时在医学图像的识别任务上时提供了巨大的诊断支持。因此，设计具有较优性能的神经网络结构能够极大提升医疗图像的识别准确率。

现有的神经架构搜索设计方法包括：基于人工设计的网络架构设计方法和自动化网络架构搜索方法(NAS)；然而基于人工设计的网络架构设计方法往往需要大量的专业知识和实验时间，诸如线性结构模型VGGNet、残差结构模型ResNet、多分支结构模型GoogLeNet这些经典的卷积神经网络结构都是专业人员精心设计的。

自动化网络架构搜索方法(NAS)主要包含三种搜索策略：基于强化学习、基于梯度优化和基于进化算法。

基于强化学习的方法主要是利用强化学习作为优化算法，将卷积神经网络结构中的操作算子建模成序列输入到RNN中，作为强化学习不断学习优化的参数，最终搜索出优秀的算子结构以及对应的连接关系。这类方法的效果优于上述经典网络，但是由于搜索空间庞大，非常的耗时。

基于梯度优化的方法主要将网络结构建模成多维优化问题中的决策变量，然后使用梯度法来优化搜索空间中的参数，从而能够在搜索空间中连续的探索结构，而不是离散的搜索，这种方法更具有可解释性。

基于进化算法的方法主要将网络结构建模成种群个体，基于多目标进化算法进行启发式的探索，由于进化算法在多目标和全局优化上有着显著的优势，因此该方法往往能有很好的性能。

上述的网络结构搜索方法相比传统的人工设计网络具有优势，但是均非常的耗时并且容易忽略搜索空间中解的分布性。

发明内容

本发明为解决现有技术的不足之处，提供了一种基于结构熵采样的多目标架构搜索骨质疏松图像识别方法，以期望能够搜索出更加有效的神经网络结构用于骨质疏松图像的识别分类，从而提高分类准确性。

为解决上述技术问题，本发明采用的技术方案是：

本发明一种基于结构熵采样的多目标架构搜索骨质疏松图像识别方法的特点是按以下步骤进行：

步骤0、采集n张含有骨密度标签的腰椎X光图像并进行预处理，得到数据集D_all，并划分为训练集D_tra和验证集D_vall，且腰椎X光图像共有I种类别；

步骤1、构建由NC个卷积块和RC个下采样块组成的卷积神经网络结构A；每个卷积块和下采样块的节点个数相同，并由m个节点构成，每个节点表示一种操作，所述操作的类型包括卷积操作、下采样操作、全连接操作以及通道注意力操作，节点之间存在连接用“1”表示，否则用“0”表示；所述卷积块中的所有操作的步长均为1，所述下采样块中所有操作的步长均为2；所述神经网络结构A由NC个卷积块和RC个下采样块依次堆叠而成；所述神经网络结构A的稠密度由所有节点连接关系中“1”的个数表示；