Adjacent Image Augmentation and Its Framework for Self-Supervised Learning in Anomaly Detection

被引:0
|
作者
Kwon, Gi Seung [1 ]
Choi, Yong Suk [1 ]
机构
[1] Hanyang Univ, Dept Comp Sci, Seoul 04763, South Korea
基金
新加坡国家研究基金会;
关键词
anomaly detection; automatic optical inspection; representation learning;
D O I
10.3390/s24175616
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Anomaly detection has gained significant attention with the advancements in deep neural networks. Effective training requires both normal and anomalous data, but this often leads to a class imbalance, as anomalous data is scarce. Traditional augmentation methods struggle to maintain the correlation between anomalous patterns and their surroundings. To address this, we propose an adjacent augmentation technique that generates synthetic anomaly images, preserving object shapes while distorting contours to enhance correlation. Experimental results show that adjacent augmentation captures high-quality anomaly features, achieving superior AU-ROC and AU-PR scores compared to existing methods. Additionally, our technique produces synthetic normal images, aiding in learning detailed normal data features and reducing sensitivity to minor variations. Our framework considers all training images within a batch as positive pairs, pairing them with synthetic normal images as positive pairs and with synthetic anomaly images as negative pairs. This compensates for the lack of anomalous features and effectively distinguishes between normal and anomalous features, mitigating class imbalance. Using the ResNet50 network, our model achieved perfect AU-ROC and AU-PR scores of 100% in the bottle category of the MVTec-AD dataset. We are also investigating the relationship between anomalous pattern size and detection performance.
引用
收藏
页数:22
相关论文
共 50 条
  • [21] Pavement anomaly detection based on transformer and self-supervised learning
    Lin, Zijie
    Wang, Hui
    Li, Shenglin
    [J]. AUTOMATION IN CONSTRUCTION, 2022, 143
  • [22] Understanding the limitations of self-supervised learning for tabular anomaly detection
    Mai, Kimberly T.
    Davies, Toby
    Griffin, Lewis D.
    [J]. PATTERN ANALYSIS AND APPLICATIONS, 2024, 27 (02)
  • [23] Anomaly classification based on self-supervised learning and its application
    Han, Yongsheng
    Qi, Zhiquan
    Tian, Yingjie
    [J]. JOURNAL OF RADIATION RESEARCH AND APPLIED SCIENCES, 2024, 17 (03)
  • [24] A Hyperspectral Image Change Detection Framework With Self-Supervised Contrastive Learning Pretrained Model
    Ou, Xianfeng
    Liu, Liangzhen
    Tan, Shulun
    Zhang, Guoyun
    Li, Wujing
    Tu, Bing
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 7724 - 7740
  • [25] A Hyperspectral Image Change Detection Framework With Self-Supervised Contrastive Learning Pretrained Model
    Ou, Xianfeng
    Liu, Liangzhen
    Tan, Shulun
    Zhang, Guoyun
    Li, Wujing
    Tu, Bing
    [J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2022, 15 : 7724 - 7740
  • [26] Image classification framework based on contrastive self-supervised learning
    Zhao, Hong-Wei
    Zhang, Jian-Rong
    Zhu, Jun-Ping
    Li, Hai
    [J]. Jilin Daxue Xuebao (Gongxueban)/Journal of Jilin University (Engineering and Technology Edition), 2022, 52 (08): : 1850 - 1856
  • [27] Decoupling Anomaly Discrimination and Representation Learning: Self-supervised Learning for Anomaly Detection on Attributed Graph
    Hu, Yanming
    Chen, Chuan
    Deng, Bowen
    Lai, Yujing
    Lin, Hao
    Zheng, Zibin
    Bian, Jing
    [J]. DATA SCIENCE AND ENGINEERING, 2024, 9 (03) : 264 - 277
  • [28] Adaptive Memory Networks With Self-Supervised Learning for Unsupervised Anomaly Detection
    Zhang, Yuxin
    Wang, Jindong
    Chen, Yiqiang
    Yu, Han
    Qin, Tao
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (12) : 12068 - 12080
  • [29] Task-Oriented Self-supervised Learning for Anomaly Detection in Electroencephalography
    Zheng, Yaojia
    Liu, Zhouwu
    Mo, Rong
    Chen, Ziyi
    Zheng, Wei-Shi
    Wang, Ruixuan
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT VIII, 2022, 13438 : 193 - 203
  • [30] Federated Graph Anomaly Detection via Contrastive Self-Supervised Learning
    Kong, Xiangjie
    Zhang, Wenyi
    Wang, Hui
    Hou, Mingliang
    Chen, Xin
    Yan, Xiaoran
    Das, Sajal K.
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024,