Hyper-YOLO: When Visual Object Detection Meets Hypergraph Computation
About
We introduce Hyper-YOLO, a new object detection method that integrates hypergraph computations to capture the complex high-order correlations among visual features. Traditional YOLO models, while powerful, have limitations in their neck designs that restrict the integration of cross-level features and the exploitation of high-order feature interrelationships. To address these challenges, we propose the Hypergraph Computation Empowered Semantic Collecting and Scattering (HGC-SCS) framework, which transposes visual feature maps into a semantic space and constructs a hypergraph for high-order message propagation. This enables the model to acquire both semantic and structural information, advancing beyond conventional feature-focused learning. Hyper-YOLO incorporates the proposed Mixed Aggregation Network (MANet) in its backbone for enhanced feature extraction and introduces the Hypergraph-Based Cross-Level and Cross-Position Representation Network (HyperC2Net) in its neck. HyperC2Net operates across five scales and breaks free from traditional grid structures, allowing for sophisticated high-order interactions across levels and positions. This synergy of components positions Hyper-YOLO as a state-of-the-art architecture in various scale models, as evidenced by its superior performance on the COCO dataset. Specifically, Hyper-YOLO-N significantly outperforms the advanced YOLOv8-N and YOLOv9-T with 12\% $\text{AP}^{val}$ and 9\% $\text{AP}^{val}$ improvements. The source codes are at ttps://github.com/iMoonLab/Hyper-YOLO.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Object Detection | MS-COCO (val) | mAP0.52 | 211 | |
| Object Detection | DUT Anti-UAV | Precision94.12 | 12 | |
| Object Detection | custom UAV dataset | Precision94.55 | 12 | |
| Instance Segmentation | Trans10K | Mask mAP@.5079.9 | 9 | |
| Instance Segmentation | GVD | Mask mAP@0.5080.9 | 9 | |
| Object Detection | Trans10K | Box mAP@.5080.1 | 9 | |
| Object Detection | GVD | Box mAP@0.5081.9 | 9 |