Search papers, labs, and topics across Lattice.
This paper introduces the Masked Diffusion Time-series Imputation Model (MDTIM), which addresses the limitations of existing time series imputation methods by structurally separating missing and observed values and directly predicting original values. By employing a MASK token and Stochastic Discretization, MDTIM effectively maps continuous time series data to ordinal-aware tokens, enhancing the model's alignment with the imputation task. Experimental results demonstrate that MDTIM outperforms state-of-the-art methods in robustness and scalability across various missing data scenarios.
MDTIM not only separates missing from observed values but also directly predicts original time series values, leading to superior imputation performance.
Time series imputation is a crucial area for reliable time series analysis, yet it remains challenging due to the complex temporal dynamics and noise of real-world data. Existing approaches, however, exhibit two limitations: missing and observed values are embedded within the same representation space without explicit structural separation, and continuous diffusion-based methods are trained to predict added noise rather than the original signal. To address these, we propose the Masked Diffusion Time-series Imputation Model (MDTIM), which leverages the training paradigm of masked diffusion model for imputation tasks. The MASK token is structurally orthogonal to valid observations, and the model directly predicts the original values, naturally aligning both the representation and the learning objective with the imputation task. To bridge the gap between discrete masked diffusion and the continuous, ordinal nature of time series, we further introduce Stochastic Discretization, which maps continuous values to ordinal-aware tokens while preserving continuous dynamics. Our experiments on diverse benchmarks confirm that MDTIM achieves superior robustness and scalability, consistently outperforming state-of-the-art deterministic and generative baselines across various missing scenarios.
