Few-shot Object Detection Algorithm Based on YOLOv8-Mamba-Like Linear Attention

In the production and assembly process, the accuracy of detection is usually low due to limited resources and small quantities. In this paper, we propose a few-shot learning method for object detection based on the combination of YOLOv8 and Mamba-Like Linear Attention (MLLA) module. This method utilizes a high-pixel camera to capture the shape and size of the workpiece. By consistency-based mutual augmentation structure for few-shot samples data augmentation. By introducing the efficient and lightweight MLLA module, the model’s speed and accuracy in object detection and segmentation tasks on mobile devices are significantly improved. Specifically, MLLA mainly integrates the two key elements of “forget gate” and “block design” in Mamba, and MLLA replaces forget gate by using position coding (RoPE) to provide necessary location information while maintaining parallel computation and fast inference speed. This makes MLLA improves the accuracy of detection and segmentation recognition. Experimental results show that compared to the original YOLOv8 model, the model with the MLLA module exhibits improvements in segmentation precision and recognition accuracy. This method demonstrates high accuracy and robustness in workpiece recognition, making it suitable for real-time industrial applications.