【跨层注意力：多层次融合】

Multi-level features fusion via cross-layer guided attention for hyperspectral pansharpening

&＃xff08;基于跨层注意力引导的多层次特征融合高光谱全色锐化&＃xff09;

近年来&＃xff0c;卷积神经网络&＃xff08;CNN&＃xff09;在计算机视觉中的成功应用引起了人们的广泛关注。特别地&＃xff0c;具有注意力机制的深度学习模型在高光谱&＃xff08;HS&＃xff09;全色锐化中表现出令人印象深刻的性能。然而&＃xff0c;这些现有模型大多遵循早/晚融合策略&＃xff0c;没有充分利用层次特征。为此&＃xff0c;设计了一种新的基于跨层注意力引导的端到端多级特征融合网络&＃xff08;HP-MFFN&＃xff09;&＃xff0c;该网络能够逐层提取层次特征。由于网络的不同层次具有不同的感受域和包含不同的细节&＃xff0c;利用跨层引导注意力融合模块&＃xff08;CLGAF&＃xff09;对从HS图像和全色图像中提取的层次特征进行细化&＃xff0c;得到细节精细、语义丰富的空谱特征。

介绍

高光谱图像是一个包含空间域和光谱域信息的三维数据立方体。它在军事、农业、环境等诸多领域显示出显著的优越性。然而&＃xff0c;由于传感器物理特性等因素的限制&＃xff0c;空间分辨率和光谱分辨率之间存在着关键的折衷。HS图像通常具有几十条甚至上百条连续条带&＃xff0c;空间分辨率较低&＃xff0c;极大地限制了其应用。因此&＃xff0c;现有的全色&＃xff08;PAN&＃xff09;传感器可以捕获具有高空间分辨率但低光谱分辨率的图像。因此&＃xff0c;PAN图像作为一种互补信息&＃xff0c;被用于融合同一场景的低分辨率HS&＃xff08;LRHS&＃xff09;图像&＃xff0c;以生成高分辨率HS&＃xff08;HRHS&＃xff09;图像。此过程称为全色锐化。在地物精细分类、目标检测和环境检测等许多领域都表现出了良好的性能。
已经开发了大量的HS全色锐化方法&＃xff0c;这些方法大致可以分为两类&＃xff0c;即&＃xff1a;传统算法和基于深度学习的算法。传统的算法可以进一步分为四类&＃xff1a;成分替代&＃xff08;CS&＃xff09;方法、矩阵分解方法、基于贝叶斯的方法和多分辨率分析&＃xff08;MRA&＃xff09;。第一类算法主要包括Gram-Schmidt&＃xff08;GS&＃xff09;、自适应GS&＃xff08;GSA&＃xff09;和主成分分析&＃xff08;PCA&＃xff09;这些方法通过采用逆变换用PAN图像代替HS图像的空间分量。然而&＃xff0c;这些方法可能导致一些明显的频谱失真。作为矩阵分解的代表&＃xff0c;耦合非负矩阵分解&＃xff08;CNMF&＃xff09;算法交替分解HS和多光谱图像&＃xff0c;以获得有利的端元和丰度矩阵。最近&＃xff0c;作为矩阵分解的高阶推广的张量分解已被广泛应用于HS图像处理&＃xff0c;例如HS图像分类、HS去噪和HS图像融合。例如&＃xff0c;Dian等人提出的非局部稀疏张量分解方法&＃xff08;NLSTF_SMBF&＃xff09;实现了HS图像和多光谱图像的融合。随后&＃xff0c;他们提出了一种基于低张量训练秩&＃xff08;LTTR&＃xff09;的融合算法&＃xff0c;其中通过LTTR先验学习非局部相似HS图像立方体的空间、光谱和非局部模式之间的相关性。“…剩下的略…”

总之&＃xff0c;一些学者的工作已经证明&＃xff0c;来自多个层的特征有利于生成更好的结果。网络越深&＃xff0c;接受域越大&＃xff0c;这是一个被广泛接受的真理。具有小感受野的低级特征保留了高分辨率重建的细节&＃xff0c;而具有大感受野的高级特征提供了图像区域类别识别的语义知识。
本文提出了一种端到端的多层次特征融合网络&＃xff08;HP-MFFN&＃xff09;&＃xff0c;旨在挖掘尽可能多的不同层次的细节特征&＃xff0c;更有效地传递信息。该方法基于两重性提出。一方面&＃xff0c;HP-MFFN能够重点提取每一层的PAN和HS细节&＃xff0c;并基于注意力机制逐步整合这些提取的特征&＃xff0c;以减轻直接重构高分辨率细节HS图像的困难。另一方面&＃xff0c;采用空间-频谱关注机制进一步细化每一层的信息&＃xff0c;提高特征表示能力。

贡献

1&＃xff09;提出了一种双分支多级特征聚合网络HP-MFFN&＃xff0c;该网络有效地提取不同层次的层次特征&＃xff0c;提高了全色锐化的性能。

2&＃xff09;受网络深度不同、感受野不同的启发&＃xff0c;设计了跨层引导注意力融合模块&＃xff08;CLGAF&＃xff09;&＃xff0c;生成细节精细、语义丰富的精细化空谱特征,在每个CLGAF之间&＃xff0c;通过密集连接形成连续的存储机制&＃xff0c;以减轻一步恢复详细HS图像的难度。

3&＃xff09;与基于自然图像任务注意机制的HS全色锐化方法不同&＃xff0c;本文提出了一种全局-局部空间-谱注意机制&＃xff08;GL-PySSAM&＃xff09;&＃xff0c;该机制由局部金字塔通道注意&＃xff08;LPCA&＃xff09;和全局金字塔空间注意&＃xff08;GPSA&＃xff09;组成&＃xff0c;用于多核选择建模。GLPySSAM通过使用具有多个核的通道关注来局部地维持光谱相似性&＃xff0c;并使用多个核来全局地捕获不同级别的对象信息&＃xff0c;从而有助于产生更有效的空间-光谱特征。

相关工作

Method overview

设PAN ∈





R



1


×


M


×


N





R^{1×M×N}


R1×M×N是可用的PAN图像&＃xff0c;大小为M×N的。设H_l∈





R



C


×


m


×


n





R^{C×m×n}


RC×m×n表示观察到的LRHS图像&＃xff0c;其中C表示频带的数目&＃xff0c;m×n是每个带上的像素数。让H_u∈





R



C


×


M


×


N





R^{C×M×N}


RC×M×N表示上采样HS图像&＃xff0c;其从LRHS H_l被内插到与输入PAN图像相同的大小&＃xff0c;并且H∈





R



C


×


M


×


N





R^{C×M×N}


RC×M×N表示期望的HS图像。

设计了高光谱全色锐化网络&＃xff0c;利用PAN和H_u对HS图像H进行复原。大多数现有的基于深度学习的现有技术水平的全色锐化算法可以大致分为两类&＃xff1a;1&＃xff09;早期融合算法&＃xff0c;2&＃xff09;后期融合算法&＃xff0c;其可以被描述为
然而&＃xff0c;早期的融合全色锐化方法是将PAN图像和预插值后的HRHS图像连接起来作为输入直接送入网络&＃xff0c;未能充分利用PAN图像复杂的空间信息。对于后期融合全色锐化方法&＃xff0c;这些高级模型主要集中于分别从PAN图像和HS图像中的深卷积层提取的高级特征的非线性组合。这些方法的输出结果常常由于缺乏低级信息&＃xff08;例如&＃xff0c;对象边缘和细节&＃xff09;。

由具有小区域的浅卷积层生成的低层特征映射适合捕获输入图像的精细细节&＃xff0c;这一点被广泛接受。随着网络的卷积层增加&＃xff0c;每个卷积层中的特征将具有不同的感受域。CNN的本质是当前卷积层的每个像素是前一卷积层的感受野中所有像素的加权平均。CNN可以通过增加网络深度获得更大的感受野。因此&＃xff0c;不同层次的卷积层可以捕获具有不同外观和形状的物体。在此基础上&＃xff0c;提出了HPMFFN算法&＃xff0c;分别用于提取HS图像和PAN图像的潜在多层次特征。HP-MFFN的流程图如图1所示。

HP-MFFN通过提取每一层的多层次特征&＃xff0c;保证了融合结果的多样性&＃xff0c;为生成满意的融合结果提供了更多的空间细节信息。简单地说&＃xff0c;这个过程可以用公式表示为&＃xff1a;

The Architecture of HP-MFFN

拟议的HP-MFFN由三个部分组成&＃xff1a;两分支多层次特征提取模块、基于空谱关注度的多层次特征融合模块和特征重构模块。为了保证双分支网络结构的高度相似性&＃xff0c;HP-MFFN将预插值后的HRHS图像和PAN图像分别作为双分支的输入。HS图像分支开始于卷积层&＃xff0c;其被采用以从预内插HS图像H_u提取初始浅层表示。
CBs跟随第一卷积层。CB由两个卷积层和两个整流线性单元&＃xff08;ReLU&＃xff09;激活层组成。针对深度学习中“网络越深&＃xff0c;感受野越大”的特点&＃xff0c;提出的HP-MFFN采用3个CBs来提取多层次特征。在第一个convblock和最后一个convblock之间添加剩余连接&＃xff0c;以避免以下公式的梯度色散问题表示HS图像分支中第N个convblock的操作。PAN分支也由用于浅层特征提取的初始卷积层和用于多层特征提取的三个CBs组成。从PAN图像PAN获得的初始特征表示可以用公式表示为。
然后&＃xff0c;利用CBs挖掘PAN图像的潜在多层次特征。同样地&＃xff0c;残差连接也被嵌入到多级特征提取中。这个过程可以写成:

在获得PAN和HS图像不同层次的特征后&＃xff0c;如何有效地融合层次特征是一个值得考虑的问题。现有的基于深度学习的HS全色锐化算法大多假设提取的特征具有相同的重要性。因此&＃xff0c;针对上述问题&＃xff0c;如图1所示&＃xff0c;我们提出了一种**新的跨层引导注意力融合模块&＃xff08;CLGAF&＃xff09;**来融合层次特征,它可以表示为。

然后&＃xff0c;我们采用两个卷积层和一个ReLU激活函数来重建一个满意的HRHS图像H。该过程用以下公式表示。

The Architecture of CLGAF

近年来&＃xff0c;人们发现人类视觉系统的一个重要特性是每个人都能自动聚焦于感兴趣的区域而不是聚焦于整个场景。这种科学机制被称为注意机制&＃xff0c;以往的文献对此进行了广泛的研究。如图1所示&＃xff0c;本文采用了一种基于注意机制的多层次特征融合模型CLGAF&＃xff0c;利用每一层的所有早期细化特征进行特征增强。在每个CLGAF之间&＃xff0c;采用密集连接形成连续记忆机制&＃xff0c;提高了空间-光谱特征的有效传递。具体地说&＃xff0c;在网络的每一层&＃xff0c;我们利用由局部金字塔通道注意力&＃xff08;LPCA&＃xff09;和全局金字塔空间注意力&＃xff08;GPSA&＃xff09;组成的全局-局部注意力模块&＃xff08;GL-PySSAM&＃xff09;来获得初始的精细特征。考虑到不同感受野的每一层特征所包含的信息不同&＃xff0c;我们创新性地采用了层叠GL-PySSAM算法&＃xff0c;利用早期提取的层次特征进一步增强初始细化特征&＃xff0c;以捕获更详细的信息&＃xff0c;促进更有效的信息流动。这个过程被称为跨层引导注意力融合。与传统的特征融合策略直接采用逐元素或串接运算不同&＃xff0c;基于注意力的特征融合策略可以直接关注图像的高亮区域&＃xff0c;提高特征的表示能力。设





F



F


u





F^{Fu}


FFu_K表示





F



H


S


I





F^{HSI}


FHSI_K和





F



P


A


N





F^{PAN}


FPAN_K的和&＃xff0c;





F



H


S


I





F^{HSI}


FHSI_K和





F



P


A


N





F^{PAN}


FPAN_K分别表示第K层提取的特征。E_K表示CLGAF在第K层的输出。以第K层的CLGAF为例&＃xff0c;可表示为

The Architecture of GL-PySSAM

虽然现有的注意模块在视觉图像处理中表现出了良好的性能&＃xff0c;但是将这些注意机制直接转移到与HS图像处理相关的任务中可能会产生一些问题。例如&＃xff0c;传统的通道注意力&＃xff08;CA&＃xff09;算法直接压缩输入特征图的空间维度来计算通道注意力图&＃xff0c;而使用通道注意力图来细化整体特征不可避免地破坏了HS图像的原始局部谱向量。同时&＃xff0c;传统的空间注意&＃xff08;SA&＃xff09;模块采用核大小为3×3的卷积层或7×7生成空间注意掩模时&＃xff0c;未能充分利用上下文感知信息&＃xff0c;导致局部细节纹理平滑。因此&＃xff0c;提出了LPCA和GPSA算法&＃xff0c;有效缓解了传统注意带来的问题&＃xff0c;进一步提高了全色锐化的性能。在GL-PySSAM中&＃xff0c;我们利用LPCA和GPSA形成复合注意掩码&＃xff0c;以便在网络中有选择地传递更有效的信息。

The Architecture of LPCA

LPCA是局部金字塔渠道关注的对象。基于同一HS图像中的相邻像素通常是相关或相关的概念&＃xff0c;如图2所示&＃xff0c;我们通过使用具有不同大小核的最大池和平均池&＃xff0c;在局部区域采用通道关注&＃xff0c;以保持局部光谱矢量相似性&＃xff0c;并在通道维度上为每个局部区域定制不同的关注水平。
我们首先整合输入特征地图F2RC的局部空间信息通过采用平均池化和和最大池操作在四个尺度3×3,5×5,7×7、9×9和步幅分别为3、5、7和9&＃xff0c;其输出可分别表示为





F



k


×


k





F^{k×k}


Fk×k_Lavg和





F



k


×


k




 


L


m


a


x


 


&＃xff0c;


k


∈



3


&＃xff0c;


5


&＃xff0c;


7


&＃xff0c;


9



&＃xff0c;



F^{k×k}~Lmax~&＃xff0c;k∈{3&＃xff0c;5&＃xff0c;7&＃xff0c;9}&＃xff0c;


Fk×k Lmax &＃xff0c;k∈3&＃xff0c;5&＃xff0c;7&＃xff0c;9&＃xff0c;F^{{k×k}$~Lavg~和$F}{k×k}_Lmax将四个尺度转发到每个共享网络以生成四个局部信道注意力图M_Lavgc∈





R



C


×


(


(


M


−


k


)


)


/


/


k


&＃43;


1


)


×


(


(


N


−


k


)


/


/


k


&＃43;


1


)





R^{C×((M-k))//k&＃43;1)×((N-k)//k&＃43;1)}


RC×((M−k))//k&＃43;1)×((N−k)//k&＃43;1)&＃xff0c;k∈{3,5,7,9}和M_Lmaxc∈





R



C


×


(


(


M


−


k


)


)


/


/


k


&＃43;


1


)


×


(


(


N


−


k


)


/


/


k


&＃43;


1


)





R^{C×((M-k))//k&＃43;1)×((N-k)//k&＃43;1)}


RC×((M−k))//k&＃43;1)×((N−k)//k&＃43;1),k∈{3,5,7,9}。具有一个隐层的多层感知器&＃xff08;MLP&＃xff09;是每个尺度的共享网络的组成部分。为了最小化参数开销&＃xff0c;隐层参数设置为





R



C


/


r


×


1


×


1





R^{C/r×1×1}


RC/r×1×1 ,其中r是缩减比&＃xff0c;1是核大小。然后&＃xff0c;我们合并输出特征图





F



k


×


k





F^{k×k}


Fk×k_Lavgc。
分别使用元素方式求和





F



k


×


k





F^{k×k}


Fk×k_Lmaxc的四个标度。这个过程可以写成

为了融合4个尺度的局部信道关注度图&＃xff0c;采用最近插值的方法保持局部区域的关注度权重不变。通过Sigmoid激活函数分别对各尺度的注意力图进行插值融合&＃xff0c;得到最终的注意力图F_LC。简而言之&＃xff0c;LPCA的计算公式为

The Architecture of GPSA

GPSA是全局金字塔空间注意力。基于同一对象类别在同一幅图像中可以出现在不同尺度上的知识&＃xff0c;如图3所示&＃xff0c;我们采用了具有四个层次的不同类型核的空间注意力来捕捉不同层次的细节&＃xff0c;使特征更加锐化。
输入特性的信道信息映射F是第一聚合特性映射采用maxpooling和avgpooling,分别生产两种特性表示:
为了获得更精细的空间注意力图&＃xff0c;我们采用金字塔结构对输入进行多尺度解析。





F


s




F^s


Fs_avg∈





R



1


×


H


×


W





R^{1×H×W}


R1×H×W和





F


s




F^s


Fs_max∈





R



1


×


H


×


W





R^{1×H×W}


R1×H×W由金字塔核大小分别为3×3,5×5,7×7,9×9进行级联的卷积&＃xff0c;然后在Sigmoid函数之后生成2D空间注意力图。简而言之&＃xff0c;这个过程可以用公式表示为

Loss function

{





H



(


i


)





H^{(i)}


H(i)_u、





P



(


i


)





P^{(i)}


P(i)&＃xff0c;





R



(


i


)





R^{(i)}


R(i)}^N_i&＃61;1表示训练样本&＃xff0c;其中N表示样本数。





H



(


i


)





H^{(i)}


H(i)_u、





P



(


i


)





P^{(i)}


P(i)和





R



(


i


)





R^{(i)}


R(i)分别代表第i个样本的输入预插值HRHS图像、输入PAN图像和参考HS图像。一些先前的HS超分辨率任务已经证明&＃xff0c;均方误差&＃xff08;称为L2损失&＃xff09;可能产生过平滑结果&＃xff0c;因为平方惩罚在图像边缘处不起作用。
因此,我们采用损失函数的平均绝对误差(称为L1损失),广泛应用于自然超分辨率商品形象的任务,导致微妙的细节。Pavia Center dataset的训练损失曲线图4所示。的损失函数HP-MFFN可以写成。