arXiv:2606.21956cs.CV2026-06中稿 · ECCV

通过注意力引导的知识蒸馏,提升红外小目标检测精度与效率

Denoising-Enhanced Coarse-to-Fine Infrared Small Target Detection with Attention Prior-Guided Knowledge Distillation

论文配图:Denoising-Enhanced Coarse-to-Fine Infrared Small Target Detection with Attention Prior-Guided Knowledge Distillation
图 1 · 摘自论文原文
  • 分两阶段检测:先粗筛含目标区域,再精检,降低冗余计算
  • 引入去噪训练策略,使模型更清晰区分目标与复杂背景
  • 知识蒸馏聚焦关键目标区域,适合实时无人机监控场景

高分辨率红外图像中的小目标检测在无人机监视和地基监测等应用中至关重要。然而,由于目标尺寸小、特征弱,且存在复杂动态背景干扰,检测仍具挑战性。现有方法常在非目标区域进行冗余计算,且对目标上下文信息利用不足。为此,提出一种高效的细粒度红外小目标检测框架ECFNet。在粗略阶段,基于网格的多尺度特征图设计区域二值分类网络(RBCN),高效识别含目标的上下文区域提案;创新性地引入去噪辅助训练策略,将带噪真实标签掩码注入RBCN特征图,并通过去噪任务训练网络重建原始标签,促使模型显式学习目标-背景上下文关系,更好区分目标与背景。在精细阶段,针对粗阶段的区域提案定制轻量级检测器,兼顾精度与效率;进一步提出由教师-学生交叉注意力先验引导的知识蒸馏策略,引导学生关注关键目标区域,增强判别性特征表示。在三个真实红外数据集上的大量实验表明,该方法优于现有单阶段与双阶段方法,同时保持高实时处理效率。

原文摘要 · Abstract (English)

Infrared small target detection (IRSTD) in high-resolution images is crucial for many practical applications, such as surveillance of unmanned aerial vehicles (UAVs) and UAV-based ground monitoring. However, IRSTD remains challenging due to the small size and weak features of targets, as well as significant interference from complex dynamic backgrounds. Existing detection methods often suffer from redundant computations on non-target background regions and insufficient exploitation of target context information, which limits their performance in complex backgrounds. To address these issues, we propose an efficient coarse-to-fine infrared small target detection framework with attention prior-guided knowledge distillation, termed ECFNet. In the coarse stage, we design a region binary classification network (RBCN) on grid-based multi-scale feature maps to efficiently recognize target-containing context region proposals. Moreover, we introduce a novel denoising-assisted training strategy that incorporates noisy ground-truth (GT) masks into RBCN feature maps and trains the network to reconstruct the original GT masks through a denoising task, thereby encouraging it to explicitly learn target-background context and thus better distinguish target proposals from background regions. In the fine stage, we customize a lightweight target detector to the coarse stage's region proposals for balancing accuracy and efficiency. Furthermore, we propose a knowledge distillation strategy guided by the teacher-student cross-attention prior. This mechanism directs the student to focus on critical target regions, thereby enhancing the discriminative feature representation for infrared small targets. Extensive experiments on three real infrared datasets demonstrate that our method outperforms both existing single-stage and two-stage approaches while maintaining high real-time processing efficiency.

红外检测小目标知识蒸馏无人机

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