用热扩散原理增强点标注,提升红外小目标检测精度与效率
Diffuse to Detect: Bi-Level Sample Rebalancing with Pseudo-Label Diffusion for Point-Supervised Infrared Small-Target Detection

- 基于热扩散物理规律,将单点标注扩展为可靠伪掩码
- 实现五倍标注加速,仅用30%数据达到接近全量数据性能
- 适合标注成本高、样本不平衡的红外小目标检测场景
点标注已成为解决红外小目标检测中密集标注难题的可扩展方案,但其性能受限于两个耦合瓶颈:复杂低对比度红外图像中伪标签演化不稳定,以及严重的样本分布不平衡。本文提出一种更自适应、更稳定的框架以解决上述问题。利用热辐射模式与热扩散的内在一致性,我们设计了一种物理启发的标注策略,将单点标签扩展为可靠的伪掩码。为进一步增强监督并缓解样本不平衡,我们构建了双层双更新框架,联合优化检测器权重、样本权重与扩散参数。元分类器动态预测样本级损失权重,可微扩散模块结合检测反馈精化伪标签,实现训练与超参数优化的自适应交互。在多个数据集上的大量实验表明,本方法实现五倍标注加速,检测精度更优,且仅使用30%训练数据即达到与全量数据相当的性能,验证了方法的高效性与实用性。代码已公开于https://github.com/yuanhang-yao/diffuse-to-detect。
原文摘要 · Abstract (English)
Point supervision has become a scalable solution to address dense annotation for infrared small target detection, but its performance is limited by two coupled bottlenecks: unstable pseudo-label evolution in cluttered, low-contrast infrared imagery and severe sample-distribution imbalance. In this paper, we present a more adaptive and stable framework to address these issues. Leveraging the intrinsic consistency between thermal radiation patterns and heat diffusion, we propose a physics-induced annotation strategy that expands single-point labels into reliable pseudo-masks. To further enhance supervision and alleviate sample imbalance, we develop a bi-level dual-update framework that jointly optimizes detector weights, sample weights, and diffusion parameters. A meta-classifier dynamically predicts sample-wise loss weights, while a differentiable diffusion module refines pseudo-labels with detection feedback, enabling adaptive interaction between training and hyperparameter optimization. Extensive experiments across multiple datasets demonstrate five-fold annotation acceleration, superior detection accuracy, and comparable performance with 30% of the training data, validating the efficiency and practicality of our approach. Our code is available at https://github.com/yuanhang-yao/diffuse-to-detect.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。