arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.18924cs.CVeess.IV

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

Swin2-MoSE: A New Single Image Super-Resolution Model for Remote Sensing

Department of Engineering and Architecture(工程与建筑系)

Leonardo Rossi, Vittorio Bernuzzi, Tomaso Fontanini, Massimo Bertozzi, Andrea Prati

AI总结:

本文提出Swin2-MoSE遥感超分辨率模型,通过增强混合专家和损失函数改进,在多个数据集上显著超越现有模型,并验证了其在语义分割中的潜力。

摘要:

由于当前光学和传感器技术的局限性以及更新它们的高成本,卫星的光谱和空间分辨率可能并不总是满足期望的要求。因此,遥感单图像超分辨率(RS-SISR)技术引起了广泛关注。在本文中,我们提出了Swin2-MoSE模型,它是Swin2SR的增强版本。我们的模型引入了MoE-SM,一种增强的混合专家(MoE)来替换所有Transformer块内的前馈网络。MoE-SM设计有Smart-Merger,这是一个用于合并各个专家输出的新层,并且采用了一种新的方式来在专家之间分配工作,定义了新的每示例策略,而不是常用的每令牌策略。此外,我们分析了位置编码之间的相互作用,证明了每通道偏置和每头偏置可以积极协作。最后,我们提出使用归一化互相关(NCC)和结构相似性指数(SSIM)损失的组合,以避免典型的MSE损失限制。实验结果表明,在2倍、3倍和4倍分辨率放大任务(Sen2Venus和OLI2MSI数据集)中,Swin2-MoSE在PSNR上优于任何基于Swin的模型,最高可达0.377-0.958 dB。它还以较大优势超越了最先进的模型,证明了其竞争力和巨大潜力,尤其是在复杂任务中。此外,还进行了计算成本的分析。最后,我们展示了Swin2-MoSE的有效性,将其应用于语义分割任务(SeasoNet数据集)。代码和预训练模型可在https://github.com/IMPLabUniPr/swin2-mose/tree/official_code获取。

英文摘要:

Due to the limitations of current optical and sensor technologies and the high cost of updating them, the spectral and spatial resolution of satellites may not always meet desired requirements. For these reasons, Remote-Sensing Single-Image Super-Resolution (RS-SISR) techniques have gained significant interest. In this paper, we propose Swin2-MoSE model, an enhanced version of Swin2SR. Our model introduces MoE-SM, an enhanced Mixture-of-Experts (MoE) to replace the Feed-Forward inside all Transformer block. MoE-SM is designed with Smart-Merger, and new layer for merging the output of individual experts, and with a new way to split the work between experts, defining a new per-example strategy instead of the commonly used per-token one. Furthermore, we analyze how positional encodings interact with each other, demonstrating that per-channel bias and per-head bias can positively cooperate. Finally, we propose to use a combination of Normalized-Cross-Correlation (NCC) and Structural Similarity Index Measure (SSIM) losses, to avoid typical MSE loss limitations. Experimental results demonstrate that Swin2-MoSE outperforms any Swin derived models by up to 0.377 - 0.958 dB (PSNR) on task of 2x, 3x and 4x resolution-upscaling (Sen2Venus and OLI2MSI datasets). It also outperforms SOTA models by a good margin, proving to be competitive and with excellent potential, especially for complex tasks. Additionally, an analysis of computational costs is also performed. Finally, we show the efficacy of Swin2-MoSE, applying it to a semantic segmentation task (SeasoNet dataset). Code and pretrained are available on https://github.com/IMPLabUniPr/swin2-mose/tree/official_code

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

↑