TerraTorch: The Geospatial Foundation Models Toolkit

要約

Terratorchは、Pytorch Lightning上に構築され、衛星、天候、気候データに合わせて調整された地理空間基盤モデル向けの微調整およびベンチマークキットです。
ドメイン固有のデータモジュール、事前に定義されたタスク、およびバックボーンを多様なデコーダーヘッドと組み合わせたモジュラーモデルファクトリーを統合します。
これらのコンポーネントにより、研究者と実践者は、トレーニング構成を編集するだけで、サポートされていないモデルを微調整できません。
モデル開発のためのベストプラクティスを統合し、自動化されたハイパーパラメーターの最適化拡張繰り返しを組み込むことにより、Terratorchは、新しい地球観測の使用ケースのモデルを微調整またはベンチマークするのに必要な専門知識と時間を短縮します。
さらに、Terratorchはジオベンチと直接統合し、地理空間基礎モデルの体系的で再現可能なベンチマークを可能にします。
TerratorChは、https://github.com/ibm/terratorchで入手可能、Apache 2.0の下で開いており、PIP Install Terratorchを介してインストールできます。

要約(オリジナル)

TerraTorch is a fine-tuning and benchmarking toolkit for Geospatial Foundation Models built on PyTorch Lightning and tailored for satellite, weather, and climate data. It integrates domain-specific data modules, pre-defined tasks, and a modular model factory that pairs any backbone with diverse decoder heads. These components allow researchers and practitioners to fine-tune supported models in a no-code fashion by simply editing a training configuration. By consolidating best practices for model development and incorporating the automated hyperparameter optimization extension Iterate, TerraTorch reduces the expertise and time required to fine-tune or benchmark models on new Earth Observation use cases. Furthermore, TerraTorch directly integrates with GEO-Bench, allowing for systematic and reproducible benchmarking of Geospatial Foundation Models. TerraTorch is open sourced under Apache 2.0, available at https://github.com/IBM/terratorch, and can be installed via pip install terratorch.

arxiv情報

提供元, 利用サービス

arxiv.jp, Google