Feature Aggregation Network for Memory-Based Industrial Anomaly Detection
Min Li, Delong Han, Rui Shao, Jinghui He · 2024
In the process of large-scale industrial production, complex defects of different shapes and sizes can occur in industrial products and their components, and these defects seriously affect the performance of industrial products and their components, and being able to quickly and accurately identify defective industrial products and their components is an important task. Memory-based methods are usually used for anomaly detection, and these methods often rely on pre-trained models for feature extraction, but have high requirements on feature quality. In addition, patch-based methods mainly focus on local differences in feature embedding when detecting anomalies without explicitly exploiting broader contextual information, thus greatly affecting the robustness and accuracy of anomaly detection. Aiming at the above problems and requirements, this paper proposes a Feature Aggregation Network for Memory-Based Industrial Anomaly Detection (FANet), which is based on the pre-extraction of anomalous features, attentional attention to key features, information aggregation of multi-level and multi-scale features, and the FANet provides advanced performance in industrial image anomaly detection and localisation by pre-extraction of anomalous features, attentional attention to key features, information aggregation of multi-layered and multi-scale features, and introduction of environmental information in the anomaly score computation to perform industrial anomaly detection tasks, and using core set subsampling to reduce storage requirements. On the MVTec AD, FANet has an image-level AUROC score of 99.1% and a pixel-level AUROC score of 98.2%. Similarly high detection results were achieved on the ViSA dataset, further demonstrating the robustness of FANet.