Research Article 2026-04-21 under-review v1

YOLO-HCS: A YOLO-Based Framework for Small Defect Detection in Industrial Automated Optical Inspection

C
Chun-Cheng Lin National Yang Ming Chiao Tung University
H
Hsiang-En Weng National Yang Ming Chiao Tung University
S
Shin-Hang Lu Yuan Ze University
H
Heng-Yih Liu Yuan Ze University

Abstract

Connectors are used in electronic and industrial systems to establish electrical connections between components, with terminal surfaces playing a role in ensuring current transmission. Accurate defect detection on these terminals is vital for maintaining product reliability and quality. However, detecting small, irregular, and scale-varying defects remains a challenge for automated visual inspection systems. To tackle this challenge, this work presents You Only Look Once with Hierarchical Coordinate Sampling (YOLO-HCS), a defect detection framework that integrates hierarchical positive sample selection (HPSS) to improve training stability, coordinate attention (CA) to enhance spatial localization of small targets, and scylla-Intersection over Union (SIoU) loss to refine bounding box regression through geometric constraints. We validate the approach on two datasets collected using automated optical inspection (AOI) systems. The proposed YOLO-HCS framework integrates hierarchical positive sample selection, coordinate attention, and SIoU loss to improve the detection of small and irregular defects in industrial AOI environments. Experimental results on two real-world AOI datasets demonstrate that the YOLO- HCS model achieves higher mean average precision (mAP), outperforming YOLOv7 and other state-of-the-art detectors. Beyond accuracy gains, the proposed method improves production line reliability, reduces reliance on manual inspection, and satisfies the real-time speed requirements of high-throughput manufacturing. Its lightweight design enables efficient deployment in industrial AOI systems while maintaining real-time inspection capability. While validated on connector terminals, the approach is broadly transferable to semiconductor packaging, printed circuit board (PCB) inspection, and precision component manufacturing, offering a practical pathway to strengthen industrial quality control and reduce operational costs.

Citation Information

@article{chunchenglin2026,
  title={YOLO-HCS: A YOLO-Based Framework for Small Defect Detection in Industrial Automated Optical Inspection},
  author={Chun-Cheng Lin and Hsiang-En Weng and Shin-Hang Lu and Heng-Yih Liu},
  journal={Machine Vision and Applications},
  year={2026},
  doi={https://doi.org/10.21203/rs.3.rs-9130149/v1}
}
Back to Top
Home
Paper List
Submit
0.023089s