[发明专利]神经网络结构的生成方法、装置、存储介质及电子设备

说明书

本申请公开了神经网络结构的生成方法、装置、存储介质及电子设备，涉及人工智能、深度学习技术领域，该方法可应用于图像处理或语音识别。具体实现方案为：对搜索空间内的网络结构集合进行采样，以得到多个候选网络结构，其中，候选网络结构包括多条边，各条边对应至少一种节点选取操作；采用训练数据集分别对多个候选网络结构进行训练，并获取训练后的多个候选网络结构基于各种节点选取操作的目标检测效果；以及根据目标检测效果，在搜索空间的网络结构集合之中进行搜索以生成目标神经网络结构，能够有效增强目标神经网络结构对目标检测的训练数据集的特征提取能力，提升神经网络结构的搜索效率，提升神经网络结构的搜索效果。

技术领域

本申请涉及计算机技术领域，具体涉及人工智能、深度学习技术领域，尤其涉及神经网络结构的生成方法、装置、存储介质及电子设备。

背景技术

近年来，随着深度学习以及计算机视觉技术的发展，人工设计的神经网络结构数量开始呈现爆发式增长，而人工设计的神经网络结构存在设计经验化、任务针对性不足等问题，由此，更多的用户希望针对当前需求的模型任务，产生特定化的网络结构以及参数，这种建模需求促使神经网络结构搜索方法的形成。

发明内容

提供了一种神经网络结构的生成方法、装置、存储介质及电子设备，能够有效增强目标神经网络结构对目标检测的训练数据集的特征提取能力，提升神经网络结构的搜索效率，提升神经网络结构的搜索效果。

根据第一方面，提供了一种神经网络结构的生成方法，包括：对搜索空间内的网络结构集合进行采样，以得到多个候选网络结构，其中，所述候选网络结构包括多条边，各条所述边对应至少一种节点选取操作；采用训练数据集分别对所述多个候选网络结构进行训练，并获取训练后的多个所述候选网络结构基于各种所述节点选取操作的目标检测效果；以及根据所述目标检测效果，在所述搜索空间的所述网络结构集合之中进行搜索以生成目标神经网络结构。

本申请实施例的神经网络结构的生成方法，通过对搜索空间内的网络结构集合进行采样，以得到多个候选网络结构，其中，候选网络结构包括多条边，各条边对应至少一种节点选取操作，并采用训练数据集分别对多个候选网络结构进行训练，并获取训练后的多个候选网络结构基于各种节点选取操作的目标检测效果；以及根据目标检测效果，在搜索空间的网络结构集合之中进行搜索以生成目标神经网络结构，能够有效增强目标神经网络结构对目标检测的训练数据集的特征提取能力，提升神经网络结构的搜索效率，提升神经网络结构的搜索效果。

根据第二方面，提供了一种神经网络结构的生成装置，包括：采样模块，用于对搜索空间内的网络结构集合进行采样，以得到多个候选网络结构，其中，所述候选网络结构包括多条边，各条所述边对应至少一种节点选取操作；训练模块，用于采用训练数据集分别对所述多个候选网络结构进行训练，并获取训练后的多个所述候选网络结构基于各种所述节点选取操作的目标检测效果；以及生成模块，用于根据所述目标检测效果，在所述搜索空间的所述网络结构集合之中进行搜索以生成目标神经网络结构。

本申请实施例的神经网络结构的生成装置，通过对搜索空间内的网络结构集合进行采样，以得到多个候选网络结构，其中，候选网络结构包括多条边，各条边对应至少一种节点选取操作，并采用训练数据集分别对多个候选网络结构进行训练，并获取训练后的多个候选网络结构基于各种节点选取操作的目标检测效果；以及根据目标检测效果，在搜索空间的网络结构集合之中进行搜索以生成目标神经网络结构，能够有效增强目标神经网络结构对目标检测的训练数据集的特征提取能力，提升神经网络结构的搜索效率，提升神经网络结构的搜索效果。

根据第三方面，提供了一种电子设备，包括：至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中，所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行本申请实施例的神经网络结构的生成方法。