0%

论文笔记:Similarity to Availability

关于《Similarity to Availability: Synthetic Data Assisted SAR Target Recognition via Global Feature Compensation》的论文笔记

论文去向

该论文被TAES2025收录

主要内容

核心创新点

本论文针对SAMPLE仿真实测数据集,基于当前SAR图像仿真数据(以CAD方法仿真获得)与实测数据相比质量不高,仿真与实测数据的统计分布具有明显差异这一问题,提出了改进GAN网络的GFC-Net网络,使该网络可以学习实测数据的全局信息和背景信息并以此提升输入的仿真数据的质量,在提升仿真图像与实测图像相似度的同时进一步提升了仿真图像的有效性,可以视为仿真图像的图像质量提升插件。

主要方法

作者借鉴GAN网络的思想设计了生成器(Generator)和鉴别器(Discriminator),对于生成器部分采用了UNet架构进行图像生成并将其中的卷积模块更换为Transformer模块并改进为W-MMA(window multihead mix-attention),使UNet可以更好地捕捉到全局信息。针对SAR实测图像背景信息提出了BFC(background feature compensation)模块;对于判别器则采用了PatchGAN网络和二进制判别器来获取最终的判别损失。网络的架构示意图如下。

6994907c74197863081df7dab3d8a562

BFC

在图像生成过程中,BFC模块的主要操作是将图像切割为多个blocks并在blocks之间建立跳跃连接(skip connections),以更好地区分出图像中的背景和目标。这些连接可以在仿真图像到实测图像的转换过程中转换low-level的纹理信息,同时仿真图像的原有语义信息,使转换后的图像不易失真。作者采用了swin transformer blocks(STBs)和UNet架构作为信息提取模块,以实现转换后的图像有更接近实测数据的背景纹理,UNet的主要优点在于使用更少的参数和训练数据便可以产生更高质量的分割掩膜。

Image 1 Image 2

W-MMA

尽管SAR实测数据和仿真数据在统计分布上有所不同,但二者存在结构特征的高度相似(例如图像中目标所在位置,和背景的语义关系),而注意力机制可以更好地学习到这些特征。W-MMA模块则是通过Transformer架构提高生成器对相关信息的学习效果。

W-MMA将C*H*W的特征图分割为M*M的小窗,然后在小窗上分别进行操作。一方面,计算两对应窗之间的交叉注意力值,另一方面将两对应窗进行拼接成2*M*M的特征窗图并计算其自注意力值。相应的结果会被拼接并投射为C*H*W矩阵格式。

对于输入的特征图$f_1$和$f_2$,M-WWA将其分割为各N个窗后首先计算$f_1$中各窗的自注意力值,计算公式如下:

接着计算$f_1$和$f_2$对应小窗的交叉注意力值,计算方法与上式相似但$W_Q$矩阵选自$f_1$而$W_K$与$W_V$矩阵选自$f_2$。此外,为进行多头操作,多头注意力函数会同时进行交叉注意力计算来获取不同位置和表征子空间的信息,对应计算公式如下:

以上两注意力结果分别包含了特征图本身的特征信息和不同特征图同位置的对应信息,分别记为$O_1$和$O_2$,将二者拼接并重新映射到输入特征图维度,即可获得W-MMA模块的最终结果。在W-MMA模块中,实测图像的特征作为q来计算与之最接近的仿真图像特征。将W-MMA替换W-MSA即获得新的CSTB(cross STB)模块,用于最终的GAN生成器中,相应结构如下。

4ec5d7dd68eebdeb3761912aeb9f9e83

GFC-Net

GFC-Net的具体网络架构如下图所示。对于其中的判别器,PatchGAN的作用为在patch尺度对图像进行惩罚判别,尝试却分N*N的图像patch是来自实测数据还是仿真数据,而二进制判别器更关注全局尺度。综合两者的判别器损失函数如下:

6742160bc472a198df6abeade970ee73

总结

该方法提出了使用改进的GAN网络提升仿真数据质量,使之在下游任务中有更好的表现,更加贴近于真实数据。对于结合SAR仿真数据的对抗样本设计也可以从此角度出发,通过引入类似的translator提升基于仿真数据的对抗样本对由真实数据训练得到的目标网络的攻击性能。