用视觉大模型提升红外小目标检测的稳定性和精度
Learning with Semantic Priors: Stabilizing Point-Supervised Infrared Small Target Detection via Hierarchical Knowledge Distillation

- 通过分层知识蒸馏,利用冻结的视觉大模型指导轻量CNN训练
- 在多个数据集上实现检测准确率提升,训练过程更稳定
- 适合做红外小目标检测且缺乏标注数据的研究者使用
单帧红外小目标检测(ISTD)旨在复杂背景中定位微弱目标,但密集像素级标注成本高昂。点监督结合在线标签演化可降低标注负担;然而,轻量级CNN检测器语义能力不足,导致伪掩码噪声大、优化不稳定。为此,我们提出一种分层VFM驱动的知识蒸馏框架,在训练中使用冻结的视觉基础模型(VFM)。将点监督学习建模为双层优化:内层循环基于重加权样本适配嵌入VFM的教师模型,外层循环将验证引导的知识传给轻量学生模型,以缓解伪标签噪声和训练集偏差。进一步引入语义条件仿射调制(SCAM),在多层注入VFM语义信息。同时,采用聚类级样本重加权的动态协同学习策略,增强对不完美伪掩码的鲁棒性。在多个主流ISTD骨干网络和多样挑战场景下的实验表明,该方法在检测精度与训练稳定性方面均有持续提升。代码已开源:https://github.com/yuanhang-yao/semantic-prior。
原文摘要 · Abstract (English)
Single-frame Infrared Small Target Detection (ISTD) aims to localize weak targets under heavy background clutter, yet dense pixel-wise annotations are expensive. Point supervision with online label evolution reduces annotation cost; however, lightweight CNN detectors often lack sufficient semantics, leading to noisy pseudo-masks and unstable optimization. To address this, we propose a hierarchical VFM-driven knowledge distillation framework that uses a frozen Vision Foundation Model (VFM) during training. We formulate point-supervised learning as a bilevel optimization process: the inner loop adapts a VFM-embedded teacher on reweighted training samples, while the outer loop transfers validation-guided knowledge to a lightweight student to mitigate pseudo-label noise and training-set bias. We further introduce Semantic-Conditioned Affine Modulation (SCAM) to inject VFM semantics into CNN features at multiple layers. In addition, a dynamic collaborative learning strategy with cluster-level sample reweighting enhances robustness to imperfect pseudo-masks. Experiments on diverse challenging cases across multiple ISTD backbones demonstrate consistent improvements in detection accuracy and training stability. Our code is available at https://github.com/yuanhang-yao/semantic-prior.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。