一种基于卷积神经网络和自注意力机制的智能矿物分选方法技术

本发明专利技术公开了一种基于卷积神经网络和自注意力机制的智能矿物分选方法，实现了高精度、高效率、低体积、低计算复杂度的智能矿物分选。该架构通过结合卷积运算中局部性、平移不变性的归纳偏执和自注意力机制的全局性及长距离依赖性，建立了识别能力更强、特征捕获能力更优的矿物图像分类模型，相较于单一基于卷积神经网络或基于自注意力机制的网络，本发明专利技术提出的架构有着更高的分类精度和更低的计算复杂度。其次，本发明专利技术所提供的架构适用于多种类、多类别矿物图像的细粒度分类任务，对表观特征差异较小的矿物图像有着优异的判别能力。特征差异较小的矿物图像有着优异的判别能力。特征差异较小的矿物图像有着优异的判别能力。

【技术实现步骤摘要】
一种基于卷积神经网络和自注意力机制的智能矿物分选方法


[0001]本专利技术涉及一种基于卷积神经网络和自注意力机制的智能矿物分选方法，名为OreFormer，属于矿物分选和计算机视觉


技术介绍

[0002]矿产资源是现代工业生产的中流砥柱，对经济发展、基础建设、国防安全等作出了巨大贡献。合理、高效且可持续化的矿产资源分配是当前矿业领域发展的第一要务。随着发展速度的不断加快，矿产保有量呈明显下降趋势，且具有贫、细、杂、难选的矿产资源特质。同时，矿物开采过程中围岩量剧增也导致运输、破碎、磨矿能耗居高不下，因此亟须在开采后和洗选前进行矿物快速识别与分选，以达到预先排矸和降本增效的目的。
[0003]近年来，基于射线传感器的智能矿物分选技术备受关注，可实现两产品(矿石排矸)乃至多产品的高效率、批量化分选。同时，随着传感器设备的快速更新，基于γ射线,罗曼光谱分析、X射线衍射(XRD)分析和X射线荧光(XRF)分析的射线分选设备在智能矿物分选领域的应用前景日益广阔，具有精度高、操作简单、用户友好等优点，且在大块矿物的分选任务中效果突出。然本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于卷积神经网络和自注意力机制的智能矿物分选方法，其特征在于包括以下步骤：步骤1：获取待处理的矿物图像数据，并将获取的矿物图像数据划分为训练集和验证集；步骤2：对步骤1获得的矿物图像数据进行标注，获取矿物图像数据的标签；步骤3：根据步骤1获得的矿物图像数据和步骤2获取的矿物图像数据标签，构建基于卷积神经网络和自注意力机制的矿物图像分类模型；步骤4：使用步骤1中划分的训练集和验证集对步骤3构建的基于卷积神经网络和自注意力机制的矿物图像分类模型进行训练；步骤5：终止所述矿物图像分类模型的训练过程，得到所述训练后的基于卷积神经网络和自注意力机制的矿物图像分类模型；步骤6:通过步骤5获得的基于卷积神经网络和自注意力机制的矿物图像分类模型对新采集的矿物图像数据进行预测，完成矿物分选。2.根据权利要求1所述的一种基于卷积神经网络和自注意力机制的智能矿物分选方法，其特征在于：所述的步骤1中，获取矿物图像数据时，搭建矿物图像数据采集平台；所述的矿物图像数据是在工业相机在传送带正上方拍摄的，按照比例8:2划分训练集和验证集。3.根据权利要求1或2所述的一种基于卷积神经网络和自注意力机制的智能矿物分选方法，其特征在于：所述步骤3中，基于卷积神经网络和自注意力机制的矿物图像分类模型包括卷积颈、下采样层、自注意力编码器、均值池化层、投影层和分类器。4.根据权利要求3所述的一种基于卷积神经网络和自注意力机制的智能矿物分选方法，其特征在于：所述卷积颈由3个卷积层构成，负责引入卷积运算的归纳偏执，其中一个卷积层的步长为2、卷积核大小为3
×
3，另外两个卷积层的步长为1、卷积核大小为3
×
3；所述下采样层由1个卷积层构成，负责降低特征图像维度并生成用于自注意力编码器特征运算的特征数据，其中卷积层的步长为2、卷积核大小为2
×
2；所述自注意力编码器由1个卷积层、2个层归一化层、1个卷积多头自注意力模块、1个倒置瓶颈前馈神经网络和3条残差连接构成，负责计算输入图像中各区域的注意力权重；所述均值池化层对输入特征值进行平均池化运算，负责汇聚重要的矿物图像特征信息；所述投影层由1个卷积层构成，负责整合前序网络所提取到的矿物图像特征信息，并将输出传输给分类器，其中卷积层的步长为1、卷积核大小为1
×
1；所述分类器由1个全连接层构成，负责建立特征信息与类别信息之间的关系。5.根据权利要求4所述的一种基于卷积神经网络和自注意力机制的智能矿物分选方法，其特征在于：所述自注意力编码器中的卷积多头自注意力模块包含1个深度卷积层、3个线性投影层和1个多头自注意力层；卷积多头自注意力模块负责计算输入图像中各区域的注意力权重；所述自注意力编码器...

【专利技术属性】
技术研发人员：邓方，刘洋，王雪旖，陈杰，
申请(专利权)人：北京理工大学，
类型：发明
国别省市：