Hyper-YOLO: When Visual Object Detection Meets Hypergraph Computation
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% and 9% improvements.
Authors
- Rongrong Ji (ORCID: https://orcid.org/0000-0001-9163-2932)
- Jiangang Huang (ORCID: https://orcid.org/0000-0002-6120-1530)
- Shaoyi Du
- Guiguang Ding (ORCID: https://orcid.org/0000-0003-0137-9975)
- Yipeng Li (ORCID: https://orcid.org/0000-0002-1443-3064)
- Shihui Ying (ORCID: https://orcid.org/0000-0001-9423-0146)
- Jun‐Hai Yong (ORCID: https://orcid.org/0000-0002-4326-4167)
- Yue Gao (ORCID: https://orcid.org/0000-0002-4971-590X)
- Yifan Feng (ORCID: https://orcid.org/0000-0003-0878-2986)
Institutions
- Shanghai University (CN)
- Xiamen University (CN)
- Beijing Academy of Artificial Intelligence (CN)
- Artificial Intelligence in Medicine (Canada) (CA)
- Xi'an Jiaotong University (CN)
- Tsinghua University (CN)
Publication Details
- Journal
- IEEE Transactions on Pattern Analysis and Machine Intelligence
- Published
- 2024-12-30
- DOI
- https://doi.org/10.1109/tpami.2024.3524377
- Citations
- 179
- Primary Topic
- Machine Learning and Data Classification
- Type
- article
- Field-Weighted Citation Impact
- 62.18
Funders
- National Natural Science Foundation of China