arXiv Machine Learning

DiFA: Inference-Time Forward-Process Alignment for Diffusion Models

arXiv:2607. 17972v1 Announce Type: new Abstract: The prevailing inference framework for diffusion models formulates generation fundamentally as a problem of numerical integration.

arXiv Machine Learning
Sep 3

DynG-Diff: A State-Aware Dynamic Guidance Diffusion Framework for Probabilistic Time Series Forecasting

DynG-Diff is a new diffusion-based framework for probabilistic multivariate time‑series forecasting that addresses the challenge of information heterogeneity across variables. It uses a two‑stage training strategy with an unconditional diffusion backbone and introduces a lightweight state‑aware policy network that dynamically adjusts guidance strength based on real‑time variable reliability. The dynamic guidance is mathematically framed as local precision, allowing the model to focus on high‑confidence variables and suppress anomalous noise, leading to competitive performance and robustness on real‑world benchmarks.

By Zhente Zhang, Zhengwei Ni, Wei Fan
arXiv Computer Vision
Sep 21

Quantization-Aware Kalman Estimation for Diffusion Sampling

The paper introduces QuAKE, a Quantization-Aware Kalman Estimator designed to correct errors in diffusion model sampling when using quantized denoisers. By treating sampling as an online estimation problem, QuAKE leverages the history of quantized outputs to recover full-precision estimates, updating a posterior in closed form at each step. The method is lightweight, plug‑and‑play, and works with any high‑order multistep ODE sampler, outperforming existing correction techniques on W4A4‑quantized text‑to‑image diffusion models.

By Qitan Shi, Cheng Jin, Jiawei Zhang, Yuantao Gu
arXiv Machine Learning
Sep 11

RDDMPI: Residual Denoising Diffusion Model for Probabilistic Multivariate Time Series Imputation

RDDMPI introduces a residual denoising diffusion model for multivariate time series imputation. By decomposing the missing signal into a baseline reconstruction and a residual uncertainty component, the method conditions the diffusion process on both the completed signal and its latent representation, using a reliability-aware mechanism to balance baseline influence. Experiments on benchmark datasets show that this approach improves reconstruction accuracy and uncertainty quantification compared to prior diffusion-based methods.

By Ramiro Valdes Jara, David Chapman, Adam Meyers