Advances in Automated Optical Inspection Using Attention-based Convolutional Neural Networks
Researchers at Chungbuk National University have proposed an attention-based convolutional neural network (A-CNN) model for automated optical inspection (AOI) of small objects. The A-CNN model uses an attention module to adaptively extract regions of interest centered on small objects, improving classification accuracy. Experimental results show that the proposed A-CNN model achieves a classification accuracy of 99.92% and an inference speed of 33 frames/s on the NVIDIA Jetson Nano platform.
Key Takeaways:
- The A-CNN model integrates an attention module to adaptively extract regions of interest centered on small objects, increasing the object-to-image area ratio and improving classification accuracy.
- The A-CNN model achieves a classification accuracy of 99.92% and an inference speed of 33 frames/s on the NVIDIA Jetson Nano platform.
- The proposed model outperforms state-of-the-art object detection algorithms, including YOLOv5, YOLOv7, YOLOv8, YOLOv9, and YOLOv10, in terms of both accuracy and latency.
- The A-CNN model has a faster inference speed than the fastest YOLO model, providing $2.0\times$ speedup.
- The A-CNN model is a practical solution for small object classification, with potential applications in real-time inspection and quality control.
- The researchers propose that their model can be effectively trained end-to-end with minimal data labeling compared with object detection methods.
Statistics:
- Classification accuracy: 99.92%
- Inference speed: 33 frames/s
- Object-to-image area ratio (OAR): Improved by using the attention module
- Training time: Minimal data labeling required
- Inference speedup: $2.0\times$ faster than the fastest YOLO model
Sources:
- "An Attention-based Convolutional Neural Network With Spatial Transformer Module for Automated Optical Inspection of Small Objects" (IEEE Transactions on Instrumentation and Measurement, 2025;74)
- Journal of Engineering (May 19, 2025, p 2618)
- IEEE Transactions on Instrumentation and Measurement (www.ieee.org/)