Search papers, labs, and topics across Lattice.
This paper introduces GATE-AD, a novel reconstruction-based approach for few-shot industrial visual anomaly detection that uses a masked, representation-aligned Graph Attention Network (GAT) to learn normal sample appearance patterns. The model encodes patch-level visual features as graph nodes and uses stacked self-attentional layers to capture complex local relations, enhanced by a representation alignment component and a Scaled Cosine Error (SCE) objective function for defect assessment. Experiments on MVTec AD, VisA, and MPDD datasets demonstrate state-of-the-art performance in few-shot settings, achieving higher detection accuracy and lower inference latency compared to existing methods.
Achieve state-of-the-art few-shot anomaly detection with a Graph Attention Network that's not only more accurate but also 25% faster than existing methods.
Few-Shot Industrial Visual Anomaly Detection (FS-IVAD) comprises a critical task in modern manufacturing settings, where automated product inspection systems need to identify rare defects using only a handful of normal/defect-free training samples. In this context, the current study introduces a novel reconstruction-based approach termed GATE-AD. In particular, the proposed framework relies on the employment of a masked, representation-aligned Graph Attention Network (GAT) encoding scheme to learn robust appearance patterns of normal samples. By leveraging dense, patch-level, visual feature tokens as graph nodes, the model employs stacked self-attentional layers to adaptively encode complex, irregular, non-Euclidean, local relations. The graph is enhanced with a representation alignment component grounded on a learnable, latent space, where high reconstruction residual areas (i.e., defects) are assessed using a Scaled Cosine Error (SCE) objective function. Extensive comparative evaluation on the MVTec AD, VisA, and MPDD industrial defect detection benchmarks demonstrates that GATE-AD achieves state-of-the-art performance across the $1$- to $8$-shot settings, combining the highest detection accuracy (increase up to $1.8\%$ in image AUROC in the 8-shot case in MPDD) with the lowest per-image inference latency (at least $25.05\%$ faster), compared to the best-performing literature methods. In order to facilitate reproducibility and further research, the source code of GATE-AD is available at https://github.com/gthpapadopoulos/GATE-AD.
