arXiv Computer Vision

ToPO: Token-Conditioned Preference Routing for Attention-Based Latent Diffusion Models

ToPO (Token-Oriented Preference Optimization) is a method for attention-based latent diffusion models that constructs a per‑minibatch, detached, separable spatial‑temporal route using branchwise squared‑residual contrast from a frozen reference denoiser. It employs preferred‑branch cross‑attention with content tokens to modulate spatial factors and adds an auxiliary pixel‑midpoint ordering term, all without local labels or a learned reward model. In experiments with shared update schedules, ToPO outperforms Diffusion‑DPO on five SD‑1.5 metrics and on HPSv2, ImageReward, and CLIP for SDXL, and achieves higher raw win shares in a blind SDXL A/B study.

arXiv Computer Vision
Sep 28

Where Compute Matters: Heterogeneous Attention for Efficient Video Diffusion

The paper introduces HetA-DiT, a heterogeneous attention mechanism for video diffusion models that allocates computation based on token difficulty. A lightweight uncertainty branch predicts denoising difficulty, routing uncertain tokens through dense global attention while applying efficient local attention to reliable tokens. This adaptive routing retains global context where needed, offers a single parameter to balance quality and efficiency, and achieves competitive generation quality while only about 20% of tokens use dense attention.

By Olga Zatsarynna, Denis Korzhenkov, Juergen Gall, Amir Habibian, Mohsen Ghafoorian
arXiv Machine Learning
Sep 7

Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Models

The paper introduces Diffusion LAIR, a listwise preference optimization technique that leverages continuous reward scores instead of binary pairwise comparisons to align text‑to‑image diffusion models. LAIR transforms reward scores into centered advantage weights and optimizes an advantage‑weighted regression objective on an implicit reward defined by denoising‑loss improvement over a reference model, with a quadratic penalty to regulate reward magnitude. Experiments demonstrate that Diffusion LAIR surpasses strong baseline methods on SD1.5 and SDXL across generation, compositional, and editing tasks.

By Austin Wang, Jiaqi Han, Stefano Ermon, Yisong Yue
arXiv Computer Vision
Oct 16

ReState: Noise-Level-Aware State Correction for Training-Free Object Removal in Few-Step Diffusion Models

ReState is a training‑free method for object removal in few‑step diffusion models that corrects the masked clean prediction toward background evidence using a noise‑level‑aware weight that fades out at the final forward pass. It complements attention rerouting, requiring no additional denoiser evaluations, and outperforms existing training‑free techniques on two paired benchmarks and two distilled backbones, improving PSNR by an average of 3.17 dB. On the OBER‑Test set, ReState also surpasses a 50‑step attention‑forcing reference while running 6.7× faster.

By Donghwan Lee, Sanghyun Jo, Sihyun Kim, Ziseok Lee, Kyungsu Kim
arXiv AI
Jun 10

Attention-Discounted Adaptive Sampler for Masked Diffusion Language Models

arXiv:2606. 10829v1 Announce Type: cross Abstract: Masked diffusion language models can reduce inference steps by revealing multiple tokens per denoising iteration, but this parallelism is fragile: positions that are individually confident may be unsafe to commit together when their predictions are coupled.

By Yusuf Sahin, Ahmed Rockey Saikia, Volkan Cevher, Paolo Favaro
arXiv AI
Sep 10

MAST: Mask-Guided Attention Control for Training-Free Regional-Multi Style Transfer

MAST (Mask‑Guided Attention Control for Training‑Free Regional‑Multi Style Transfer) is a framework that enables diffusion models to apply multiple reference styles to user‑specified regions of a content image without any training or optimization. It introduces logit‑level attention mass allocation, sharpness‑aware temperature scaling, and discrepancy‑aware detail injection to address mass allocation, selectivity, and detail loss problems in regional‑multi style transfer. Experiments with two to five styles show that MAST outperforms baselines in ArtFID, FID, and R‑FID, achieving high regional style fidelity, content preservation, and scalability.

By Dongkyung Kang, Jaeyeon Hwang, Junseo Park, Minji Kang, Yeryeong Lee, Beomseok Ko, Hanyoung Roh, Jeongmin Shin, Hyeryung Jang
arXiv Computation and Language
Oct 12

Exploring Attention Sinks in Text-to-Image Transformers from Diffusion and Autoregressive Models

The paper investigates attention sinks—tokens or positions that attract disproportionate attention—in text-to-image transformers, comparing bidirectional diffusion models and causal autoregressive models. It finds that both model types exhibit attention sinks, but with distinct patterns: autoregressive models focus on the first and global tokens, while diffusion models develop sinks near modality boundaries, within text regions, and at spatial corners. Perturbation experiments show that sink-related heads influence generation more than non-sink heads, and introducing register tokens can redistribute attention without harming performance.

By Binglei Li, Mengping Yang, Zhiyu Tan, Wenxu Qian, Junping Zhang, Hao Li