Hyperbolic geometry enhanced feature filtering network for industrial anomaly detection.

In recent years, Cutting-edge machine learning algorithms and systems in Industry 4.0 enhance quality control and increase production efficiency. The visual perception algorithms have become extensively utilized in surface defect detection, progressively replacing manual inspection methods. As a crucial component of the Industrial Internet of Things (IIoT), this technology is pivotal for ensuring industrial production quality and has garnered significant attention from the military and aerospace sectors. Nonetheless, most existing methods rely on Euclidean space, which constrains their effectiveness in handling non-Euclidean space data. Additionally, challenges such as addressing pre-trained feature redundancy and bias in the pre-training process persist. This paper presents HADNet, a hyperbolic space-based anomaly detection method. Specifically, we begin by mapping the extracted features to hyperbolic space, a non-Euclidean geometric space. This mapping leverages the unique geometric properties of hyperbolic space, particularly the hyperbolic distance metric, to represent the distances between features more effectively. Next, the most relevant features for anomaly detection are selected through the anomaly-aware feature subset selection module, enhancing anomaly detection performance. Finally, we introduce adaptive residuals discrimination, an adaptive analysis technique that discards residuals lacking anomaly information, thereby isolating the most effective regions for anomaly detection. Extensive experiments on four benchmark datasets NEU-Seg, MT-Defect, FSSD-12, and UCF-EL demonstrate the efficacy of HADNet, achieving mIoU scores of 87%, 81.46%, 77.04%, and 59.41% respectively, significantly surpassing the current state-of-the-art methods.