中文

OlmoEarth v1.1:更高效的 OlmoEarth 模型系列

计算机视觉与模式识别 2026-05-21 v1 机器学习

摘要

我们提出了一组改进 OlmoEarth 系列的方案。这些改进允许我们在训练期间削减计算成本(Base 模型训练所需 GPU 小时减少 1.7 倍),在推理阶段(Sentinel-2 任务上的 MACs 减少 2.9 倍),同时保持模型的整体性能。所有训练代码均可用 github.com/allenai/olmoearth_pretrain 获取。

关键词

引用

@article{arxiv.2605.20804,
  title  = {OlmoEarth v1.1: A more efficient family of OlmoEarth models},
  author = {Gabriel Tseng and Yawen Zhang and Favyen Bastani and Henry Herzog and Joseph Redmon and Hadrien Sablon and Piper Wolters and Patrick Alan Johnson and Christopher Wilhelm and Patrick Beukema},
  journal= {arXiv preprint arXiv:2605.20804},
  year   = {2026}
}