12月8日,国际仪器与测量协会著名期刊《IEEE Transactions on Instrumentation and Measurement,IEEE TIM》(影响因子:5.6)在线刊发了我院国家数控系统工程技术研究中心副教授余文勇课题组在工业表面缺陷智能检测领域的新成果,题为“A-Net: An A-shape Lightweight Neural Network for Real-time Surface Defect Segmentation(A-Net: 一种用于实时表面缺陷分割的A形轻量化神经网络)”。我院2020级本科生陈彪为第一作者,2018级博士生牛通之为共同一作。余文勇副教授为通讯作者,李斌教授为学业导师。
机器学习已成为工业表面缺陷检测的研究热点,但由于其日益增加的参数量和计算消耗,机器学习在工业现场应用的硬件成本日益增加。课题组设计了一种轻量级网络架构A-NET,通过编码器-解码器结构,提取低层次细节特征和高层次语义特征,将这些特征通过一个跳跃连接进行融合,使网络架构呈“A”形,这种方式不仅保留了多尺度特征提取和融合特性,而且最大限度地减少了由多个跳跃连接引起的内存使用, 由此降低了网络参数量和计算复杂度。
通过多个标准测试集的评估,与同行的U-Net方法相比,A-Net在NEU-seg、DAGM-seg、MCSD-seg和MT数据集上分别获得了最佳性能-轻量化权衡。其网络参数量仅为0.39M,减少了98.8%,计算负载减少了99%。由于A-Net具有较低的FLOPs,因此在没有GPU的边缘设备上显示出极快的推理速度。通过A-Net开发的缺陷分割算法,适用于资源有限的工业现场应用。
无GPU加速下的高效语义分割网络
同年11月,陈彪还以共同第一作者身份在第49届IEEE工业电子学会年会(IECON 2023)上发表论文,题为“Background-Adaptive Surface Defect Detection Neural Networks via Positive Samples(基于正样本的背景自适应表面缺陷检测神经网络,BANet)”。BANet利用孪生网络同时提取正样本和负样本的特征,再将二者相同特征对齐并做差,解决了背景纹理变化、平移和旋转噪声的问题。该方法在光通讯器件缺陷数据集上取得最佳分割性能。
背景自适应表面缺陷检测网络
陈彪是我院2020级机械本硕博班本科生,大一期间就主动进入课题组参与科研工作,经严格的学术培训逐步成长起来,是我校优秀本科生的代表。
上述研究工作得到了国家自然科学基金(No. 52375494,51775214)的支持。
论文链接:
https://ieeexplore.ieee.org/document/10352342
https://ieeexplore.ieee.org/document/10312239