Aug, 2023

基于种子特征图的卫星遥感服务的 CNN 模型

TL;DR在低地球轨道卫星上部署高性能卷积神经网络模型(CNN)进行快速遥感图像处理引起了工业界和学术界的极大兴趣,不过,LEO 卫星上有限的资源与资源密集型 CNN 模型的需求相矛盾,因此需要采用地面站服务器的协助来训练和更新这些模型。本文介绍了一种地面站服务器辅助框架,其中 CNN 模型的每一层只包含一个学习特征图(称为种子特征图),其他特征图是基于特定规则从种子特征图生成的,而这些规则的超参数是随机生成的,而不是通过训练得到,从而实现了从种子特征图生成多个特征图并显著减少 FLOPs 的目的。此外,由于可以使用少量随机种子保存随机超参数,可以便于地面站服务器在更新部署在 LEO 卫星上的 CNN 模型。在 ISPRS Vaihingen、ISPRS Potsdam、UAVid 和 LoveDA 等数据集上进行的语义分割服务的实验结果表明,所提出的框架优于现有的最先进方法。特别地,在 UAVid 数据集上,基于 SineFM 的模型相比 UNetFormer 模型在 mIoU 上实现了较高的性能,且参数数量减少了 3.3 倍,FLOPs 减少了 2.2 倍。