Self-Supervised Representation Learning for Wafer Bin Map Defect Pattern Classification

    Research output: Contribution to journalArticlepeer-review

    Abstract

    Automatic identification of defect patterns in wafer bin maps (WBMs) stands as a challenging problem for the semiconductor manufacturing industry. Deep convolutional neural networks have recently shown decent progress in learning spatial patterns in WBMs, but only at the expense of explicit manual supervision. Unfortunately, a clean set of labeled WBM samples is often limited in both size and quality, especially during rapid process development or early production stages. In this study, we propose a self-supervised learning framework that makes the most out of unlabeled data to learn beforehand rich visual representations for data-efficient WBM defect pattern classification. After self-supervised pre-training based on noise-contrastive estimation, the network is fine-tuned on the available labeled data to classify WBM defect patterns. We argue that self-supervised pre-training with a vast amount of unlabeled data substantially improves classification performance when labels are scarce. We demonstrate the effectiveness of our work on a real-world public WBM dataset, WM-811K. The code is available at https://github.com/hgkahng/WaPIRL.

    Original languageEnglish
    Article number9260238
    Pages (from-to)74-86
    Number of pages13
    JournalIEEE Transactions on Semiconductor Manufacturing
    Volume34
    Issue number1
    DOIs
    Publication statusPublished - 2021 Feb

    Bibliographical note

    Funding Information:
    Manuscript received September 9, 2020; revised October 9, 2020 and October 31, 2020; accepted November 11, 2020. Date of publication November 16, 2020; date of current version February 3, 2021. This work was supported in part by the Korea Institute for Advancement of Technology (KIAT) Grant funded by the Korean Government (MOTIE) under Grant P0008691 (Competency Development Program for Industry Specialists), and in part by the National Research Foundation of Korea Grant funded by the Korean Government (MSIT) under Grant NRF-2019R1A4A1024732. (Corresponding author: Seoung Bum Kim.) The authors are with the School of Industrial Management Engineering, Korea University, Seoul 02841, South Korea (e-mail: [email protected]; [email protected]).

    Publisher Copyright:
    © 1988-2012 IEEE.

    Keywords

    • Wafer bin map defect pattern classification
    • contrastive learning
    • convolutional neural networks
    • self-supervised learning
    • semiconductor manufacturing
    • unsupervised learning

    ASJC Scopus subject areas

    • Electronic, Optical and Magnetic Materials
    • Condensed Matter Physics
    • Industrial and Manufacturing Engineering
    • Electrical and Electronic Engineering

    Fingerprint

    Dive into the research topics of 'Self-Supervised Representation Learning for Wafer Bin Map Defect Pattern Classification'. Together they form a unique fingerprint.

    Cite this