5 papers
Variational Test-time Optimization for Diffusion Synchronization
Hyunsoo Lee, Farrin Marouf Sofian, Kushagra Pandey +1
Collaborative generation, which coordinates multiple diffusion trajectories to extend the capabilities of pretrained priors, has emerged as a powerful paradigm for extending the ap…
Hierarchical Variational Policies for Reward-Guided Diffusion
Kushagra Pandey, Farrin Marouf Sofian, Jan Niklas Groeneveld +2
Adapting pretrained diffusion models to downstream objectives such as inverse problems often requires expensive test-time guidance or optimization. We propose a principled framewor…
Control-Augmented Autoregressive Diffusion for Data Assimilation
Prakhar Srivastava, Farrin Marouf Sofian, Francesco Immorlano +2
Despite advances in test-time scaling and diffusion finetuning, guidance for Auto-Regressive Diffusion Models (ARDMs) remains underexplored. We introduce an amortized framework tha…
Parallel Token Prediction for Language Models
Felix Draxler, Justus Will, Farrin Marouf Sofian +3
Autoregressive decoding in language models is inherently slow, generating only one token per forward pass. We propose Parallel Token Prediction (PTP), a general-purpose framework f…
Variational Control for Guidance in Diffusion Models
Kushagra Pandey, Farrin Marouf Sofian, Felix Draxler +2
Diffusion models exhibit excellent sample quality, but existing guidance methods often require additional model training or are limited to specific tasks. We revisit guidance in di…