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

Hyungu Kahng, Seoung Bum Kim

Research output: Contribution to journalArticlepeer-review

14 Citations (Scopus)


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

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


  • 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


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