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

用于遥感中密集预测问题的监督与对比自监督域内表示学习

Supervised and Contrastive Self-Supervised In-Domain Representation Learning for Dense Prediction Problems in Remote Sensing

Ali Ghanbarzade, Hossein Soleimani

AI总结:

本文针对遥感密集预测问题中自然图像与遥感图像的领域差异,提出结合监督与自监督的域内表示学习方法,利用SimSiam算法在遥感数据上预训练,在语义分割和目标检测任务中取得最先进结果。

摘要:

近年来,卷积神经网络(CNN)在计算机视觉领域取得了显著进展。这些进步已被应用于遥感等其他领域,并显示出令人满意的结果。然而,缺乏大型标注数据集以及遥感问题固有的复杂性,使得训练深度CNN来解决密集预测问题变得困难。为了解决这一问题,ImageNet预训练权重被用作各种密集预测任务的起点。尽管这种迁移学习带来了改进,但自然图像与遥感图像之间的领域差异也限制了深度CNN的性能。另一方面,在过去两年中,从大量未标记图像中学习视觉表示的自监督学习方法得到了大幅发展。因此,在本文中,我们探索了以监督和自监督两种形式进行域内表示学习的有效性,以解决遥感与ImageNet数据集之间的领域差异问题。从遥感图像中获得的权重被用作解决语义分割和目标检测任务的初始权重,并取得了最先进的结果。对于自监督预训练,我们使用了SimSiam算法,因为它简单且不需要巨大的计算资源。从遥感图像中获取通用视觉表示的最具影响力的因素之一是预训练数据集。为了检验预训练数据集的影响,我们使用了同等规模的遥感数据集进行预训练。我们的结果表明,在自监督表示学习中使用具有高空间分辨率的数据集,能够在下游任务中带来高性能。

英文摘要:

In recent years Convolutional neural networks (CNN) have made significant progress in computer vision. These advancements have been applied to other areas, such as remote sensing and have shown satisfactory results. However, the lack of large labeled datasets and the inherent complexity of remote sensing problems have made it difficult to train deep CNNs for dense prediction problems. To solve this issue, ImageNet pretrained weights have been used as a starting point in various dense predictions tasks. Although this type of transfer learning has led to improvements, the domain difference between natural and remote sensing images has also limited the performance of deep CNNs. On the other hand, self-supervised learning methods for learning visual representations from large unlabeled images have grown substantially over the past two years. Accordingly, in this paper we have explored the effectiveness of in-domain representations in both supervised and self-supervised forms to solve the domain difference between remote sensing and the ImageNet dataset. The obtained weights from remote sensing images are utilized as initial weights for solving semantic segmentation and object detection tasks and state-of-the-art results are obtained. For self-supervised pre-training, we have utilized the SimSiam algorithm as it is simple and does not need huge computational resources. One of the most influential factors in acquiring general visual representations from remote sensing images is the pre-training dataset. To examine the effect of the pre-training dataset, equal-sized remote sensing datasets are used for pre-training. Our results have demonstrated that using datasets with a high spatial resolution for self-supervised representation learning leads to high performance in downstream tasks.

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

↑