HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

CoSTI: Consistency Models for (a faster) Spatio-Temporal Imputation

Solís-García Javier ; Vega-Márquez Belén ; Nepomuceno Juan A. ; Nepomuceno-Chamorro Isabel A.

CoSTI: Consistency Models for (a faster) Spatio-Temporal Imputation

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.

Code Repositories

javiersgjavi/costi
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
multivariate-time-series-imputation-on-metrCOSTI
1 step MAE: 1.76

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
CoSTI: Consistency Models for (a faster) Spatio-Temporal Imputation | Papers | HyperAI