NN2ViT: Neural Networks and Vision Transformers based approach for Visual Anomaly Detection in Industrial Images

被引:0
|
作者
Wahid, Junaid Abdul [1 ]
Ayoub, Muhammad [2 ]
Xu, Mingliang [1 ]
Jiang, Xiaoheng [1 ]
Shi, Lei [3 ]
Hussain, Shabir [4 ]
机构
[1] Zhengzhou Univ, Sch Comp Sci & Artificial Intelligence, Zhengzhou 450001, Peoples R China
[2] Shanghai Jiao Tong Univ, Sch Comp Sci & Engn, Shanghai, Peoples R China
[3] Zhengzhou Univ, Sch Cyber Space & Secur, Zhengzhou 450002, Peoples R China
[4] Tsinghua Univ, Inst Biopharmaceut & Hlth Engn, Tsinghua Shenzhen Int Grad Sch, Shenzhen, Peoples R China
关键词
Anomaly detection; Anomaly segmentation; Industrial data; Neural network; Vision transformers;
D O I
10.1016/j.neucom.2024.128845
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Ensuring product quality through automated anomaly detection is crucial in manufacturing. Traditional methods often struggle to capture both local and global features effectively, relying heavily on predefined templates that limit their adaptability and accuracy. To address these challenges, this study propose NN2ViT, a novel approach that integrates a Single Shot Detector (SSD) for local feature detection and the Segment Anything Model (SAM) for global feature segmentation. This integration allows fora comprehensive analysis of anomalies in industrial images. Our method improves anomaly segmentation performance by fine-tuning SAM for precise segmentation in industrial product images. Experiments on the MVTec benchmark dataset demonstrate that NN2ViT outperforms traditional models and achieved the highest 95.54% and 96.23% Image AUROC and AP scores, respectively thus enhancing interpretability and adaptability to various anomaly patterns. This research presents a significant advancement in manufacturing quality control, contributing to improved product quality and operational efficiency.
引用
收藏
页数:13
相关论文
共 22 条
  • [1] A Novel Anomaly Detection Approach based on Neural Networks
    Ergen, Tolga
    Kerpicci, Mine
    2018 26TH SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS CONFERENCE (SIU), 2018,
  • [2] Unsupervised Anomaly Detection of Industrial Images Based on Dual Generator Reconstruction Networks
    Gu, Cong
    Ren, Siyv
    Duan, Qiqiang
    INFORMATION TECHNOLOGY AND CONTROL, 2024, 53 (02): : 331 - 341
  • [3] Unsupervised Visual-to-Geometric Feature Reconstruction for Vision-Based Industrial Anomaly Detection
    Hoang, Dinh-Cuong
    Tan, Phan Xuan
    Nguyen, Anh-Nhat
    Tran, Duc-Thanh
    Duong, Van-Hiep
    Mai, Anh-Truong
    Pham, Duc-Long
    Phan, Khanh-Toan
    Do, Minh-Quang
    Duong, Ta Huu Anh
    Huynh, Tuan-Minh
    Bui, Son-Anh
    Nguyen, Duc-Manh
    Trinh, Viet-Anh
    Tran, Khanh-Duong
    Nguyen, Thu-Uyen
    IEEE ACCESS, 2025, 13 : 3667 - 3682
  • [4] Comparative analysis of vision transformers and convolutional neural networks in osteoporosis detection from X-ray images
    Sarmadi, Ali
    Razavi, Zahra Sadat
    van Wijnen, Andre J.
    Soltani, Madjid
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [5] Employing a hybrid model based on texture-biased convolutional neural networks and edge-biased vision transformers for anomaly detection of signal bonds
    Hoshi, Takuro
    Shibayama, Seiya
    Jiang, Xiaonan
    JOURNAL OF ELECTRONIC IMAGING, 2023, 32 (02)
  • [6] Optimizing Strawberry Disease and Quality Detection with Vision Transformers and Attention-Based Convolutional Neural Networks
    Aghamohammadesmaeilketabforoosh, Kimia
    Nikan, Soodeh
    Antonini, Giorgio
    Pearce, Joshua M.
    FOODS, 2024, 13 (12)
  • [7] Vision Transformers for Anomaly Detection and Localisation in Leather Surface Defect Classification Based on Low-Resolution Images and a Small Dataset
    Smith, Antony Douglas
    Du, Shengzhi
    Kurien, Anish
    APPLIED SCIENCES-BASEL, 2023, 13 (15):
  • [8] A Continuous Terminal Sliding-Mode Observer-Based Anomaly Detection Approach for Industrial Communication Networks
    Xu, Long
    Xiong, Wei
    Zhou, Minghao
    Chen, Lei
    SYMMETRY-BASEL, 2022, 14 (01):
  • [9] Human and bird detection and classification based on Doppler radar spectrograms and vision images using convolutional neural networks
    Gokaraju, Jnana Sai Abhishek Varma
    Song, Weon Keun
    Ka, Min-Ho
    Kaitwanidvilai, Somyot
    INTERNATIONAL JOURNAL OF ADVANCED ROBOTIC SYSTEMS, 2021, 18 (03)
  • [10] Transmission line detection in aerial images: An instance segmentation approach based on multitask neural networks
    Li, Bo
    Chen, Cheng
    Dong, Shiwen
    Qiao, Junfeng
    SIGNAL PROCESSING-IMAGE COMMUNICATION, 2021, 96 (96)