연구진은 0-3시간 강수량 예측 정확도를 높이기 위해 Mamba와 self-attention을 결합한 MambaRain 프레임워크를 제안했어요.
MambaRain은 Mamba 블록으로 장기적인 시간 의존성을 모델링하고 self-attention 모듈로 공간 상관관계를 분석하여 2-3시간 예측 정확도를 크게 향상시켰어요.
새로운 스펙트럴 손실 함수를 도입하여 예측 영상의 흐릿함 문제를 완화하고, 기존 방식보다 성능이 뛰어나다는 것을 실험적으로 검증했어요.