‘AI로 보는 폭풍’ 기상 예보 혁신하는 NVIDIA의 새로운 생성형 AI 모델

허레케인 시즌이 다가오면서, NVIDIA는 고해상도 대기 역학을 모방하는 혁신적인 생성형 AI 모델을 공개했습니다
by NVIDIA Korea

허리케인과 토네이도를 비롯한 기타 기상이변의 발생 빈도와 심각성이 증가하면서 최신 기술을 사용해 기후 연구와 예측을 개선하고 가속화하는 것이 그 어느 때보다 중요합니다.

현재 대서양 허리케인 시즌이 절정에 이른 시점에서, NVIDIA Research는 고해상도 대기 역학을 에뮬레이션하는 새로운 생성형 AI 모델인 StormCast를 발표했습니다. StormCast는 폭풍보다는 크지만 사이클론보다는 작은 규모인 중규모에서 신뢰할 수 있는 날씨 예측을 가능하게 합니다. 이는 재난 계획과 완화에서 매우 중요하죠.

로렌스 버클리 국립 연구소(Lawrence Berkeley National Laboratory)와 워싱턴 대학교(University of Washington)가 공동으로 작성한 논문에 따르면, StormCast는 극심한 기상 현상으로 미국에서만 매년 1,500억 달러 이상의 금전 피해, 인명 피해, 주택 파괴가 발생하고 있는 상황에서 등장했습니다.

이는 생성형 AI가 기후 연구와 실행 가능한 극한 기상 예측에서 어떻게 엄청난 혁신을 촉진하고 있는지 보여주는 한 가지 예시에 불과합니다. 과학자들이 생명과 세계를 구하는 가장 중요한 과제를 해결하는 것을 도우면서 말이죠.

NVIDIA Earth-2는 AI, 물리 시뮬레이션, 컴퓨터 그래픽의 힘을 결합한 디지털 트윈 클라우드 플랫폼입니다. 전례 없는 정확성과 속도로 전 세계적 규모의 날씨와 기후 예측을 시뮬레이션하고, 시각화할 수 있도록 지원합니다.

지난 6월 열린 컴퓨텍스(COMPUTEX) 2024에서 NVIDIA 창립자 겸 CEO인 젠슨 황(Jensen Huang)은 Earth-2를 활용할 수 있는 CorrDiff를 발표했습니다.

예를 들어, 대만의 NCDR(National Science and Technology Center for Disaster Reduction)에서는 Earth-2의 일부로 제공되는 NVIDIA 생성형 AI 모델 CorrDiff를 사용해 태풍의 세밀한 세부 사항을 예측할 계획입니다.

CorrDiff는 기존 방법보다 1,000배 더 빠르고 3,000배 더 적은 에너지를 사용해, 25km 규모의 대기 데이터를 2km로 12.5배 수준으로 높여 초고해상도 변환이 가능하죠.

즉 CPU에 거의 3백만 달러가 소요됐던 이 센터의 잠재적 생명 구조 작업이 NVIDIA H100 Tensor 코어 GPU가 탑재된 단일 시스템에서는 약 6만 달러로 수행할 수 있게 된 것입니다. 이것이 바로 생성형 AI와 가속 컴퓨팅이 어떻게 에너지 효율성을 높이고 비용을 낮출 수 있는지 보여주는 엄청난 절감 효과의 예시입니다.

이 센터는 CorrDiff를 활용해 도시 지역에서 강풍이 도로까지 내려와 건물을 파손하고 보행자에게 영향을 미치는 다운워시(downwash) 현상도 예측할 계획입니다.

이제 StormCast는 CorrDiff에 시간별 자기회귀 예측(hourly autoregressive prediction) 기능을 추가해, 과거 데이터를 기반으로 미래의 결과를 예측할 수 있습니다.

지역 중심의 세계적 영향

세계 기후 연구는 지역 수준에서 시작됩니다.

날씨와 기후 변화의 물리적 위험은 지역 규모에 따라 크게 달라질 수 있는데요. 그러나 지역 수준에서 신뢰할 수 있는 수치적 날씨 예측에는 상당한 계산 비용이 수반됩니다. 이는 중규모에서 기본적인 유체 역학적 움직임을 표현하는 데 필요한 높은 공간 해상도 때문이죠.

대류 허용 모델(Convection-Allowing Models, CAMs)이라고 하는 지역 기상 예측 모델은 전통적으로 연구자들이 해상도, 앙상블 크기, 경제성 측면에서 다양한 트레이드오프(trade off)에 직면해야 했습니다.

CAMs는 기상학자들이 폭풍의 진화와 구조를 추적하고, 대류 모드(convective mode)나 폭풍 형성 시 폭풍이 결성되는 상황을 모니터링하는 데 유용하죠. 예를 들어, 토네이도의 발생 가능성은 폭풍의 구조와 대류 모드를 기반으로 합니다.

미국 국립해양대기청(National Oceanic and Atmospheric Administration, NOAA)의 정지 궤도 운영 환경 위성을 사용해 시각화한 중규모 대류 시스템. 이미지 제공: NOAA

또한 CAMs는 연구자들이 인프라구조 수준에서 날씨와 관련된 물리적 위험에 대한 영향을 이해하는 데 도움이 됩니다.

예를 들어, 범 지구 기후 모델 시뮬레이션이 CAMs에 제공한 정보는 대규모 하천의 대기 수분 함량 변화를 취약한 해안 지역에서 나타나는 돌발 홍수 예측으로 변환하는 데 도움이 될 수 있죠.

저해상도에서는 글로벌 데이터로 훈련된 머신 러닝 모델이 수치 기상 예측 모델의 유용한 에뮬레이터로 부상하고 있습니다. 극한 상황에 대한 조기 경보 시스템을 개선하는 데 사용되는 이 모델은 일반적으로 약 30km의 공간 해상도와 6시간의 시간 해상도를 가지고 있습니다.

이제 StormCast는 생성형 확산(generative diffusion)의 도움으로 3km와 시간 단위의 시공간 해상도를 가능하게 합니다.

이 모델은 아직 초키 단계임에도 불구하고 이미 강수 레이더와 함께 적용했을 때 최대 6시간의 리드 타임(lead times)을 가집니다. 미국 NOAA의 최첨단 3km 단위 CAMs보다 최대 10% 더 정확한 예보를 제공하죠.

또한 StormCast의 출력물은 사실적인 열과 수분의 물리적 역학을 보여줍니다. 온도, 수분 농도, 바람과 강우 레이더 반사율 값 등 100가지가 넘는 변수를 미세한 간격의 여러 고도에서 예측할 수 있습니다. 이를 통해 과학자들은 폭풍이 가진 부력의 사실적 3D 진화를 확인할 수 있는데요. 이는 AI 기상 시뮬레이션 분야에서 최초로 달성한 성과입니다.

NVIDIA 연구원들은 계산 속도를 높이는 NVIDIA 가속 컴퓨팅을 사용해 미국 중부 지역에서 약 3년 반 동안 수집한 NOAA 기후 데이터로 StormCast를 훈련시켰습니다.

더 많은 혁신의 탄생

과학자들은 이미 이 모델의 이점을 활용하기 위해 노력하고 있습니다.

더 웨더 컴퍼니(The Weather Company)의 혁신 책임자 톰 해밀(Tom Hamill)은 “조직화된 뇌우와 겨울 강수의 엄청난 영향, 그리고 이를 확실하게 예측하는 데 따르는 주요 어려움을 고려할 때, 컴퓨터로 추적 가능한 폭풍 규모의 앙상블 일기 예보 생성은 수치 기상 예측의 큰 과제 중 하나입니다. StormCast는 이러한 과제를 해결하는 주목할 만한 모델이죠. 더 웨더 컴퍼니는 NVIDIA와 협력해 이러한 딥 러닝 예측 모델을 개발, 평가하고 잠재적으로 활용할 수 있게 돼 매우 기쁩니다”고 말했습니다.

콜로라도 주립대학교 대기협동조합연구소(Colorado State University Cooperative Institute for Research in the Atmosphere, CIRA)의 머신 러닝 책임자 임 에버트-우포프(Imme Ebert-Uphoff)는 “고해상도 기상 모델을 개발하려면 대류를 해결하는 AI 알고리즘이 필요한데, 이는 매우 어려운 과제입니다. NVIDIA의 새로운 연구는 StormCast와 같은 확산 모델을 통해 이를 달성할 가능성을 탐구합니다. 고해상도 기상 예측을 위한 미래 AI 모델 개발로 가는 중요한 진전이라고 말할 수 있죠”라고 언급했습니다.

물리적으로 정확한 기후 시뮬레이션의 가속화와 시각화, 지구의 디지털 트윈과 함께 이러한 연구 혁신은 NVIDIA Earth-2가 어떻게 기후 연구의 새롭고 중요한 시대를 열어가고 있는지 보여줍니다.

지속 가능한 컴퓨팅NVIDIA Research에 대해 자세히 알아보세요. NVIDIA Researh는 수백 명의 과학자와 엔지니어로 구성된 글로벌 팀으로, 기후 AI, 컴퓨터 그래픽, 컴퓨터 비전, 자율주행 자동차, 로보틱스 등의 주제를 탐구합니다.

썸네일 이미지 제공: NASA

소프트웨어 제품 정보에 관한 공지도 확인해보세요.