A prototype-aligned multi-scale attention network for few-shot industrial surface defect detection

Rubing Huang and Lianyi Zhao

PLOS ONE, 2026, vol. 21, issue 9, 1-31

Abstract: Few-shot industrial surface defect detection remains challenging because annotated defect images are scarce, defect scales vary considerably, and textured backgrounds can cause false responses. This study proposes a Prototype-Aligned Multi-Scale Attention Network (PA-MSAN) that integrates multi-receptive-field feature extraction, spatial-channel attention, and region of interest (RoI)-level prototype matching within a two-stage detector. Experiments are conducted on NEU-DET under 1-shot, 3-shot, and 5-shot episode settings, with 5-shot cross-domain evaluation on GC10-DET using target-domain support prototypes. Multi-scale feature-pyramid fusion represents local textures and broader defect morphology, while attention weighting enhances defect-sensitive regions and channels before region proposal network (RPN) proposal generation and RoI Align. Few-shot classification is performed by cosine similarity between RoI features and class prototypes, and training jointly optimizes detection classification, bounding-box regression, region proposal, and Prototype Alignment Loss (PAL). Under the 5-shot NEU-DET setting, PA-MSAN achieves 92.2 ± 0.4% mean average precision (mAP)@0.5, 88.7 ± 0.4% small-object average precision (APsmall), 0.89 ± 0.01 average intersection over union (IoU), and a 4.1 ± 0.2% background false response rate, with 23.4 ms detection time and 18.7 × 10⁶ parameters. Compared with Faster R-CNN, PA-MSAN improves mAP@0.5 by 2.9 percentage points and APsmall by 3.9 percentage points while reducing the background false response rate by 1.7 percentage points. Under the 5-shot target-domain support setting, PA-MSAN achieves 86.4 ± 0.4% mAP@0.5 on GC10-DET, 5.8 percentage points (6.3%) below the corresponding NEU-DET result. The results support region-level prototype alignment for limited-label metallic surface inspection under the evaluated datasets and two-stage configuration.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0359400 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 59400&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0359400

DOI: 10.1371/journal.pone.0359400

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().