一种语义分割辅助的单教师-多学生无监督异常检测方法

SINGLE TEACHER AND MULTI STUDENT UNSUPERVISED ANOMALY DETECTION METHOD ASSISTED BY SEMANTIC SEGMENTATION

  • 摘要: 为了改善目前常见的异常检测算法受到的异常数据难以获取以及繁多的异常类别之间数据不均衡的制约,提出一种语义分割辅助的单教师-多学生无监督异常检测算法。在知识蒸馏模型中使用多个学生网络参与预测,根据多个学生网络的均值、方差等统计信息生成特征图,增强预测结果的稳定性与准确性;利用特征金字塔的思想,提取多个尺度的视觉特征以提升模型检测不同尺度异常区域的能力;利用语义分割模型良好的边界提取能力,引入语义分割网络对异常检测结果辅助修正。与现有的无监督异常检测算法相比,该方法能够更好地定位与检测工业场景下的物体表面异常。

     

    Abstract: In order to improve the restraint that caused by the difficulty of obtaining abnormal data and the imbalance of data among various abnormal categories for current common anomaly detection algorithms, this paper proposes a semantic segmentation-assisted single teacher and multi student unsupervised anomaly detection algorithm. Multiple student networks were used in the knowledge distillation model to participate in prediction together, and generated the feature maps through statistical information such as the mean and variance of multiple student networks to enhance the stability and accuracy of the prediction results. By using the idea of feature pyramids, the method extracted multiple scales of visual features to improve the model's ability to detect abnormal regions of different scales. By using the boundary extraction capabilities of the semantic segmentation model, the semantic segmentation network was introduced to assist in correcting the anomaly detection results. Compared with the existing unsupervised anomaly detection algorithms, the proposed method has better ability in locating and detecting object surface anomalies in industrial scenes.

     

/

返回文章
返回