arXiv:2410.19965cs.CVcs.AI2024-10被引 18

用百亿参数训练高分辨率遥感模型,突破行业规模瓶颈。

OReole-FM: successes and challenges toward billion-parameter foundation models for high-resolution satellite imagery

  • 基于前沿超算与高分辨率数据,实现百亿参数遥感模型预训练
  • 验证了数据量扩展对模型性能提升的关键作用
  • 公开数据集与模型,助力学术界高效复现与研究

尽管遥感图像领域基础模型(FMs)的预训练日益增多,现有模型参数量仍局限于数亿级别。将模型扩展至百亿参数已被证明能带来前所未有的能力涌现,但需要大规模数据和计算资源,通常仅限于工业研发实验室。本文结合美国首台埃克斯级超算Frontier与高分辨率光学遥感数据,成功实现了百亿级模型的预训练。研究评估了多种视觉变换器变体在图像分类、语义分割和目标检测任务上的表现,凸显了数据规模化对模型有效扩增的重要性。此外,本文构建了新型TIU预训练数据集,讨论了模型初始化策略,并计划公开数据与预训练模型。通过剖析技术挑战与文献中常被忽略的细节,本工作旨在为地理空间社区提供大模型高效训练与基准测试的最佳实践。

原文摘要 · Abstract (English)

While the pretraining of Foundation Models (FMs) for remote sensing (RS) imagery is on the rise, models remain restricted to a few hundred million parameters. Scaling models to billions of parameters has been shown to yield unprecedented benefits including emergent abilities, but requires data scaling and computing resources typically not available outside industry R&D labs. In this work, we pair high-performance computing resources including Frontier supercomputer, America's first exascale system, and high-resolution optical RS data to pretrain billion-scale FMs. Our study assesses performance of different pretrained variants of vision Transformers across image classification, semantic segmentation and object detection benchmarks, which highlight the importance of data scaling for effective model scaling. Moreover, we discuss construction of a novel TIU pretraining dataset, model initialization, with data and pretrained models intended for public release. By discussing technical challenges and details often lacking in the related literature, this work is intended to offer best practices to the geospatial community toward efficient training and benchmarking of larger FMs.

遥感百亿参数超算视觉变换器

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。