arXiv Machine Learning

LatticeSMC: Where to Spend Inference-Time Compute in Chunked Sequence Generators

The paper introduces LatticeSMC, a sampler for chunked sequence generators that balances inference-time compute across chunk indices and denoising steps using a Feynman‑Kac model on a two‑dimensional lattice. By resampling at chunk boundaries and, when scoring is free, within chunks, it achieves higher beat alignment and prompt adherence on music‑to‑dance diffusion and text‑to‑music generation while maintaining quality. The method also outperforms existing steering techniques on long‑range rewards and is preferred by human raters in most pairwise comparisons.

arXiv AI
Oct 7

Best-of-$N$ Guidance for Test-time Diffusion Alignment

The paper introduces Best‑of‑N Guidance (BoNG), a method that embeds the Best‑of‑N sampling principle directly into the reverse diffusion process of generative models. By performing online BoN selection over denoising particles and using the current best particle as a guidance signal, BoNG steers the particle population toward higher‑reward regions during generation. Empirical results show that BoNG outperforms both standard BoN and SMC across 36 comparisons, achieving the best performance in 29 cases and supporting multi‑output generation with significant speedup.

By Richard Lee Kim, Yeongmin Kim, Gyuwon Sim, Taekyu Kim, Minsang Park, Il-chul Moon
arXiv Computer Vision
Oct 12

Spend Where It Matters: Particle Search with MH Refinement for Budget-Constrained Audio-Visual Alignment

The paper introduces “Spend Where It Matters”, a two‑stage, training‑free framework that combines particle search with Metropolis‑Hastings refinement for budget‑constrained audio‑visual alignment. In the first stage, particles are adaptively allocated across denoising checkpoints based on online variance estimates, reducing estimation variance while focusing compute where it matters most. The second stage uses MH refinement with local noise‑and‑denoise proposals to further improve reward, starting from the strong candidate produced by the particle search. Experiments on “JavisBench‑Mini” show that this combined approach yields higher rewards than either particle search or MH alone, achieving significant relative gains over baseline methods. The method also demonstrates benefits for image generation on SD1.5, improving aesthetic and image reward scores compared to uniform particle allocation.

By Shivanshu Shekhar, Raghavendra Addanki, Tong Zhang, Uttaran Bhattacharya
arXiv Computer Vision
Oct 16

Speculative Forcing for Efficient Autoregressive Video Generation

Speculative Forcing is a framework for efficient autoregressive video generation that treats intermediate clean estimates as drafts and predicts when individual tokens can be safely accepted early. Tokens that are accepted are frozen and no longer undergo further denoising, while their cached keys and values remain available for context. Two verifier designs—one training‑free based on similarity and stability, and one learned that uses intermediate features—enable this selective refinement, preserving full‑step quality while reducing computation across short‑video, long‑video, and world generation tasks.

By Hyunwook Choi, Seungryong Kim, Dongyoon Han, Sangdoo Yun
arXiv AI
Sep 25

Accelerating Video Diffusion via Training-Free Trajectory Routing

The paper introduces TRACK, a training‑free trajectory routing method that accelerates video diffusion by selectively switching between large and small models during denoising steps. A calibration process generates a disagreement score map, guiding the selection of the appropriate model at each step to maintain quality while reducing computational cost. Experiments on Wan 2.1, Cosmos 3, TurboDiffusion, and FastVideo show speedups ranging from 1.95× to 2.73× with comparable quality and diversity.

By Mustafa Munir, Huy Vu, Shreyas Misra, Rohit Jena, Sajad Norouzi, Ali Taghibakhshi, Anis Ahmad, Anjul Patney, Pavlo Molchanov, Nima Tajbakhsh
arXiv Machine Learning
Oct 9

The Lattice of Transition Laws

The paper introduces a framework that unifies diffusion and autoregressive (AR) generative models as paths on a corruption lattice, defining a cost for decoding schedules based on discarded parallel steps. It shows that the minimal number of steps for a zero‑cost schedule is determined by the data’s geometry—specifically, the graph’s treedepth for Markov data—providing a theoretical limit for both token and continuous field models. By estimating pairwise dependencies from pretrained weights, the authors predict the ranking of decoding schedules before execution, and validate these predictions across text, image, and video generation tasks.

By T. Y. Tsui, Jiatao Gu, Lingjie Liu
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.

arXiv Machine Learning
Sep 21

Reading Less While Writing: A Closed-Form Bandwidth Dial for Streaming Multimodal Decoders

The paper introduces ZENDAYA, a closed‑form bandwidth dial that replaces fixed wait‑k rules in streaming multimodal decoders with a single continuous parameter γ. This schedule ties the visible source prefix to generation progress and the input’s predicted length, allowing a single offline decoder to serve as both offline and real‑time streaming models. Experiments on video‑to‑text and audio‑to‑text tasks show that reading less of the source can improve text quality, with significant METEOR gains across three public corpora.

By Yasir Mehmood, Kashif Javed
arXiv Machine Learning
Aug 28

GRAS: Guided Reduced-Variance Proposals and Adaptive Selection for Training-Free Reward Alignment in Discrete Diffusion

The paper introduces GRAS, a method that improves training‑free reward alignment for discrete diffusion models by reducing variance in guided proposals and adapting the resampling temperature during search. It achieves this without adding denoiser cost, using Rao‑Blackwellized estimates for differentiable rewards and a leave‑one‑out baseline for non‑differentiable ones. Experiments on regulatory DNA and protein design show GRAS outperforms existing training‑free techniques and rivals reward‑fine‑tuned models.

By Kwanyoung Kim
arXiv Machine Learning
Aug 5

Latent Reward Registers for Diffusion Preference Alignment

arXiv:2608. 03929v1 Announce Type: new Abstract: Aligning diffusion models with human preferences usually relies on a sparse terminal reward evaluated on the final generated samples, presenting a severe temporal credit-assignment challenge across the multi-step denoising process.

By Yuanshen Guan, Zipeng Feng, Zhiwei Xiong, Peiqin Sun