arXiv:2505.06694cs.CVcs.AI2025-05被引 3

用NAS搜索优化的Transformer模型,提升声呐图像目标检测精度与效率

Underwater object detection in sonar imagery with detection transformer and Zero-shot neural architecture search

  • 基于最大熵原理的零样本NAS搜索,自动发现适合声呐图像的高效骨干网络
  • 在两个数据集上达到当前最优性能,同时保持低计算开销和实时性
  • 首次将DETR与NAS结合,适合水下探测、智能船舶等实际应用

利用声呐图像进行水下目标检测已成为海洋技术中的关键研究方向。然而,相较于光学图像,声呐图像分辨率更低、特征更稀疏,严重制约检测性能。为此,本文提出一种基于神经架构搜索(NAS)优化的检测变压器(NAS-DETR)架构。首先,提出一种基于最大熵原理的改进型零样本NAS方法,用于高效发现兼具高表征能力与实时性的CNN-Transformer骨干网络;其次,将该骨干网络与特征金字塔网络(FPN)及可变形注意力机制的Transformer解码器结合,构建完整检测架构,并融合多种先进组件与训练策略以提升整体性能。大量实验表明,该架构在两个代表性数据集上均达到当前最优效果,且计算复杂度与实时性开销极小。此外,通过分析关键参数与基于微分熵的适应度函数之间的相关性,增强了框架的可解释性。据我们所知,这是首个将DETR架构与NAS搜索机制结合应用于声呐目标检测的工作。

原文摘要 · Abstract (English)

Underwater object detection using sonar imagery has become a critical and rapidly evolving research domain within marine technology. However, sonar images are characterized by lower resolution and sparser features compared to optical images, which seriously degrades the performance of object detection.To address these challenges, we specifically propose a Detection Transformer (DETR) architecture optimized with a Neural Architecture Search (NAS) approach called NAS-DETR for object detection in sonar images. First, an improved Zero-shot Neural Architecture Search (NAS) method based on the maximum entropy principle is proposed to identify a real-time, high-representational-capacity CNN-Transformer backbone for sonar image detection. This method enables the efficient discovery of high-performance network architectures with low computational and time overhead. Subsequently, the backbone is combined with a Feature Pyramid Network (FPN) and a deformable attention-based Transformer decoder to construct a complete network architecture. This architecture integrates various advanced components and training schemes to enhance overall performance. Extensive experiments demonstrate that this architecture achieves state-of-the-art performance on two Representative datasets, while maintaining minimal overhead in real-time efficiency and computational complexity. Furthermore, correlation analysis between the key parameters and differential entropy-based fitness function is performed to enhance the interpretability of the proposed framework. To the best of our knowledge, this is the first work in the field of sonar object detection to integrate the DETR architecture with a NAS search mechanism.

声呐检测TransformerNAS搜索水下感知

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。