YOLO-PRO: Enhancing Instance-Specific Object Detection with Full-Channel Global Self-Attention

Lin Huang, Yujuan Tan, Weisheng Li, Shitai Shan, Liu Liu +3 more
3/4/2025
cs.CV

No confident code implementation found for this paper. 2 possible matches can be shown below.

Abstract

This paper addresses the inherent limitations of conventional bottleneck structures (diminished instance discriminability due to overemphasis on batch statistics) and decoupled heads (computational redundancy) in object detection frameworks by proposing two novel modules: the Instance-Specific Bottleneck with full-channel global self-attention (ISB) and the Instance-Specific Asymmetric Decoupled Head (ISADH). The ISB module innovatively reconstructs feature maps to establish an efficient full-channel global attention mechanism through synergistic fusion of batch-statistical and instance-specific features. Complementing this, the ISADH module pioneers an asymmetric decoupled architecture enabling hierarchical multi-dimensional feature integration via dual-stream batch-instance representation fusion. Extensive experiments on the MS-COCO benchmark demonstrate that the coordinated deployment of ISB and ISADH in the YOLO-PRO framework achieves state-of-the-art performance across all computational scales. Specifically, YOLO-PRO surpasses YOLOv8 by 1.0-1.6% AP (N/S/M/L/X scales) and outperforms YOLO11 by 0.1-0.5% AP in critical N/M/L/X groups, while maintaining competitive computational efficiency. This work provides practical insights for developing high-precision detectors deployable on edge devices.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 2 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{huang2025yolopro,
  title  = {YOLO-PRO: Enhancing Instance-Specific Object Detection with Full-Channel Global Self-Attention},
  author = {Lin Huang and Yujuan Tan and Weisheng Li and Shitai Shan and Liu Liu and Linlin Shen and Jing Yu and Yue Niu},
  year   = {2025},
  eprint = {2503.02348},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2503.02348v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2503.02348.svg)](https://codeofpaper.com/paper/2503.02348)

Discussion