Damba-ST: Domain-Adaptive Mamba for Efficient Urban Spatio-Temporal Prediction
Created by
Haebom
저자
Rui An, Yifeng Zhang, Ziran Liang, Wenqi Fan, Yuxuan Liang, Xuequn Shang, Qing Li
개요
본 논문은 다양한 지역과 도시에서 일반화가 잘 되는 도시 공간-시간 기반 모델을 훈련시키는 것에 초점을 맞추고 있습니다. 기존의 Transformer 기반 모델들은 계산 복잡도가 높고 메모리 오버헤드가 커서 확장성과 실제 배포에 제약이 있었습니다. 본 논문에서는 선형 시간 복잡도를 갖는 상태 공간 모델인 Mamba에 착안하여 효율적인 도시 공간-시간 예측을 위한 새로운 모델인 Damba-ST를 제안합니다. Damba-ST는 공간-시간 이질성 문제를 해결하기 위해 도메인 적응형 상태 공간 모델을 도입하여 잠재 표현 공간을 공유 하위 공간과 도메인별 하위 공간으로 분할하고, 세 가지 도메인 어댑터를 통해 도메인 간 차이를 줄이고 일반적인 특징을 정렬합니다. 실험 결과, Damba-ST는 최첨단 성능을 달성하고 강력한 제로샷 일반화 능력을 보여줍니다.