An Observed Value Consistent Diffusion Model for Imputing Missing Values in Multivariate Time Series
Xu Wang, Hongbo Zhang, Pengkun Wang, Yudong Zhang, Binwu Wang, Zhengyang Zhou, Yang Wang
Abstract
Missing values, which are common in multivariate time series, is most important obstacle towards the utilization and interpretation of those data. Great efforts have been employed on how to accurately impute missing values in multivariate time series, and existing works either use deep learning networks to achieve deterministic imputations or aim at generating different plausible imputations by sampling multiple noises from a same distribution and then denoising them. However, these models either fall short of modeling the uncertainties of imputations due to their deterministic nature or perform poorly in terms of interpretability and imputation accuracy due to their ignorance of the correlations between the latent representations of both observed and missing values which are parts of samples from a same distribution. To this end, in this paper, we explicitly take the correlations between observed and missing values into account, and theoretically re-derive the Evidence Lower BOund (ELBO) of conditional diffusion model in the scenario of multivariate time series imputation. Based on the newly derived ELBO, we further propose a novel multivariate imputation diffusion model (MIDM) which is equipped with novel noise sampling, adding and denoising mechanisms for multivariate time series imputation, and the series of newly designed technologies jointly ensure the involving of the consistency between observed and missing values. Extensive experiments on both the tasks of multivariate time series imputation and forecasting witness the superiority of our proposed MIDM model on generating conditional estimations.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 4c66f4ba-e213-4dc4-b194-84d446ca7cd4Cited by top-tier papers14
- CrossGNN: Confronting Noisy Multivariate Time Series Via Cross Interaction RefinementQihe Huang, Lei Shen, Ruixin Zhang, Shouhong Ding et al.NeurIPS 2023 · 162 citations
- HDMixer: Hierarchical Dependency with Extendable Patch for Multivariate Time Series ForecastingQihe Huang, Lei Shen, Ruixin Zhang, Jiahuan Cheng et al.AAAI 2024 · 91 citations
- Frequency-aware Generative Models for Multivariate Time Series ImputationXinyu Yang, Yu Sun, Xiaojie Yuan, Xinyang ChenNeurIPS 2024 · 41 citations
- Rethinking the Diffusion Models for Missing Data Imputation: A Gradient Flow PerspectiveZhichao Chen, Haoxuan Li, Fangyikang Wang, Odin Zhang et al.NeurIPS 2024 · 38 citations
- Towards Dynamic Spatial-Temporal Graph Learning: A Decoupled PerspectiveBinwu Wang, Pengkun Wang, Yudong Zhang, Xu Wang et al.AAAI 2024 · 34 citations
Builds on14
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Efficiently Modeling Long Sequences with Structured State SpacesAlbert Gu, Karan Goel, Christopher RéICLR 2022 · 3,482 citations
- Adaptive Graph Convolutional Recurrent Network for Traffic ForecastingLei Bai, Lina Yao, Can Li, Xianzhi Wang et al.NeurIPS 2020 · 2,206 citations
- DiffWave: A Versatile Diffusion Model for Audio SynthesisZhifeng Kong, Wei Ping, Jiaji Huang, Kexin Zhao et al.ICLR 2021 · 1,902 citations
- GMAN: A Graph Multi-Attention Network for Traffic PredictionChuanpan Zheng, Xiaoliang Fan, Cheng Wang, Jianzhong QiAAAI 2020 · 1,858 citations
Related papers
- CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series ImputationYusuke Tashiro, Jiaming Song, Yang Song, Stefano ErmonNeurIPS 2021 · 1,245 citations
- Time-Frequency Conditioned Diffusion for Multivariate Time Series ImputationYumeng Liu, Zheng Wang, Jikui Liu, Kaisa Zhang et al.ICDE 2026
- Conditional Information Bottleneck Approach for Time Series ImputationMinGyu Choi, Changhee LeeICLR 2024 · 24 citations
- Probabilistic Imputation for Time-series Classification with Missing DataSeunghyun Kim, Hyunsu Kim, Eunggu Yun, Hwangrae Lee et al.ICML 2023 · 37 citations
- Diffusion Transformers for Imputation: Statistical Efficiency and Uncertainty QuantificationZeqi Ye, Minshuo ChenNeurIPS 2025
