Wafer Defect Classification Algorithm With Label Embedding Using Contrastive Learning
Classifying wafer defects in the wafer manufacturing process is increasingly critical for ensuring high-quality production, optimizing processes, and reducing costs. Most existing methods for wafer map defect classification primarily rely on images alone for model training and prediction. However, t...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2025-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10835094/ |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832592932836737024 |
|---|---|
| author | Jeongjoon Hwang Somi Ha Dohyun Kim |
| author_facet | Jeongjoon Hwang Somi Ha Dohyun Kim |
| author_sort | Jeongjoon Hwang |
| collection | DOAJ |
| description | Classifying wafer defects in the wafer manufacturing process is increasingly critical for ensuring high-quality production, optimizing processes, and reducing costs. Most existing methods for wafer map defect classification primarily rely on images alone for model training and prediction. However, these approaches often lack interpretability, which can hinder process improvement and problem-solving efforts. In other words, existing methods only calculate the probability of a specific image belonging to each class, making it difficult to visually judge why the image belongs to a particular class. Additionally, these methods make it challenging to assess the distance of new images from each class. Furthermore, it is difficult to obtain representative images of each class. To address these limitations, we propose a novel approach for wafer defect classification using contrastive learning with label embedding. The proposed method aims to map label information and wafer defect images into a shared latent space through contrastive learning using label embedding. This not only facilitates defect class prediction from images but also enhances interpretability by visualizing relationships between images and defects (labels) and providing representative defect images. Moreover, compared to previous methods, our approach demonstrates better classification performance and computational efficiency, even in situations with imbalanced labels. This method also shows significant potential in identifying unseen defects not defined in the original classification tasks. Consequently, the proposed approach extends its applicability beyond wafer map defect patterns, showing promising potential for use in various domains. |
| format | Article |
| id | doaj-art-716e129a305f40f8a8396d752a6ccf79 |
| institution | Kabale University |
| issn | 2169-3536 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj-art-716e129a305f40f8a8396d752a6ccf792025-01-21T00:01:44ZengIEEEIEEE Access2169-35362025-01-01139708971710.1109/ACCESS.2025.352749110835094Wafer Defect Classification Algorithm With Label Embedding Using Contrastive LearningJeongjoon Hwang0Somi Ha1Dohyun Kim2https://orcid.org/0000-0003-1669-6746Department of Industrial and Management Engineering, Myongji University, Yongin, Gyeonggi-do, Republic of KoreaDepartment of Industrial and Management Engineering, Myongji University, Yongin, Gyeonggi-do, Republic of KoreaDepartment of Industrial and Management Engineering, Myongji University, Yongin, Gyeonggi-do, Republic of KoreaClassifying wafer defects in the wafer manufacturing process is increasingly critical for ensuring high-quality production, optimizing processes, and reducing costs. Most existing methods for wafer map defect classification primarily rely on images alone for model training and prediction. However, these approaches often lack interpretability, which can hinder process improvement and problem-solving efforts. In other words, existing methods only calculate the probability of a specific image belonging to each class, making it difficult to visually judge why the image belongs to a particular class. Additionally, these methods make it challenging to assess the distance of new images from each class. Furthermore, it is difficult to obtain representative images of each class. To address these limitations, we propose a novel approach for wafer defect classification using contrastive learning with label embedding. The proposed method aims to map label information and wafer defect images into a shared latent space through contrastive learning using label embedding. This not only facilitates defect class prediction from images but also enhances interpretability by visualizing relationships between images and defects (labels) and providing representative defect images. Moreover, compared to previous methods, our approach demonstrates better classification performance and computational efficiency, even in situations with imbalanced labels. This method also shows significant potential in identifying unseen defects not defined in the original classification tasks. Consequently, the proposed approach extends its applicability beyond wafer map defect patterns, showing promising potential for use in various domains.https://ieeexplore.ieee.org/document/10835094/Deep learningimage classificationwafer defect classificationcontrastive learninglabel embedding |
| spellingShingle | Jeongjoon Hwang Somi Ha Dohyun Kim Wafer Defect Classification Algorithm With Label Embedding Using Contrastive Learning IEEE Access Deep learning image classification wafer defect classification contrastive learning label embedding |
| title | Wafer Defect Classification Algorithm With Label Embedding Using Contrastive Learning |
| title_full | Wafer Defect Classification Algorithm With Label Embedding Using Contrastive Learning |
| title_fullStr | Wafer Defect Classification Algorithm With Label Embedding Using Contrastive Learning |
| title_full_unstemmed | Wafer Defect Classification Algorithm With Label Embedding Using Contrastive Learning |
| title_short | Wafer Defect Classification Algorithm With Label Embedding Using Contrastive Learning |
| title_sort | wafer defect classification algorithm with label embedding using contrastive learning |
| topic | Deep learning image classification wafer defect classification contrastive learning label embedding |
| url | https://ieeexplore.ieee.org/document/10835094/ |
| work_keys_str_mv | AT jeongjoonhwang waferdefectclassificationalgorithmwithlabelembeddingusingcontrastivelearning AT somiha waferdefectclassificationalgorithmwithlabelembeddingusingcontrastivelearning AT dohyunkim waferdefectclassificationalgorithmwithlabelembeddingusingcontrastivelearning |