Vision Transformers for Anomaly Detection and Localisation in Leather Surface Defect Classification Based on Low-Resolution Images and a Small Dataset
Source abstract
Abstract
Genuine leather manufacturing is a multibillion-dollar industry that processes animal hides from varying types of animals such as sheep, alligator, goat, ostrich, crocodile, and cow. Due to the industry’s immense scale, there may be numerous unavoidable causes of damages, leading to surface defects that occur during both the manufacturing process and the bovine’s own lifespan. Owing to the heterogenous and manifold nature of leather surface characteristics, great difficulties can arise during the visual inspection of raw materials by human inspectors. To mitigate the industry’s challenges in the quality control process, this paper proposes the application of a modern vision transformer (ViT) architecture for the purposes of low-resolution image-based anomaly detection for defect localisation as a means of leather surface defect classification. Utilising the low-resolution defective and non-defective images found in the opensource Leather Defect detection and Classification dataset and higher-resolution MVTec AD anomaly benchmarking dataset, three configurations of the vision transformer and three deep learning (DL) knowledge transfer methods are compared in terms of performance metrics as well as in leather defect classification and anomaly localisation. Experiments show the proposed ViT method outperforms the light-weight state-of-the-art methods in the field in the aspect of classification accuracy. Besides the classification, the low computation load and low requirements for image resolution and size of training samples are also advantages of the proposed method.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Industrial Vision Systems and Defect Detection · Thermography and Photoacoustic Techniques · Identification and Quantification in Food
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Electrical Engineering, Faculty of Engineering and the Built Environment, Pretoria 0184, South Africa
References (50)
Show source-provided references
1. Fortune Business Insights (2023). Market Research Report, Fortune Business Insights. FBI104405.
2. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2020). An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale. arXiv.
3. ImageNet classification with deep convolutional neural networks10.1145/3065386 ↗
4. Sermanet, P., Eigen, D., Zhang, X., Mathieu, M., Fergus, R., and LeCun, Y. (2013). OverFeat: Integrated Recognition, Localization and Detection using Convolutional Networks. arXiv.
5. A Comprehensive Review of Image Segmentation Techniques10.37917/ijeee.17.2.18 ↗
6. Vaswani, A., Shazeer, N.M., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, L., and Polosukhin, I. (2017). Attention is All you Need. arXiv.
7. Kendall, E.J., Barnett, M.G., and Chytyk-Praznik, K. (2013). Automatic detection of anomalies in screening mammograms. BMC Med. Imaging, 13.10.1186/1471-2342-13-43 ↗
8. Gyimah, N.K., Girma, A., Mahmoud, M.N., Nateghi, S., Homaifar, A., and Opoku, D. (2021, January 17–20). A Robust Completed Local Binary Pattern (RCLBP) for Surface Defect Detection. Proceedings of the 2021 IEEE International Conference on Systems, Man, and Cybernetics (SMC), Melbourne, Australia.
9. Road Crack Detection Using Visual Features Extracted by Gabor Filters10.1111/mice.12042 ↗
10. Vaideliene, G., and Valantinas, J. (2016, January 27–29). Wavelet-based Defect Detection System for Grey-level Texture Images. Proceedings of the International Conference on Computer Vision Theory and Applications, Rome, Italy.10.5220/0005678901430149 ↗
11. Detection and classification of defect patterns on semiconductor wafers10.1080/07408170600733236 ↗
12. Automated Visual Defect Detection for Flat Steel Surface: A Survey10.1109/TIM.2019.2963555 ↗
13. Fabric Defect Detection in Textile Manufacturing: A Survey of the State of the Art
14. Simonyan, K., and Zisserman, A. (2015). Very Deep Convolutional Networks for Large-Scale Image Recognition. arXiv.
15. Chollet, F. (2017, January 21–26). Xception: Deep Learning with Depthwise Separable Convolutions. Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA.10.1109/CVPR.2017.195 ↗
16. A machine learning method for defect detection and visualization in selective laser sintering based on convolutional neural networks
17. Tabernik, D., Sela, S., Skvarc, J., and Skočaj, D. (2019, January 23–25). Deep-Learning-Based Computer Vision System for Surface-Defect Detection. Proceedings of the International Conference on Virtual Storytelling, Thessaloniki, Greece.10.1007/978-3-030-34995-0_44 ↗
18. Han, Y., and Yu, H. (2020). Fabric Defect Detection System Using Stacked Convolutional Denoising Auto-Encoders Trained with Synthetic Defect Data. Appl. Sci., 10.10.3390/app10072511 ↗
19. Automated Product Boundary Defect Detection Based on Image Moment Feature Anomaly10.1109/ACCESS.2019.2911358 ↗
20. Minhas, M.S., and Zelek, J.S. (2019). Anomaly Detection in Images. arXiv.
21. Wang, L., Zhang, D., Guo, J., and Han, Y. (2020). Image Anomaly Detection Using Normal Data Only by Latent Space Resampling. Appl. Sci., 10.10.3390/app10238660 ↗
22. DDR-ID: Dual deep reconstruction networks-based image decomposition for anomaly detection10.1007/s12652-021-03425-0 ↗
23. Detection and segmentation of image anomalies based on unsupervised defect reparation10.1007/s00371-021-02257-5 ↗
24. Beggel, L., Pfeiffer, M., and Bischl, B. (2019). Robust Anomaly Detection in Images using Adversarial Autoencoders. arXiv.10.1007/978-3-030-46150-8_13 ↗
25. Modified DenseNet for Automatic Fabric Defect Detection With Edge Computing for Minimizing Latency10.1109/JIOT.2020.2983050 ↗
26. Dafu, Y. (2019, January 22–24). Classification of Fabric Defects Based on Deep Adaptive Transfer Learning. Proceedings of the 2019 Chinese Automation Congress (CAC), Hangzhou, China.10.1109/CAC48633.2019.8996969 ↗
27. Pistori, H., Amorim, W.P., Martins, P.S., Pereira, M.C., Pereira, M.M., and Jacinto, M.A. (2006). Computational Modeling of Objects Represented in Images, CRC Press.
28. Deep learning and machine learning neural network approaches for multi class leather texture defect classification and segmentation10.1186/s42825-022-00080-9 ↗
29. Moganam, P.K., and Sathia Seelan, D.A. (2022, August 12). Leather Defect Detection and Classification. Available online: https://www.kaggle.com/datasets/praveen2084/leather-defect-classification.
30. Perceptron neural network-based machine learning approaches for leather defect detection and classification10.18280/i2m.190603 ↗
31. An improved automatic defect identification system on natural leather via generative adversarial network10.1080/0951192X.2022.2048421 ↗
32. Surface Defect Detection of Wet-Blue Leather Using Hyperspectral Imaging10.1109/ACCESS.2021.3112133 ↗
33. Parmar, N., Vaswani, A., Uszkoreit, J., Kaiser, L., Shazeer, N.M., Ku, A., and Tran, D. (2018, January 10–15). Image Transformer. Proceedings of the International Conference on Machine Learning, Stockholm, Sweden.
34. All Tokens Matter: Token Labeling for Training Better Vision Transformers
35. Wang, Y., Huang, R., Song, S., Huang, Z., and Huang, G. (2021). Not All Images are Worth 16x16 Words: Dynamic Vision Transformers with Adaptive Sequence Length. arXiv.
36. Mishra, P., Verk, R., Fornasier, D., Piciarelli, C., and Foresti, G.L. (2021, January 20–23). VT-ADL: A Vision Transformer Network for Image Anomaly Detection and Localization. Proceedings of the 2021 IEEE 30th International Symposium on Industrial Electronics (ISIE), Kyoto, Japan.10.1109/ISIE45552.2021.9576231 ↗
37. Xu, Y., Zhang, Q., Zhang, J., and Tao, D. (2021). ViTAE: Vision Transformer Advanced by Exploring Intrinsic Inductive Bias. arXiv.
38. Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., Lin, S., and Guo, B. (2021, January 10–17). Swin Transformer: Hierarchical Vision Transformer using Shifted Windows. Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision (ICCV), Montreal, QC, Canada.10.1109/ICCV48922.2021.00986 ↗
39. Fang, Y., Yang, S., Wang, S., Ge, Y., Shan, Y., and Wang, X. (2022). Unleashing Vanilla Vision Transformer with Masked Image Modeling for Object Detection. arXiv.
40. Deng, J., Dong, W., Socher, R., Li, L., Li, K., and Fei-Fei, L. (2009, January 20–25). ImageNet: A large-scale hierarchical image database. Proceedings of the 2009 IEEE Conference on Computer Vision and Pattern Recognition, Miami, FL, USA.10.1109/CVPR.2009.5206848 ↗
41. He, K., Zhang, X., Ren, S., and Sun, J. (2016, January 27–30). Deep Residual Learning for Image Recognition. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.90 ↗
42. Wang, S., Xia, X., Ye, L., and Yang, B. (2021). Automatic Detection and Classification of Steel Surface Defect Using Deep Convolutional Neural Networks. Metals, 11.10.3390/met11030388 ↗
43. Szegedy, C., Vanhoucke, V., Ioffe, S., Shlens, J., and Wojna, Z. (2016, January 27–30). Rethinking the Inception Architecture for Computer Vision. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.308 ↗
44. Google Cloud Developers (2023, January 30). Advanced Guide to Inception V3. Available online: https://cloud.google.com/tpu/docs/inception-v3-advanced.
45. Tan, M., and Le, Q.V. (2019). EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks. arXiv.
46. Bergmann, p., Fauser, M., Sattlegger, D., and Steger, C. (2019, January 15–20). A Comprehensive Real-World Dataset for Unsupervised Anomaly Detection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA.10.1109/CVPR.2019.00982 ↗
47. Honda, H. (2023, March 23). Vision Transformer Pipeline (Image). Available online: https://github.com/hirotomusiker/schwert_colab_data_storage/blob/master/images/vit_demo/vit_input.png.
48. Multilayer perceptron and neural networks
49. Grosse, R. (2019). Lecture Notes for the Course CSC321, “Intro to Neural Networks,” for Undergraduates at the University of Toronto, University of Toronto.
50. AnoViT: Unsupervised Anomaly Detection and Localization with Vision Transformer-based Encoder-Decoder10.1109/ACCESS.2022.3171559 ↗