arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.00632cs.CV

Diff-Mosaic:基于扩散先验增强红外小目标检测中的真实表示

Diff-Mosaic: Augmenting Realistic Representations in Infrared Small Target Detection via Diffusion Prior

Yukai Shi, Yupei Lin, Pengxu Wei, Xiaoyu Xian, Tianshui Chen, Liang Lin

AI总结:

提出基于扩散模型的数据增强方法Diff-Mosaic,通过Pixel-Prior网络和Diff-Prior策略两阶段处理,有效提升红外小目标检测数据集的多样性与真实感,显著增强检测网络性能。

摘要:

最近,研究人员提出了各种深度学习方法来准确检测具有形状和纹理不清晰特征的红外目标。由于红外数据集的多样性有限,训练具有良好泛化能力的深度学习模型面临挑战。为了扩充红外数据集,研究人员采用数据增强技术,这通常涉及通过组合来自不同数据集的图像来生成新图像。然而,这些方法在两个方面存在不足。在真实感方面,基于mixup的方法生成的图像缺乏真实感,难以有效模拟复杂的现实场景。在多样性方面,与现实场景相比,从另一个数据集借用知识本质上具有有限的多样性。目前,扩散模型作为一种创新的生成方法脱颖而出。大规模训练的扩散模型具有强大的生成先验,能够对图像进行现实世界建模,从而生成多样且逼真的图像。在本文中,我们提出了Diff-Mosaic,一种基于扩散模型的数据增强方法。该模型通过扩散先验有效缓解了数据增强方法在多样性和真实感方面的挑战。具体而言,我们的方法包含两个阶段。首先,我们引入了一个名为Pixel-Prior的增强网络,通过协调像素来生成高度协调且逼真的Mosaic图像。在第二阶段,我们提出了一种名为Diff-Prior的图像增强策略。该策略利用扩散先验对现实场景中的图像进行建模,进一步增强了图像的多样性和真实感。大量实验表明,我们的方法显著提高了检测网络的性能。代码可在https://github.com/YupeiLin2388/Diff-Mosaic获取

英文摘要:

Recently, researchers have proposed various deep learning methods to accurately detect infrared targets with the characteristics of indistinct shape and texture. Due to the limited variety of infrared datasets, training deep learning models with good generalization poses a challenge. To augment the infrared dataset, researchers employ data augmentation techniques, which often involve generating new images by combining images from different datasets. However, these methods are lacking in two respects. In terms of realism, the images generated by mixup-based methods lack realism and are difficult to effectively simulate complex real-world scenarios. In terms of diversity, compared with real-world scenes, borrowing knowledge from another dataset inherently has a limited diversity. Currently, the diffusion model stands out as an innovative generative approach. Large-scale trained diffusion models have a strong generative prior that enables real-world modeling of images to generate diverse and realistic images. In this paper, we propose Diff-Mosaic, a data augmentation method based on the diffusion model. This model effectively alleviates the challenge of diversity and realism of data augmentation methods via diffusion prior. Specifically, our method consists of two stages. Firstly, we introduce an enhancement network called Pixel-Prior, which generates highly coordinated and realistic Mosaic images by harmonizing pixels. In the second stage, we propose an image enhancement strategy named Diff-Prior. This strategy utilizes diffusion priors to model images in the real-world scene, further enhancing the diversity and realism of the images. Extensive experiments have demonstrated that our approach significantly improves the performance of the detection network. The code is available at https://github.com/YupeiLin2388/Diff-Mosaic

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑