Command Palette
Search for a command to run...
CoSTI: Consistency Models for (a faster) Spatio-Temporal Imputation
CoSTI: Consistency Models for (a faster) Spatio-Temporal Imputation
Javier Solís-García Belén Vega-Márquez Juan A. Nepomuceno Isabel A. Nepomuceno-Chamorro
Abstract
Multivariate Time Series Imputation (MTSI) is crucial for many applications,such as healthcare monitoring and traffic management, where incomplete data cancompromise decision-making. Existing state-of-the-art methods, like DenoisingDiffusion Probabilistic Models (DDPMs), achieve high imputation accuracy;however, they suffer from significant computational costs and are notablytime-consuming due to their iterative nature. In this work, we propose CoSTI,an innovative adaptation of Consistency Models (CMs) for the MTSI domain. CoSTIemploys Consistency Training to achieve comparable imputation quality to DDPMswhile drastically reducing inference times, making it more suitable forreal-time applications. We evaluate CoSTI across multiple datasets and missingdata scenarios, demonstrating up to a 98% reduction in imputation time withperformance on par with diffusion-based models. This work bridges the gapbetween efficiency and accuracy in generative imputation tasks, providing ascalable solution for handling missing data in critical spatio-temporalsystems.