arXiv Machine Learning

GLIDE: Graph-guided Leap Inference for Diffusion Estimation of Spatio-Temporal Point Processes

arXiv:2606. 01273v1 Announce Type: new Abstract: Spatio-temporal point processes (STPPs) provide a principled framework for modeling asynchronous events in continuous time and space.

arXiv Machine Learning
Aug 31

Improved off-policy training of diffusion samplers

The paper investigates training diffusion models to sample from distributions defined by unnormalized densities or energy functions. It benchmarks various diffusion-structured inference techniques, including simulation-based variational methods and off-policy approaches such as continuous generative flow networks, highlighting their relative strengths and challenging some prior claims. Additionally, the authors introduce a new exploration strategy for off-policy methods that employs local search in the target space with a replay buffer, demonstrating improved sample quality across multiple target distributions.

By Marcin Sendera, Minsu Kim, Sarthak Mittal, Pablo Lemos, Luca Scimeca, Jarrid Rector-Brooks, Alexandre Adam, Yoshua Bengio, Esmeralda S. Whitammer
arXiv Statistics ML
Oct 21

Divide-and-Conquer Neural Bayes Estimation for Spatial and Spatiotemporal Point Process Models

The paper introduces a divide-and-conquer neural Bayes estimation framework for spatial and spatiotemporal point process models. It uses transformer-based architectures on partitions of the domain and an attention mechanism to combine local summaries, enabling direct inference from raw point pattern data. Simulation studies show reliable inference with better scalability than existing neural and classical likelihood-based methods, and the approach is applied to whale call data in Cape Cod Bay, producing parameter estimates and credible intervals at negligible cost after training.

By Bokgyeong Kang, Andrew Zammit-Mangion, Erin M. Schliep, Robert S. Schick
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
Hugging Face Trending Papers
Sep 24

Accelerating Video Diffusion via Training-Free Trajectory Routing

Accelerating Video Diffusion via Training-Free Trajectory Routing (TRACK) introduces a heterogeneous denoising strategy that switches between large and small diffusion models at selected steps, determined by a calibration process that measures disagreement between model predictions. By routing quality-sensitive steps to the large model and low-disagreement steps to the small model, TRACK achieves significant speedups—up to 2.73×—across several video diffusion benchmarks while maintaining comparable quality and diversity. The method requires no retraining, architectural changes, or online dual-model evaluation, making it a practical acceleration paradigm for video diffusion.