arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.04476cs.CV

迈向基于持续预训练的地理空间基础模型

Towards Geospatial Foundation Models via Continual Pretraining

University of Central Florida(中佛罗里达大学);Amazon Web Services(亚马逊云科技);Boson AI(玻色人工智能)

Matias Mendieta, Boran Han, Xingjian Shi, Yi Zhu, Chen Chen

AI总结:

本文提出一种基于持续预训练的多目标范式,利用ImageNet-22k模型并构建GeoPile数据集,以低资源成本构建高效地理空间基础模型,在七个下游数据集上超越现有方法。

摘要:

地理空间技术在我们的世界中正变得越来越重要,广泛应用于农业、城市规划和灾害响应等领域。为了提高深度学习模型在这些地理空间任务上的适用性和性能,各种工作已开始研究该领域的基础模型。研究人员探索了两种主要方法来引入此类模型到地理空间应用中,但两者都存在性能提升有限或训练成本过高的问题。因此,在本工作中,我们提出了一种新颖的范式,以最小的资源成本和碳影响构建高效的地理空间基础模型。我们首先从多个来源构建一个紧凑但多样化的数据集,以促进特征多样性,我们称之为GeoPile。然后,我们研究了从大规模ImageNet-22k模型进行持续预训练的潜力,并提出了一种多目标持续预训练范式,该范式利用ImageNet的强大表示,同时提供学习有价值的领域内特征的自由。在涵盖变化检测、分类、多标签分类、语义分割和超分辨率等任务的七个下游数据集上的广泛评估中,我们的方法优于先前最先进的地理空间预训练方法。

英文摘要:

Geospatial technologies are becoming increasingly essential in our world for a wide range of applications, including agriculture, urban planning, and disaster response. To help improve the applicability and performance of deep learning models on these geospatial tasks, various works have begun investigating foundation models for this domain. Researchers have explored two prominent approaches for introducing such models in geospatial applications, but both have drawbacks in terms of limited performance benefit or prohibitive training cost. Therefore, in this work, we propose a novel paradigm for building highly effective geospatial foundation models with minimal resource cost and carbon impact. We first construct a compact yet diverse dataset from multiple sources to promote feature diversity, which we term GeoPile. Then, we investigate the potential of continual pretraining from large-scale ImageNet-22k models and propose a multi-objective continual pretraining paradigm, which leverages the strong representations of ImageNet while simultaneously providing the freedom to learn valuable in-domain features. Our approach outperforms previous state-of-the-art geospatial pretraining methods in an extensive evaluation on seven downstream datasets covering various tasks such as change detection, classification, multi-label classification, semantic segmentation, and super-resolution.

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

↑