Energy Time-Series Imputation with Differentially Private Diffusion Models via Clipping-Aware Objective Conditioning
On the map: Diffusion models → Flow matching, time series and science also fits: Diffusion foundations and sampling
Abstract (arXiv)
Reliable recovery of missing measurements is important for monitoring and analysis in energy time-series systems, where fine-grained measurements may contain sensitive temporal information. Diffusion models trained with differentially private stochastic gradient descent (DP-SGD) provide a promising framework for privacy-sensitive energy time-series imputation. Under cosine diffusion schedules, late timesteps correspond to low signal-to-noise ratio (SNR) conditions, where standard varepsilon-prediction can induce large pre-clipping gradients. Such gradients are more likely to be clipped, reducing the retained optimization signal. The artificial intelligence (AI) contribution lies in formulating this objective--clipping interaction as an objective optimization problem under fixed-threshold DP-SGD and developing timestep-aware objective conditioning for diffusion-based energy time-series imputation. The method adopts v-prediction to mitigate late-timestep gradient amplification, uses static loss weighting as a uniform-scaling control, and introduces diffusion-schedule-aware dynamic weighting for stronger attenuation before clipping. For the engineering application, we evaluate the method on five real-world energy time-series datasets across random point missingness, contiguous block missingness, persistent outages, and multiple missing-data severities. Under matched DP-SGD settings, the proposed method consistently improves imputation utility over the varepsilon-prediction baseline. Gradient diagnostics reveal lower upper-tail pre-clipping gradient norms, reduced clipping fractions, and stronger attenuation at late low-SNR timesteps, supporting the effectiveness of clipping-aware objective conditioning for energy time-series imputation.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 11 of its 21 references that are in the map. First posted on arXiv 21 Sep 2026.
Builds on
11 of 21 references.
Closest ancestors · 2
Same direction, newest first
- 2024TimeDiT: General-purpose Diffusion Transformers for Time Series Foundation Model3 Sep
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputation7 Jul
Other references · 9
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2016Deep Learning with Differential Privacycited by 2,682
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2022TimesNet: Temporal 2D-Variation Modeling for General Time Series Analysiscited by 1,030
- 2023iTransformer: Inverted Transformers Are Effective for Time Series Forecastingcited by 906
- 2023Efficient Diffusion Training via Min-SNR Weighting Strategycited by 242
- 2022Differentially Private Diffusion Modelscited by 118
- 2024Differentially Private Fine-Tuning of Diffusion Modelscited by 18
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024EnergyDiff: Universal Time-Series Energy Data Generation using Diffusion Models
- 2026DP-aware AdaLN-Zero: Taming Conditioning-Induced Heavy-Tailed Gradients in Differentially Private Diffusion
- 2026Analyzing and Improving Diffusion Models for Time-Series Data Imputation: A Proximal Recursion Perspective
- 2025LSCD: Lomb-Scargle Conditioned Diffusion for Time series Imputation
- 2026RDDMPI: Residual Denoising Diffusion Model for Probabilistic Multivariate Time Series Imputation
- 2025Exploiting the Prior of Generative Time Series Imputation
- 2026Discretizing Continuous Time Series for Imputation with Masked Diffusion Training
- 2023Towards More Accurate Diffusion Model Acceleration with A Timestep Tuner