# EFEN-YOLOv8: Surface defect detection network based on spatial feature capture and multi-level weighted attention

## TL;DR

This paper introduces EFEN-YOLOv8, a new deep learning framework for detecting surface defects in industrial settings with improved accuracy and efficiency.

## Contribution

EFEN-YOLOv8 introduces novel modules for spatial feature capture and multi-level attention to enhance surface defect detection in industrial environments.

## Key findings

- EFEN-YOLOv8 achieves a 7.4% mAP improvement on the NEU-DET dataset compared to baseline models.
- The framework shows a 3.3% performance enhancement on the GC10-DET dataset.
- The method demonstrates robust generalization across different train-test data splits.

## Abstract

Surface defects in industrial environments severely the impact product aesthetics, quality, and operational efficiency. Although deep learning approaches show promise, current architectures often demonstrate inadequate feature extraction in industrial settings. We introduce EFEN-YOLOv8, a novel defect detection framework that prioritizes efficient feature extraction to enhance detection accuracy. Our approach incorporates a β-FEIoU loss function that concurrently tackles defect-background discrimination and positive-negative sample imbalance. The Shallow Attention Convolution (SAConv) module strengthens feature localization in early network layers, while Large Separable Kernel Attention (LSKA) expands receptive fields and augments processing efficiency. Additionally, our Weighted Atrous Spatial Pyramid Pooling (WASPP) feature fusion module facilitates multi-scale integration, enabling richer abstract information capture and improved model representation. Comprehensive experimental validation, including statistical significance testing across diverse data splits, confirms superior performance over existing methods. Our framework achieves 7.4% mAP improvement on NEU-DET and 3.3% enhancement on GC10-DET compared to baseline models, maintaining consistent performance across both 8:2 and 9:1 train-test configurations. These findings validate the method’s robust generalization capacity and establish its effectiveness for industrial surface defect detection applications. Code and datasets are available at: https://github.com/01WineCool/YOLO.