Work

DTA-Head: Dynamic Task Alignment Head for Regression and Classification in Small Object Detection

Year: 2025

Type: article

Abstract: Detecting small targets poses significant challenges due to the limited feature information and the aggregation of features in deep feature maps. Existing single-stage detectors handle classification and regression separately, leading to inconsistent predictions and potential filtering by Non-Maximum Suppression (NMS). To address these issues, we propose the Dynamic Task Alignment (DTA) Head. This novel approach comprises two branches: regression and classification. The regression branch computes offsets and masks for feature alignment adopted by Deformable ConvNets v2 (DCNv2), while the classification branch enhances feature interaction through dynamic selection. The task-decomposition module separates features for each branch. Additionally, we introduce Diverse-Scale Channel-Specific Convolution (DSCSC) to apply diverse convolutions across specific channels and exchange channel information. Our methods achieved an [email protected] of 30.9% on the TinyPerson dataset, a 3.3% improvement over the original model’s 27.6% and outperforming other common models.

Source: Applied Sciences

DOI: 10.3390/app15169060

Authors: Kaiqi Ye, Qi Li, Yunfeng Yan, Xianbo Wang, Donglian Qi

Institution: Zhejiang University

Language: English

Topic: Advanced Neural Network Applications

Subfield: Computer Vision and Pattern Recognition

Field: Computer Science

Domain: Physical Sciences

Open Access status: gold

Estimated APC paid: $2,896

Funders: National Natural Science Foundation of China, Zhejiang University, State Grid Corporation of China, Natural Science Foundation of Zhejiang Province