Density-aware multi-modal feature fusion for wafer map defect classification in semiconductor manufacturing
摘要
Accurate classification of wafer defect maps is essential for improving semiconductor yield. Yet, conventional methods often underutilize domain-specific knowledge and consequently achieve limited performance. To address this gap, we propose a knowledge-data dual-driven framework that embeds wafer-specific expertise into deep learning. Our method comprises two main components: the Knowledge-Aware Feature Enhancement (KAFE) module and the Knowledge-Data Dual-Driven Network (KDDNet). KAFE transforms raw wafer maps into continuous defect-density heatmaps, selectively enhancing regions with concentrated defects while suppressing background noise. Then, KDDNet builds on these enhanced representations through a dual-branch design. A multi-scale convolutional subnetwork extracts explicit spatial defect patterns, while a variational feature-pyramid subnetwork learns latent statistical distributions. These complementary features are fused for final classification. On the MixedWM38 benchmark, our method achieves 98.64% accuracy, 98.65% precision, 98.64% recall, and 98.64% F1-score with only 100 training iterations-delivering nearly 40