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

FuseFormer:一种用于视觉与热成像图像融合的Transformer

FuseFormer: A Transformer for Visual and Thermal Image Fusion

Middle East Technical University(中东技术大学)

Aytekin Erdogan, Erdem Akagündüz

AI总结:

本研究提出FuseFormer,一种基于Transformer的多尺度图像融合方法,结合CNN与Transformer提取并融合局部与全局特征,采用两阶段训练并设计新损失函数以克服传统指标偏向,在多个基准上表现优越。

摘要:

由于图像融合问题缺乏明确的真实标签,损失函数基于评估指标构建,例如结构相似性指数测量(SSIM)。然而,这样做会引入对SSIM的偏向,进而导致对输入可见光波段图像的偏向。本研究的目标是提出一种用于图像融合问题的新颖方法,以缓解使用经典评估指标作为损失函数所带来的局限性。我们的方法整合了基于Transformer的多尺度融合策略,能够熟练处理局部和全局上下文信息。这种整合不仅优化了图像融合过程的各个独立组件,还显著提升了该方法的整体效能。我们提出的方法遵循两阶段训练方法,其中自编码器在第一阶段被训练以在多个尺度上提取深度特征。在第二阶段,我们整合了融合模块并如上所述更改了损失函数。多尺度特征使用卷积神经网络(CNN)和Transformer的组合进行融合。CNN用于捕获局部特征,而Transformer负责整合全局上下文特征。通过对各种基准数据集进行大量实验,我们提出的方法连同新颖的损失函数定义,与其他有竞争力的融合算法相比展现出了优越的性能。

英文摘要:

Due to the lack of a definitive ground truth for the image fusion problem, the loss functions are structured based on evaluation metrics, such as the structural similarity index measure (SSIM). However, in doing so, a bias is introduced toward the SSIM and, consequently, the input visual band image. The objective of this study is to propose a novel methodology for the image fusion problem that mitigates the limitations associated with using classical evaluation metrics as loss functions. Our approach integrates a transformer-based multi-scale fusion strategy that adeptly addresses local and global context information. This integration not only refines the individual components of the image fusion process but also significantly enhances the overall efficacy of the method. Our proposed method follows a two-stage training approach, where an auto-encoder is initially trained to extract deep features at multiple scales in the first stage. For the second stage, we integrate our fusion block and change the loss function as mentioned. The multi-scale features are fused using a combination of Convolutional Neural Networks (CNNs) and Transformers. The CNNs are utilized to capture local features, while the Transformer handles the integration of general context features. Through extensive experiments on various benchmark datasets, our proposed method, along with the novel loss function definition, demonstrates superior performance compared to other competitive fusion algorithms.

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

↑