엔비디아, 오픈소스 기상 AI 모델군 '어스-2' 공개…연산 시간 90% 절감

엔비디아의 어스-2(Earth-2)는 기상 및 기후 예측을 위한 오픈소스 AI 모델 제품군이자 가속 소프트웨어 스택이다. 기존에 대형 슈퍼컴퓨터를 독점하던 물리 기반 수치예보를 기업이나 기관이 자체 인프라에서 실행하고 미세조정할 수 있도록 전환해, 초기 관측 데이터 처리부터 15일 글로벌 예보, 국지성 폭풍 예측까지 전 과정을 지원한다.

모델 구성과 작동 방식

기존 수치예보는 방대한 유체역학 방정식을 풀기 위해 대규모 CPU 슈퍼컴퓨터 클러스터를 장시간 가동해야 했다. Earth-2는 이를 물리 기반 AI로 대체하거나 보완하며, 사전 학습된 모델과 맞춤형 튜닝 레시피, 추론 라이브러리를 하나로 묶어 제공한다.

이번에 공개된 핵심 모델은 글로벌 중기 예보를 담당하는 Earth-2 Medium Range와 국지성 악천후를 예측하는 Earth-2 Nowcasting이다. 기존의 초고해상도 다운스케일링 모델 CorrDiff, 중장기 예측 모델 FourCastNet3와 함께 파이프라인을 구성하며, 관측 데이터를 통합하는 Earth-2 Global Data Assimilation 모듈도 올해 안에 추가될 예정이다.

호환성도 넓다. 유럽중기기상예보센터(ECMWF), 구글, 마이크로소프트의 오픈소스 기상 모델을 통합 지원하며, 오픈소스 파이썬 프레임워크인 PhysicsNeMo를 사용해 로컬 데이터에 맞춘 대규모 물리 AI 모델 학습 및 파인튜닝이 가능하다.

연산 시간 90% 절감과 실제 성과

현장 검증 결과는 뚜렷하다. 이스라엘 기상청이 2.5km 해상도 환경에서 Earth-2 CorrDiff를 테스트한 결과, CPU 클러스터 기반의 기존 수치예보 모델 대비 연산 시간을 90% 단축했다. 최근 폭우 발생 당시 6시간 누적 강수량 예측 검증에서는 해당 AI 모델이 기존의 모든 현업 수치 모델보다 우수한 정확도를 기록했다.

산업계 적용도 빠르게 확산 중이다. 기상 AI 스타트업 브라이트밴드(Brightband)는 Earth-2 Medium Range를 실제 현업에 도입해 매일 글로벌 예보를 발행하고 있다. 대형 태양광 기업 GCL은 태양광 발전량 예측 시스템에 이를 적용해 비용을 낮추고 예측 정확도를 크게 끌어올렸다.

미국 사우스웨스트 파워 풀(SPP)과 히타치는 일중 풍력 발전 예측에 Nowcasting을 도입했으며, 토탈에너지스, 에니, 악사(AXA) 등 글로벌 에너지 및 금융 기업들도 기후 위험 분석과 가스 수요 예측에 해당 모델군을 테스트하고 있다.

어디서 사용해 볼 수 있나

공개 범위는 전면 오픈 방식이다. Earth-2 Medium Range와 Nowcasting의 코드와 모델 가중치는 현재 깃허브(GitHub)와 허깅페이스(Hugging Face)에 공개되어 있으며, Earth2Studio 툴킷을 통해서도 바로 이용할 수 있다.

다만 슈퍼컴퓨터가 필요 없다고 해서 진입 장벽이 완전히 사라진 것은 아니다. 고해상도 글로벌 예측이나 PhysicsNeMo를 통한 모델 재학습에는 고성능 GPU 인프라가 필수적이며, 데이터 동화 모듈이 완전히 출시되기 전까지는 자체적인 파이프라인 구성이 일부 요구된다.

왜 중요한가

초정밀 기상 예측은 오랫동안 최고급 슈퍼컴퓨터를 보유한 소수 국가 기관의 전유물이었다. Earth-2가 완성도 높은 기상 AI 스택을 오픈소스로 내놓으면서, 지역 기상청은 물론 에너지 기업, 농업 및 보험사들이 자체 서버에서 맞춤형 기상 모델을 돌릴 수 있는 길이 열렸다.

엔비디아 입장에서도 영리한 전략이다. 과학 계산 분야에서 가장 연산 집약적인 기상 예측 영역에 고성능 오픈 모델을 배포해 기술 표준을 선점하고, 장기적인 추론 및 파인튜닝 수요를 자사 가속 컴퓨팅 생태계로 흡수하고 있다. 기상 변화에 민감한 에너지 거래나 발전소 운영 조직이라면 지금 즉시 벤치마크해 볼 가치가 충분하다.