Missing value imputation is a fundamental challenge in spatio-temporal data modeling. To address this issue, we propose a novel model named Bidirectional Mamba Imputation (BMI), which incorporates the selective state space model, Mamba, into the missing data recovery task. The proposed BMI leverages bidirectional information flow to capture both past and future dependencies in time series, and introduces a spatio-temporal block design to jointly learn temporal and spatial patterns. Moreover, a selective mechanism is embedded in the state transition process to adaptively focus on crucial variables and time steps. Extensive experiments conducted on four real-world datasets demonstrate that BMI significantly outperforms existing traditional and deep learning baselines under both point and block missing scenarios. Ablation studies confirm the effectiveness of the bidirectional design and the selective state space module. Additionally, efficiency analysis shows that BMI is well-suited for deployment in resource-constrained environments due to its high memory and I/O efficiency.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

BMI: A Bidirectional Spatio-Temporal Selective State Space Model for Data Imputation

  • Jigang Wen,
  • Wenqian Nie,
  • Xiaocan Li,
  • Kun Xie,
  • Yong Xie,
  • Yuxiang Chen,
  • Zigeng Fu,
  • Quan Feng

摘要

Missing value imputation is a fundamental challenge in spatio-temporal data modeling. To address this issue, we propose a novel model named Bidirectional Mamba Imputation (BMI), which incorporates the selective state space model, Mamba, into the missing data recovery task. The proposed BMI leverages bidirectional information flow to capture both past and future dependencies in time series, and introduces a spatio-temporal block design to jointly learn temporal and spatial patterns. Moreover, a selective mechanism is embedded in the state transition process to adaptively focus on crucial variables and time steps. Extensive experiments conducted on four real-world datasets demonstrate that BMI significantly outperforms existing traditional and deep learning baselines under both point and block missing scenarios. Ablation studies confirm the effectiveness of the bidirectional design and the selective state space module. Additionally, efficiency analysis shows that BMI is well-suited for deployment in resource-constrained environments due to its high memory and I/O efficiency.