轻量高效红外小目标检测,兼顾精度与计算速度
Make Both Ends Meet: A Synergistic Optimization Infrared Small Target Detection with Streamlined Computational Overhead
- 用MBConvblock和BSblock替代YOLOv8n的瓶颈结构,降低计算开销
- 引入可变卷积核的AVCStem,自适应调整感受野增强目标提取
- 通过全局通道混洗提升模型鲁棒性,适合嵌入式部署
红外小目标检测因成像信噪比低、纹理缺失和背景干扰复杂而极具挑战。现有方法多将任务视为语义分割,但存在两个关键问题:(1)远距离成像导致目标边界模糊;(2)特征堆叠造成过高计算开销。为此,提出基于YOLOv8n的轻量高效红外小目标检测框架LE-IRSTD。首先,识别出YOLOv8n骨干网络中C2f模块的多重瓶颈结构增加计算负担,采用Mobile Inverted Bottleneck Convolution块(MBConvblock)与Bottleneck Structure块(BSblock)替代,平衡效率与深层语义提取能力。其次,引入基于注意力的可变卷积茎干结构(AVCStem),以可变核卷积(VKConv)替换末端卷积,使卷积核自适应变化形状,扩展感受野以更好捕捉目标。最后,采用全局通道混洗卷积(GSConv)对不同卷积路径的通道特征进行混洗,增强模型泛化能力。实验表明,该方法在准确率与轻量化性能上均优于多个主流深度学习模型。
原文摘要 · Abstract (English)
Infrared small target detection(IRSTD) is widely recognized as a challenging task due to the inherent limitations of infrared imaging, including low signal-to-noise ratios, lack of texture details, and complex background interference. While most existing methods model IRSTD as a semantic segmentation task, but they suffer from two critical drawbacks: (1)blurred target boundaries caused by long-distance imaging dispersion; and (2) excessive computational overhead due to indiscriminate feature stackin. To address these issues, we propose the Lightweight Efficiency Infrared Small Target Detection (LE-IRSTD), a lightweight and efficient framework based on YOLOv8n, with following key innovations. Firstly, we identify that the multiple bottleneck structures within the C2f component of the YOLOv8-n backbone contribute to an increased computational burden. Therefore, we implement the Mobile Inverted Bottleneck Convolution block (MBConvblock) and Bottleneck Structure block (BSblock) in the backbone, effectively balancing the trade-off between computational efficiency and the extraction of deep semantic information. Secondly, we introduce the Attention-based Variable Convolution Stem (AVCStem) structure, substituting the final convolution with Variable Kernel Convolution (VKConv), which allows for adaptive convolutional kernels that can transform into various shapes, facilitating the receptive field for the extraction of targets. Finally, we employ Global Shuffle Convolution (GSConv) to shuffle the channel dimension features obtained from different convolutional approaches, thereby enhancing the robustness and generalization capabilities of our method. Experimental results demonstrate that our LE-IRSTD method achieves compelling results in both accuracy and lightweight performance, outperforming several state-of-the-art deep learning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。