提出基于流形的注意力机制,提升红外可见光图像融合效果
GrFormer: A Novel Transformer on Grassmann Manifold for Infrared and Visible Image Fusion
- 在流形空间构建低秩子空间映射,分离高频细节与低频语义
- 在多个基准上超越当前最优方法,显著提升融合质量
- 适合图像融合、多模态视觉研究者参考
在图像融合领域,通过将不同模态的数据建模为线性子空间已取得显著进展。然而,实际中源图像常位于非欧几里得空间,传统欧氏方法难以捕捉其内在拓扑结构。欧氏空间中的内积计算的是代数相似性而非语义相似性,导致注意力输出不理想,降低融合性能。在红外与可见光图像融合任务中,需兼顾低层细节与高层语义的平衡。为此,本文提出一种基于格拉斯曼流形的新型注意力机制(GrFormer)。具体地,通过在格拉斯曼流形上的投影约束构建低秩子空间映射,将注意力特征压缩至不同秩级别的子空间中,迫使特征解耦为高频细节(局部低秩)和低频语义(全局低秩),实现多尺度语义融合。此外,设计基于协方差掩码的跨模态融合策略(CMS),最大化不同模态间的互补性,抑制高相关性冗余特征。实验结果表明,该网络在多个图像融合基准上均优于现有最先进方法。代码已开源:https://github.com/Shaoyun2023。
原文摘要 · Abstract (English)
In the field of image fusion, promising progress has been made by modeling data from different modalities as linear subspaces. However, in practice, the source images are often located in a non-Euclidean space, where the Euclidean methods usually cannot encapsulate the intrinsic topological structure. Typically, the inner product performed in the Euclidean space calculates the algebraic similarity rather than the semantic similarity, which results in undesired attention output and a decrease in fusion performance. While the balance of low-level details and high-level semantics should be considered in infrared and visible image fusion task. To address this issue, in this paper, we propose a novel attention mechanism based on Grassmann manifold for infrared and visible image fusion (GrFormer). Specifically, our method constructs a low-rank subspace mapping through projection constraints on the Grassmann manifold, compressing attention features into subspaces of varying rank levels. This forces the features to decouple into high-frequency details (local low-rank) and low-frequency semantics (global low-rank), thereby achieving multi-scale semantic fusion. Additionally, to effectively integrate the significant information, we develop a cross-modal fusion strategy (CMS) based on a covariance mask to maximise the complementary properties between different modalities and to suppress the features with high correlation, which are deemed redundant. The experimental results demonstrate that our network outperforms SOTA methods both qualitatively and quantitatively on multiple image fusion benchmarks. The codes are available at https://github.com/Shaoyun2023.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。