Showing 2026Show all
2 papers · 1 filter
cs.AI2026
ConicFlow: constrained Inference-time optimal control for flow matching models
Shervin Khalafi, Kasra Fallah, Sung Cho +3
We propose ConicFlow, which frames constrained sampling from a flow matching model at inference time as an optimal control problem that balances deviation from the pretrained traje…
cs.LG2026
Unlearning in Diffusion Models: A Unified Framework with KL Divergence and Likelihood Constraints
Shervin Khalafi, Alejandro Ribeiro, Dongsheng Ding
Unlearning in diffusion models aims to remove undesirable data or concepts while preserving the utility of pretrained models -- two fundamentally conflicting objectives. We propose…