태풍 경로와 중기 날씨 예측 모델을 직접 실행하고 검증하는 방법
태풍이 어디로 이동할지, 며칠 뒤 날씨가 어떻게 변할지 궁금할 때 이 저장소를 열면 됩니다. 구글 딥마인드가 개발한 WeatherNext 2 모델 코드를 공개해, 연구자나 개발자가 직접 모델을 실행하고 결과를 시각화할 수 있게 합니다. 복잡한 수식 대신 코드를 통해 실제 예측 과정을 눈으로 확인하고, 태풍 추적까지 해볼 수 있습니다.
WeatherNext 2와 이전 모델들
이 저장소에는 최신 모델인 WeatherNext 2(WN2)와 태풍 특화 모델 WeatherNext Cyclones 코드가 들어 있습니다. WN2는 전 지구적 중기 날씨와 태풍을 함께 예측하며, 100m 높이의 바람까지 계산합니다. 이전 세대 모델인 GraphCast와 GenCast 코드도 함께 제공되어 비교 실험이 가능합니다. 각 모델은 독립적으로 학습된 가중치를 사용하지만, 태풍 예측 알고리즘은 동일하게 작동합니다.
모델 버전과 해상도
운영용으로 쓰이는 WeatherNext2_<2025>와 WeatherNextCyclones_<2025>는 0.25도 해상도(약 30km)를 가지며, 2024년까지의 데이터로 학습되었습니다. 특히 Cyclones 모델은 2025년 대서양 허리케인 시즌에 실시간으로 운영된 버전입니다. 메모리나 컴퓨트 자원이 제한된 환경에서는 1도 해상도의 Mini 버전을 사용할 수 있습니다. Mini 버전은 성능이 큰 모델보다 낮지만, 로컬 테스트나 단일 GPU/TPU에서 실행하기에 적합합니다.
실행 방법과 요구 사항
가장 쉬운 시작 방법은 Colab Notebook을 여는 것입니다. 기본 설정은 Mini 모델이며, 무료 v5e-1 런타임으로 실행할 수 있습니다. 더 큰 모델을 실행하려면 v5p 가속기가 필요합니다. TPU에서 실행하는 것이 최적화되어 있지만, GPU를 사용할 경우 어텐션 구현 방식을 변경해야 합니다. 비-Mini 모델은 충분한 VRAM을 위해 H100이 필요하며, Mini 모델은 P100에서도 추론이 가능합니다. 사전 학습된 가중치와 샘플 데이터는 Google Cloud Bucket에서 다운로드할 수 있습니다.
데이터와 라이선스 확인 사항
모델을 실행하려면 ECMWF의 ERA5 데이터셋을 Zarr 형식으로 받아야 하며, 운영용 미세 조정 데이터는 WeatherBench2의 HRES 데이터를 사용합니다. 코드는 Apache 2.0 라이선스, 나머지 자료는 CC BY 4.0 라이선스로 배포됩니다. 이는 공식적으로 지원되는 구글 제품이 아니며, 정부 기상 기관의 공식 경보나 통지를 대체하지 않는 실험적 연구 프로젝트입니다.
