Hyperbolic geometry enhanced feature filtering network for industrial anomaly detection.

Journal: Scientific reports
Published Date:

Abstract

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.

Authors

  • Yanjun Feng
    Shanghai Frontiers Science Center of Optogenetic Techniques for Cell Metabolism, Shanghai Key Laboratory of New Drug Design, School of Pharmacy, East China University of Science and Technology, Shanghai, China.
  • Jun Liu
    Department of Radiology, Second Xiangya Hospital, Changsha, Hunan, China.
  • Yonggang Gai
    School of Automation and Electrical Engineering, Shenyang Ligong University, Shenyang, 110159, China.

Keywords

No keywords available for this article.