arXiv AI

TraceSampler: Making Sampling Errors Predictable and Correctable for Accelerated Generative Sampling

arXiv Computer Vision
Oct 12

A Lightweight Velocity Correction Framework for Improving Instruction Following in Few-Step Text-to-Image Generation

The paper introduces a lightweight correction framework that improves instruction following in few-step text-to-image generators without altering the generator’s parameters or increasing forward-pass count. The framework adds a small corrector module (<20 M parameters) that learns residual velocity adjustments at selected sampling steps, using On‑Trajectory Flow‑Bridge Supervision to target the actual states reached during corrected generation. Experiments show a +17.60‑point NetWin over a frozen baseline with only a 0.84% increase in end‑to‑end latency.

By Yating Gao, Zhenlong Zhu, Wen Zihao, Lenan Zhang, Jiexiong Wang, Yang Yang, Guanjun Jiang, Yixiong Zou, Takuya Maekawa
arXiv AI
Jun 2

Efficient Weighted Sampling via Score-based Generative Models

arXiv:2502. 04646v2 Announce Type: replace-cross Abstract: Weighted sampling -- sampling from a probability density function (PDF) proportional to the product of a base PDF and a weight function -- is a fundamental technique with wide-ranging applications in variance reduction, biased sampling, data augmentation, and more.

By Heasung Kim, Taekyun Lee, Hyeji Kim, Gustavo de Veciana
arXiv Computer Vision
Oct 23

ParaLin++: Accelerating Parallel Diffusion Integrator via Intrinsic Partially Linear Structure

ParaLin++ is a training‑free, step‑parity parallel sampler that accelerates diffusion model sampling by preserving the full‑step discretization while removing the solver’s temporal dependency. The authors first review quality metrics for training‑free sampling and show that many step‑shortening methods lose quality on semantic‑composition and long‑text benchmarks, whereas parallel sampling remains near‑lossless. Experiments on Stable Diffusion‑v2, LSUN Church, FLUX.1‑dev, and Ming‑Flash‑Omni 2.0 demonstrate speedups of 2.6×–4.8×, 1.68×, and 1.21× respectively while maintaining the full‑step Euler quality profile.

By Jianrong Lu, Zhiyu Zhu, Hui Liu, Junhui Hou
arXiv AI
1d ago

PEEK: One-Step-Look-Ahead Exposure Bias Correction for Diffusion Sampling

The paper introduces PEEK, a training‑free deterministic sampler plugin that corrects exposure bias in diffusion image generators by performing a provisional one‑step update and interpolating between predictions at a look‑ahead state. PEEK evaluates the denoiser at this higher‑signal noisy state, thereby producing more image‑like predictions and reducing the train‑test gap that often leads to blurry or poorly detailed outputs. Experiments on eight pixel‑space and latent‑space diffusion models—including unconditional, label‑guided, and text‑guided settings—show consistent improvements over NFE‑matched baselines, with FID gains up to 49.3 % and minimal additional computation.

By Yunwei Bai, Ying Kiat Tan, Tsuhan Chen