Apr, 2024

Swin2-MoSE:一种新的遥感单幅图像超分辨率模型

TL;DR我们提出了 Swin2-MoSE 模型,它是 Swin2SR 的增强版本,引入 MoE-SM(增强的专家混合编码器)来替换 Transformer 模块中的前馈网络,并通过智能合并、新的输出融合层和新的工作分配策略,改进了专家之间的合作。同时,我们分析了位置编码之间的相互作用,证明了通道偏差和头部偏差的正向合作效果。最后,我们提出采用标准化互相关(NCC)和结构相似性指数(SSIM)损失的组合,以克服常规均方损失的局限性。实验结果表明,在 2 倍、3 倍和 4 倍分辨率放大任务上,Swin2-MoSE 在 Sen2Venus 和 OLI2MSI 数据集上相对于现有最佳方法的 PSNR 提高了 0.377~0.958 dB。我们还展示了 Swin2-MoSE 在语义分割任务(SeasoNet 数据集)中的有效性。代码和预训练模型可在给出的 URL 中获取。