A Self-Supervised Approach for Enhanced Feature Representations in Object Detection Tasks
Organizations: Menéndez Pelayo International University, Spain · Artificial Intelligence Center, University of Oviedo, Gijón, Spain
Abstract
In the fast-evolving field of artificial intelligence, where models are increasingly growing in complexity and size, the availability of labeled data for training deep learning models has become a significant challenge. Addressing complex problems like object detection demands considerable time and resources for data labeling to achieve meaningful results. For companies developing such applications, this entails extensive investment in highly skilled personnel or costly outsourcing. This research work aims to demonstrate that enhancing feature extractors can substantially alleviate this challenge, enabling models to learn more effective representations with less labeled data. Utilizing a self-supervised learning strategy, we present a model trained on unlabeled data that outperforms state-of-the-art feature extractors pre-trained on ImageNet and particularly designed for object detection tasks. Moreover, the results demonstrate that our approach encourages the model to focus on the most relevant aspects of an object, thus achieving better feature representations and, therefore, reinforcing its reliability and robustness.
Figures & tables
| Method | Classification | Localization | ||||
|---|---|---|---|---|---|---|
| Top-1 Acc | Top-3 Acc | Mean IoU | Acc IoU 0.5 | Acc IoU 0.7 | ||
| 10 | Baseline | 0.8262 | 0.9717 | 0.2594 | 0.1916 | 0.0440 |
| SSL | 0.4440 | 0.8094 | 0.4759 | 0.5215 | 0.2408 | |
| 20 | Baseline | 0.8806 | 0.9885 | 0.2818 | 0.2168 | 0.0408 |
| SSL | 0.4911 | 0.8293 | 0.4887 | 0.5455 | 0.2450 | |
| 50 | Baseline | 0.9277 | 0.9969 | 0.4095 | 0.3937 | 0.1099 |
| Method | Classification | Localization | |||||
|---|---|---|---|---|---|---|---|
| Top-1 Acc | Top-3 Acc | Top-5 Acc | Mean IoU | Acc IoU 0.5 | Acc IoU 0.7 | ||
| 3 | Baseline | 0.6259 | 0.8236 | 0.8880 | 0.1685 | 0.1206 | 0.0541 |
| SSL | 0.2223 | 0.4691 | 0.6215 | 0.4169 | 0.4080 | 0.1464 | |
| 5 | Baseline | 0.6248 | 0.8203 | 0.8984 | 0.1699 | 0.0983 | 0.0131 |
| SSL | 0.2615 | 0.4981 | 0.6412 | 0.4410 | 0.4659 | 0.1830 | |
| 10 | Baseline | 0.7160 | 0.8957 | 0.9483 | 0.2669 | 0.2108 | 0.0552 |
Object Detection
Object detection locates and classifies objects in images or video, balancing localization accuracy, robustness, and inference speed.
See all 624 papers