基金项目:国家自然科学基金(61462072,61662062)收稿日期:2021-05-11修回日期:2021-06-05第40卷第4期计算机仿真2023年4月文章编号:1006-9348(2023)04-0362-05融合掩模和注意机制的CNN图像分类算法李伟1,2,黄鹤鸣1,2(1.青海师范大学计算机学院,青海西宁810008;2.藏语智能信息处理与应用国家重点实验室,青海西宁810008)摘要:为了更好地抑制卷积神经网络在图像分类训练过程中出现的过拟合现象,提高图像分类精度和算法的稳定性,提出了融合掩模和注意机制的CNN图像分类算法。算法使用去噪神经网络提取的去噪深度特征,作为深度CNN网络模型总框架的输入,基于掩模预训练方法训练网络模型;再使用不同注意类型的激活函数,在标准图像库Corel-1000上充分进行实验。实验结果表明:使用基于去噪神经网络的掩模预训练方法,不仅较大程度提升了图像分类效果,而且整个模型的性能更趋于稳定和健壮。加入混合注意机制后,实验效果得到进一步提升。算法通过降低图像背景噪声和深度特征噪声的干扰,能有效抑制训练过程中出现的过拟合现象;且不同注意类型激活函数的作用不尽相同,其中混合注意机制的实验效果最优。关键词:卷积神经网络;过拟合;掩模操作;注意机制;预训练;去噪深度特征中图分类号:TP183文献标识码:BCNNImageClassificationAlgorithmBasedonFusionofMaskandAttentionMechanismLIWei1,2,HUANGHe-ming1,2(1.SchoolofComputerScienceandTechnology,QinghaiNormalUniversity,XiningQinghai810008,China;2.StateKeyLaboratoryofTibetanIntelligentInformationProcessingandApplication,XiningQinghai810008,China)ABSTRACT:Inordertosuppresstheover-fittingphenomenonofconvolutionneuralnetworkinthetrainingprocessofimageclassificationandimprovetheaccuracyofimageclassificationandthestabilityofalgorithm,aCNNimageclassificationalgorithmbasedonfusionofmaskandattentionmechanismisproposed.Thealgorithmwasusedtotrainthenetworkmodelbasedonmaskpre-trainingmethod,andthedenoisingneuralnetworkwasusedtoextractthede-noisingdeepfeaturesastheinputoftheoverallframeworkofdeepCNNnetworkmodel.Then,experimentswerecar-riedoutonthestandardimagedatabaseCorel-1000usingdifferentattentionactivationfunctions.Theexperimentalre-sultsshowthatusingthepre-trainingmethodbasedonmasknotonlyimprovestheimageclassificationeffect,butal-somakesthepe...
