Prototypical rectification with interpretable fusion for enhanced industrial defect segmentation
摘要
Surface defect segmentation is crucial in industrial manufacturing for ensuring product quality. Traditional methods face challenges such as visual ambiguity and data scarcity. We introduce PRIF, a novel semi-supervised framework leveraging prototypical rectification with interpretable fusion. PRIF dynamically focuses on ambiguous regions through a multi-source uncertainty mechanism, integrating epistemic, structural, and divergence-based uncertainties. Our architecture features a PRIF block with a path aggregation nexus for long-range dependency modeling and a bilateral cross-attention Ffsion module for superior feature integration. With only 3.8M parameters, PRIF is highly efficient and enables real-time deployment in industrial environments. Extensive experiments on five diverse benchmarks demonstrate PRIF’s robustness and efficiency, achieving state-of-the-art performance with 110 FPS on an NVIDIA RTX 4060 GPU. Visualizations confirm enhanced interpretability and well-calibrated uncertainty estimates.