【封面文章】大连理工大学吴玉虎等:基于轻量化卷积神经网络的金属断口图像识别

文摘   科技   2024-09-19 15:46   辽宁  




点击蓝字 关注我们 







本文引用信息

闫涵, 卢伟, 吴玉虎. 基于轻量化卷积神经网络的金属断口图像识别[J]. 控制与决策, 2024, 39(9): 2858-2866.


2024年第9期封面文章推荐

基于轻量化卷积神经网络的金属断口图像识别

闫涵,卢伟,吴玉虎


本文创新点:

本文提出了一种基于轻量化卷积神经网络的金属断口图像识别方法,其主要创新如下:

1)针对特征复杂易错分的问题,使用基于迁移学习的VGGNet-16ResNet-50作为特征提取基模型,通过设计自适应加权特征拼接方式增强特征提取能力;

2)对模型冗余效率低的问题,设计了一种混合剪枝算法,对模型进行高效压缩,降低参数量;

3)针对模型部署问题,给出了两种基于嵌入式系统的金属断口图像识别模型部署方案,以适应不同现场需求。

01

研究背景

金属材料是航空航天、交通运输、冶金制造等领域的重大工程、大型设备主要用材。在复杂环境作用下,金属材料会发生腐蚀、疲劳、断裂等各种模式的失效事故,进而造成重大经济损失与人员伤亡。因此,对金属断口的高效智能识别成为保障金属材料生产及安全使用的关键环节。
目前,绝大多数金属断口识别依赖人工经验完成,存在如下问题:1) 一些金属生产及使用环境较为恶劣,难以实现人工实时检测;2) 依赖人工经验易存在主观误判、检测效率较低等问题。近年来,随着机器学习技术在计算机视觉领域的发展,基于机器视觉的金属断口图像识别方法取得了重大突破。然而,在实际工业应用场景下的金属断口图像识别仍存在如下挑战。

首先是特征提取问题,现有的金属断口图像识别方法大多基于传统的机器学习方式,需人工设计特征提取方法,导致算法性能有限且泛化能力不足。其次是模型冗余问题,尽管CNN模型在特征提取性能上优于传统机器学习方法,但由于模型结构复杂,导致模型冗余难以在工业环境下应用。最后是模型部署问题,上述方法研究大多在高性能计算机上进行仿真实验,缺乏针对工业应用环境的部署研究。

02

方法介绍

针对上述问题,本文提出方法的总体方案如下:

首先,根据目前图像识别任务中的深度学习模型性能选择特征提取基模型。考虑到金属断口图像的数据量,对基模型进行预训练,通过迁移学习获取低阶特征提取器。结合集成学习思想,为各个基模型分配权重,以提升特征提取性能。然后,设计混合剪枝算法对模型的通道及卷积核进行修剪,降低模型复杂度。最后,针对工业背景下不同使用场景的需求,给出两种基于嵌入式系统Jetson Nano b01的模型部署方案,分别为模型离线部署方式与模型在线部署方式。

03

主要结论

目前工业环境下的金属断口图像智能识别大多基于传统机器学习方法,识别精度不高。少数基于深度学习的方法,因模型参数量及运算量过大,无法应用于真实工业场景中。为了提高工业场景下金属断口的识别性能,本文给出了一种基于轻量化CNN的金属断口图像识别方法。该方法以VGGNetResNet作为特征提取基模型;设计自适应加权特征拼接方法增强特征提取能力;利用混合剪枝算法降低模型的参数量;并通过BO模块对模型的重要参数进行自动搜索。通过在金属断口图像数据集上的实验验证了算法的有效性。同时,通过两种不同的部署方式将该模型部署到嵌入式系统中,来验证算法实际应用的可行性。


作者介绍


闫涵,大连理工大学控制科学与工程学院,博士生,从事深度学习、图像处理与模式识别等研究。

卢伟,大连理工大学控制科学与工程学院,教授,博士生导师,从事粒计算、计算智能、知识发现和表示等研究。

吴玉虎,大连理工大学控制科学与工程学院,教授,博士生导师,从事非线性系统、博弈决策和控制理论方法等研究。


相关文章推荐






精彩论文】基于脉冲卷积神经网络稀疏表征的高分辨率遥感图像场景分类方法

【精彩论文】基于深度稀疏低秩分解的深度神经网络轻量化方法


郑重声明:根据国家版权局相关规定,纸媒、网站、微博、微信公众号转载、摘编本微信信息,需包含本微信名称、二维码等相关信息,并在文首注明《控制与决策》原创。个人请按本微信原文转发、分享。


-联系我们-

《控制与决策》

网站:http://kzyjc.alljournals.cn

电话:024-83687766

邮箱:kzyjc@mail.neu.edu.cn

《Journal of Control and Decision》

网站:

https://www.tandfonline.com/toc/tjcd20/current

邮箱:tengrong@ise.neu.edu.cn


点击“阅读原文”查看全文信息

控制与决策
该服务号旨在发布期刊、网刊动态和征文通知等内容
 最新文章