
LW
Luran Wang, Linrui Ma, Hannes St\"ark, Regina Barzilay
· 1 min read
ResearcharXiv cs.LG
Specificity-Aware Diffusion Steering via Variance-Reduced Sequential Monte Carlo
arXiv:2610.00395v1 Announce Type: new
Abstract: Inference-time steering enables pretrained diffusion models to satisfy new constraints without full retraining. However, specificity-aware generation is difficult: repelling samples from a negative reference distribution can also erode the positive distribution where the two overlap. The key challenge is to suppress negative mass while minimally distorting the positive distribution. We address this problem by formulating specificity-aware steering as a target-design problem and deriving a target distribution from an overlap-based objective. The resulting target keeps the desired reference distribution only in regions where it is sufficiently preferred over the undesired reference distribution, giving a likelihood-ratio interpretation of specificity. To sample from the corresponding time-dependent target path, we develop a Sequential Monte Carlo sampler with a variance-minimized local proposal. We further introduce a practical fixed-noise optimization procedure with the Jacobian--vector products with the desired and undesired score fields. Experiments on synthetic task, class-contrastive generation, text-to-image tasks and peptide-MHC (p-MHC) binder show that the proposed method suppresses undesired regions more effectively, reduces mode shift, and improves sampling stability by decreasing the SMC weight collapse compared with negative-guidance baselines. Code is available at: https://github.com/WangLuran/Specificity-Aware-Diffusion-Steering
Original source
This story was published by arXiv cs.LG and written by Luran Wang, Linrui Ma, Hannes St\"ark, Regina Barzilay. SyncAI.news shows a preview; the complete article is on the publisher's site.
Read the full story on arxiv.org


