基于掩码自编码器的无监督钢铁表面缺陷识别方法
钢铁表面缺陷的自动化视觉检测是一项重要的质量控制任务,缺陷标注数据稀缺且获取成本高,而无标注表面图像丰富,促使采用自监督方法在无类别标签情况下学习有用表示。本文使用基于Transformer的掩码自编码器( MAE),在预训练阶段对输入图像75%的块随机掩码,由轻量解码器从剩余25%可见块重建掩盖区域。编码器联合辅助缺陷定位目标训练,辅助定位仅作为训练信号。解码器达到结构相似度0.92和均方误差0.47。预训练编码器提取特征后,结合UMAP降维和聚合聚类,实现与六类已知缺陷的匈牙利匹配精度91.3%。
这条 研究论文 信号说明,来自 arXiv cs.CV 的信息已经不只是单点新闻,而是值得放进产品、研究和行业判断里的趋势线索。