May, 2024

WeatherFormer:一个基于预训练编码器模型的小数据集天气表示学习方法

TL;DR本文介绍了 WeatherFormer,一种基于 Transformer 编码器的模型,旨在从少量观测数据中学习稳健的天气特征。它解决了在农业、流行病学和气候科学中许多预测任务中建模复杂天气动态的挑战。WeatherFormer 在大型预训练数据集上进行了预训练,包括 39 年在美洲范围内的卫星测量数据。通过独特的预训练任务和微调,WeatherFormer 在县级大豆产量预测和流感预测方面实现了最先进的性能。技术创新包括捕捉地理、年度和季节性变化的独特时空编码,将 Transformer 架构适应连续天气数据,以及一种能够学习对缺失天气特征具有稳健性的表示的预训练策略。本文首次证明了使用大型 Transformer 编码器模型在多个领域的天气相关应用中进行预训练的有效性。