DCGAN Feature-Enhancement-Based YOLOv8n Model in Small-Sample Target Detection
This paper proposes DCGAN-YOLOv8n, an integrated framework that significantly advances small-sample target detection by synergizing generative adversarial feature enhancement with multi-scale representation learning. The model’s core contribution lies in its novel adversarial feature enhancement module (AFEM), which leverages conditional generative adversarial networks to reconstruct discriminative multi-scale features while effectively mitigating mode collapse. Furthermore, the architecture incorporates a deformable multi-scale feature pyramid that dynamically fuses generated high-resolution features with hierarchical semantic representations through an attention mechanism. The proposed triple marginal constraint optimization jointly enhances intra-class compactness and inter-class separation, thereby structuring a highly discriminative feature space. Extensive experiments on the NWPU VHR-10 dataset demonstrate state-of-the-art performance, with the model achieving an mAP50 of 90.46% and an mAP50-95 of 57.06%, representing significant improvements of 4.52% and 4.08% over the baseline YOLOv8n, respectively. These results validate the framework’s effectiveness in addressing critical challenges of feature representation scarcity and cross-scale adaptation in data-limited scenarios.
Authors
- Peng Zheng (ORCID: https://orcid.org/0000-0003-4327-2308)
- Yuanyi Liu (ORCID: https://orcid.org/0000-0002-3696-2047)
- Yun Cheng (ORCID: https://orcid.org/0000-0001-5624-8628)
- Shijie Wang (ORCID: https://orcid.org/0000-0002-8175-0836)
- Shuhong Liu (ORCID: https://orcid.org/0000-0002-8997-4180)
- Wei Zhu (ORCID: https://orcid.org/0000-0002-5154-0889)
- Jinyin Bai
- Chenhao Ye
Institutions
- National Defense University (US)
- National University of Defense Technology (CN)
Publication Details
- Journal
- Computers
- Published
- 2025-09-15
- DOI
- https://doi.org/10.3390/computers14090389
- Primary Topic
- Infrared Target Detection Methodologies
- Type
- article
- Field-Weighted Citation Impact
- 0.00