English

CerberusDet: Unified Multi-Dataset Object Detection

Computer Vision and Pattern Recognition 2024-09-16 v2

Abstract

Conventional object detection models are usually limited by the data on which they were trained and by the category logic they define. With the recent rise of Language-Visual Models, new methods have emerged that are not restricted to these fixed categories. Despite their flexibility, such Open Vocabulary detection models still fall short in accuracy compared to traditional models with fixed classes. At the same time, more accurate data-specific models face challenges when there is a need to extend classes or merge different datasets for training. The latter often cannot be combined due to different logics or conflicting class definitions, making it difficult to improve a model without compromising its performance. In this paper, we introduce CerberusDet, a framework with a multi-headed model designed for handling multiple object detection tasks. Proposed model is built on the YOLO architecture and efficiently shares visual features from both backbone and neck components, while maintaining separate task heads. This approach allows CerberusDet to perform very efficiently while still delivering optimal results. We evaluated the model on the PASCAL VOC dataset and Objects365 dataset to demonstrate its abilities. CerberusDet achieved state-of-the-art results with 36% less inference time. The more tasks are trained together, the more efficient the proposed model becomes compared to running individual models sequentially. The training and inference code, as well as the model, are available as open-source (https://github.com/ai-forever/CerberusDet).

Keywords

Cite

@article{arxiv.2407.12632,
  title  = {CerberusDet: Unified Multi-Dataset Object Detection},
  author = {Irina Tolstykh and Mikhail Chernyshov and Maksim Kuprashevich},
  journal= {arXiv preprint arXiv:2407.12632},
  year   = {2024}
}

Comments

12 pages, 4 figures

R2 v1 2026-06-28T17:44:33.859Z