Representation-Augmented Diffusions
F. M. Sofian*, K. Pandey*, S. Mandt
A new class of diffusion models that are faster to train, faster to sample, and parameter-efficient, with a smooth latent space.
I'm an early fourth-year Ph.D. student at UC Irvine, advised by Prof. Stephan Mandt and supported by the Chan Zuckerberg Initiative.
I work on generative modeling with diffusion models: designing new foundation models, conditional sampling via test-time steering, and reward-guided fine-tuning, with applications from images and language to science. Previously, I was an AI Resident at the Chan Zuckerberg Initiative and at Adobe.
01
Two papers accepted to NeurIPS 2026: Hierarchical Variational Policies for Reward-Guided Diffusion and Control-Augmented Autoregressive Diffusion for Data Assimilation.
Parallel Token Prediction for Language Models accepted to ICLR 2026.
Variational Control for Guidance in Diffusion Models accepted to ICML 2025.
Joined Adobe for the summer to work on LLMs.
02
9 papers
Topic
Year
01
New model classes and training schemes for large diffusion and language models.
F. M. Sofian*, K. Pandey*, S. Mandt
A new class of diffusion models that are faster to train, faster to sample, and parameter-efficient, with a smooth latent space.
F. Draxler*, R. C. Geyer*, D. Geyfman*, F. M. Sofian*, J. Will*, et al.
Pre-training and post-training a large language model that decodes many tokens in parallel.
F. Draxler, J. Will, F. M. Sofian, T. Karaletsos, S. Mandt
Predicts several tokens in one forward pass by moving randomness into the model's inputs, giving a 2.4× speedup in speculative decoding.
* Equal contribution
02
Steering pretrained diffusion models toward rewards and constraints — at inference time or through lightweight learned control.
K. Pandey*, F. M. Sofian*, J. Groeneveld, F. Draxler, S. Mandt
Amortizes test-time control into a lightweight stochastic policy, matching test-time scaling baselines at a fraction of the compute.
P. Srivastava, F. M. Sofian, F. Immorlano, K. Pandey, S. Mandt
Adds an offline-trained controller to autoregressive diffusion models for fast, stable data assimilation in chaotic PDEs and weather.
H. Lee*, F. M. Sofian*, K. Pandey, S. Mandt
Derives diffusion synchronization from optimal control to coordinate multiple diffusion trajectories at test time, with no extra training.
K. Pandey*, F. M. Sofian*, F. Draxler, T. Karaletsos, S. Mandt
Unifies diffusion guidance as trajectory matching under variational control, reaching state of the art on inverse problems without retraining.
* Equal contribution
03
Multimodal generation and NLP from my undergraduate research at Koç University.
B. C. Biner, F. M. Sofian, U. B. Karakaş, D. Ceylan, E. Erdem, et al.
Conditions pretrained image diffusion models on audio, enabling sound-driven image generation and editing.
A. Kara, F. M. Sofian, A. Bond, G. G. Sahin
The first large-scale dataset and baselines for Turkish grammatical error correction and detection.
03
I'm always happy to talk about research collaborations or new opportunities.
If you're a woman in STEM and would like guidance, a sounding board, or just a chat, please reach out. Supporting other women in the field means a lot to me, wherever you are in your journey.
Email is the best way to reach me.