maxxxzdn

mosaic

maxxxzdn

(Sparse) Attention to the Details: Preserving Spectral Fidelity in ML-based Weather Forecasting Models [ICML'26]

AI 简介

Mosaic 是一个基于机器学习的天气预报模型,通过网格对齐的块稀疏注意力机制在原生分辨率网格上运行。该项目的核心功能包括在1.5°分辨率下利用2.14亿参数进行高效预测,同时保持频谱保真度,并且单个H100 GPU上24成员10天的预报耗时不到12秒。技术特点方面,Mosaic 通过学习的功能扰动解决了频谱衰减问题,并采用块稀疏注意力消除了高频混叠现象。该模型适合用于需要高精度和快速响应的天气预报场景,尤其是在处理大规模气象数据集时能够显著提高预测准确性和计算效率。

Python
40
Stars
1
Forks
35
Watchers
0
Issues

Star 增长

今日0
近 7 天+1
近 30 天+2
综合评分38.6
默认分支main