面向滚子表面缺陷的轻量化多视角智能分类模型
Lightweight multi-view intelligent classification model for roller surface defects
-
摘要: 为提升圆柱滚子表面缺陷的自动化识别精度与检测效率,提出1种轻量化多视角特征驱动分类模型MFDANet。该模型通过多视角特征增强模块(MVFA)融合原始纹理、边缘结构与增强信息,实现细微缺陷的多层次特征提取;采用哈尔小波下采样(HWD)策略,在降低计算量的同时有效保留关键细节;设计高效多维注意力机制(ECSA),自适应聚焦缺陷区域特征;并结合空洞卷积多尺度特征提取模块(DMFE),实现局部与全局特征的协同建模。试验结果表明,MFDANet在圆柱滚子表面缺陷分类任务中达到87.70%的准确率和87.33%的宏平均F1值,参数量仅4.41 MB、计算量0.026 GFlops。在显著降低模型复杂度的同时,该模型较传统分类模型、轻量级模型及近期先进方法均取得更优性能,验证了其高效性与实用性。Abstract: To enhance the automation accuracy and efficiency of cylindrical roller surface defect recognition, a lightweight multi-view feature-driven classification model termed MFDANet is proposed. The model integrates a multi-view feature augmentation module to fuse raw texture, edge structure, and enhanced information for hierarchical feature extraction of subtle defects. A Haar wavelet downsampling strategy is adopted to reduce computational cost while preserving critical details. Moreover, an efficient channel-spatial attention mechanism is designed to adaptively emphasize defect-related features, and a dilated multi-scale feature extraction module is introduced to achieve collaborative modeling of local and global representations. Experimental results demonstrate that MFDANet achieves an accuracy of 87.70% and a macro-averaged F1-score of 87.33% on the cylindrical roller surface defect classification task. With only 4.41 MB of parameters and 0.026 GFlops of computation, the proposed model significantly reduces complexity while outperforming traditional classifiers, lightweight networks and recent advanced approaches, thereby verifying its efficiency and practicality.
下载: