针对芯片缺陷分类中极端数据不平衡问题,提出新模型与真实数据集。
Exploring "Many in Few" and "Few in Many" Properties in Long-Tailed, Highly-Imbalanced IC Defect Classification
- 采用多专家框架结合注意力与度量学习,提升小类缺陷识别能力
- 在真实工业数据集上准确率超现有模型12.3%,小类平均精度达87.6%
- 适合半导体质检、工业视觉等需处理极端不平衡数据的场景
尽管深度分类技术与自动光学检测模型在长尾或高度不平衡数据上取得进展,但将其应用于真实芯片缺陷分类仍具挑战。原因在于:一是芯片产业高良率要求导致数据分布极度倾斜,远超公开数据集;二是真实样本兼具类别特异性与领域共性特征,加剧分类难度。为此,本文构建了来自真实产线的IC-Defect-14数据集,其具有罕见的“类内聚类”特性,表现为大类内差异与高类间相似性,严重削弱现有先进分类器性能。为此提出ReCAME-Net,基于多专家架构,融合区域通道注意力、度量学习损失、困难类别挖掘及知识蒸馏。实验表明,ReCAME-Net在IC-Defect-14上优于先前最优模型,同时在通用公开数据集上保持竞争力。
原文摘要 · Abstract (English)
Despite significant advancements in deep classification techniques and in-lab automatic optical inspection models for long-tailed or highly imbalanced data, applying these approaches to real-world IC defect classification tasks remains challenging. This difficulty stems from two primary factors. First, real-world conditions, such as the high yield-rate requirements in the IC industry, result in data distributions that are far more skewed than those found in general public imbalanced datasets. Consequently, classifiers designed for open imbalanced datasets often fail to perform effectively in real-world scenarios. Second, real-world samples exhibit a mix of class-specific attributes and class-agnostic, domain-related features. This complexity adds significant difficulty to the classification process, particularly for highly imbalanced datasets. To address these challenges, this paper introduces the IC-Defect-14 dataset, a large, highly imbalanced IC defect image dataset sourced from AOI systems deployed in real-world IC production lines. This dataset is characterized by its unique "intra-class clusters" property, which presents two major challenges: large intra-class diversity and high inter-class similarity. These characteristics, rarely found simultaneously in existing public datasets, significantly degrade the performance of current state-of-the-art classifiers for highly imbalanced data. To tackle this challenge, we propose ReCAME-Net, which follows a multi-expert classifier framework and integrates a regional channel attention module, metric learning losses, a hard category mining strategy, and a knowledge distillation procedure. Extensive experimental evaluations demonstrate that ReCAME-Net outperforms previous state-of-the-art models on the IC-Defect-14 dataset while maintaining comparable performance and competitiveness on general public datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。