文献类型：期刊文章

作　　者：叶仪铭[1] 陈锐[2] 王仁波[3] 刘凡[4]

YE Yiming;CHEN Rui;WANG Renbo;LIU Fan(School of Information Engineering,East China University of Technology,Nanchang 330013,China;Jiangzi Province Engineering Research Center of New Energy Technology and Equipment,Nanchang 330013,China;Institute of Nuclear Application Technology,East China University of Technology,Nanchang 330013,China;School of Mechanical and Electronic Engineering,East China University of Technology Nanchang 330013,China)

机构地区：[1]东华理工大学信息工程学院,南昌330013 [2]东华理工大学江西省新能源工艺及装备工程技术研究中心,南昌330013 [3]东华理工大学核应用技术研究所,南昌330013 [4]东华理工大学机械与电子工程学院,南昌330013

出　　处：《有色金属（选矿部分）》

基　　金：国家自然科学基金资助项目(U22B2077);江西省重大科技研发专项项目(20224AAC01012)。

年　　份：2023

期　　号：6

起止页码：118-124

语　　种：中文

收录情况：BDHX、BDHX2020、CAS、JST、ZGKJHX、核心刊

摘　　要：铀资源是我国重要的战略资源,在工农业生产、科学技术以及国防领域中有广泛的用途,但我国已探明铀矿床主要以中低品位为主,矿体形态复杂,不连续分散分布,开采成本高,利用率低。为了识别可冶炼铀矿石,提高资源利用率,更好地支持国家的铀矿资源储备,采用X射线透射成像技术,并以收集到的铀矿石透射图像为研究对象,提出了一种名为SAxVit的轻量化高性能铀矿识别算法。该算法采用了ShuffleNet的思想,设计了一个轻量级的主干网络用于初步的特征提取。其次,提出了一种轴向平均注意力机制来进行深度特征提取,该机制通过计算特征图横轴纵轴上的平均值获取特征分布趋势矩阵,然后加权该矩阵计算特征图上每个像素点与其他像素点之间的关联性,以符合透射图像根据不同灰度值区分元素的工作原理。最后,基于Vision Transformer模型,剔除特征图分割操作,设计了一种即插即用的轻量化的ViT模块用作轴向平均注意力机制的载体。试验结果表明,SAxVit算法的参数量低至0.077M,识别速度低至3.481 ms,在铀矿透射图像测试集上该算法的识别准确率达到了95.7%。相比于MobileNetV2、ShuffleNetV2_0.5、Vit_64x64以及MobileVit_small,SAxVit在识别准确率、轻量化和识别速度等方面都取得了显著的改进。

关 键 词：深度学习  X射线矿石图像识别  注意力机制  轻量化网络  

分 类 号：TD679] TP312]