<p>We propose a novel multi-scale attention mechanism (MSAM) that specifically addresses the challenges of environmental noise in industrial defect segmentation. The MSAM is composed of two submodules, namely the multi-scale feature filtering (MSFF) submodule and the scale selection (SS) submodule. The MSFF generates spatial attention weights that provide precise spatial location guidance for multi-scale features. This enhances the regions of interest while suppressing irrelevant regions. Additionally, the SS adaptively models the interdependencies between the channels of multi-scale features and assigns different weights to different channels. Experiments on the three industrial datasets demonstrate that the application of the MSAM module can effectively tackle the influence of complex noise in industrial segmentation and significantly improve the accuracy of industrial defect segmentation.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

MSAM: a multi-scale attention mechanism for improving industrial defect segmentation

  • Menghao Han,
  • Zhenyan Ji,
  • Yanyan Yang,
  • Qibo Feng,
  • Shen Yin

摘要

We propose a novel multi-scale attention mechanism (MSAM) that specifically addresses the challenges of environmental noise in industrial defect segmentation. The MSAM is composed of two submodules, namely the multi-scale feature filtering (MSFF) submodule and the scale selection (SS) submodule. The MSFF generates spatial attention weights that provide precise spatial location guidance for multi-scale features. This enhances the regions of interest while suppressing irrelevant regions. Additionally, the SS adaptively models the interdependencies between the channels of multi-scale features and assigns different weights to different channels. Experiments on the three industrial datasets demonstrate that the application of the MSAM module can effectively tackle the influence of complex noise in industrial segmentation and significantly improve the accuracy of industrial defect segmentation.