Wafer Map Defect Classification Using Autoencoder-Based Data Augmentation and Convolutional Neural Network

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bao, Yin-Yin, Li, Er-Chao, Yang, Hong-Qiang, Jia, Bin-Bin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929594817314816
author Bao, Yin-Yin
Li, Er-Chao
Yang, Hong-Qiang
Jia, Bin-Bin
author_facet Bao, Yin-Yin
Li, Er-Chao
Yang, Hong-Qiang
Jia, Bin-Bin
contents In semiconductor manufacturing, wafer defect maps (WDMs) play a crucial role in diagnosing issues and enhancing process yields by revealing critical defect patterns. However, accurately categorizing WDM defects presents significant challenges due to noisy data, unbalanced defect classes, and the complexity of failure modes. To address these challenges, this study proposes a novel method combining a self-encoder-based data augmentation technique with a convolutional neural network (CNN). By introducing noise into the latent space, the self-encoder enhances data diversity and mitigates class imbalance, thereby improving the model's generalization capabilities. The augmented dataset is subsequently used to train the CNN, enabling it to deliver precise classification of both common and rare defect patterns. Experimental results on the WM-811K dataset demonstrate that the proposed method achieves a classification accuracy of 98.56%, surpassing Random Forest, SVM, and Logistic Regression by 19%, 21%, and 27%, respectively. These findings highlight the robustness and effectiveness of the proposed approach, offering a reliable solution for wafer defect detection and classification.
format Preprint
id arxiv_https___arxiv_org_abs_2411_11029
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Wafer Map Defect Classification Using Autoencoder-Based Data Augmentation and Convolutional Neural Network
Bao, Yin-Yin
Li, Er-Chao
Yang, Hong-Qiang
Jia, Bin-Bin
Computer Vision and Pattern Recognition
Artificial Intelligence
Image and Video Processing
68T07, 68U10
I.2.10; I.5.1; I.5.4; I.4.8
In semiconductor manufacturing, wafer defect maps (WDMs) play a crucial role in diagnosing issues and enhancing process yields by revealing critical defect patterns. However, accurately categorizing WDM defects presents significant challenges due to noisy data, unbalanced defect classes, and the complexity of failure modes. To address these challenges, this study proposes a novel method combining a self-encoder-based data augmentation technique with a convolutional neural network (CNN). By introducing noise into the latent space, the self-encoder enhances data diversity and mitigates class imbalance, thereby improving the model's generalization capabilities. The augmented dataset is subsequently used to train the CNN, enabling it to deliver precise classification of both common and rare defect patterns. Experimental results on the WM-811K dataset demonstrate that the proposed method achieves a classification accuracy of 98.56%, surpassing Random Forest, SVM, and Logistic Regression by 19%, 21%, and 27%, respectively. These findings highlight the robustness and effectiveness of the proposed approach, offering a reliable solution for wafer defect detection and classification.
title Wafer Map Defect Classification Using Autoencoder-Based Data Augmentation and Convolutional Neural Network
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Image and Video Processing
68T07, 68U10
I.2.10; I.5.1; I.5.4; I.4.8
url https://arxiv.org/abs/2411.11029