3 papers
cs.LG2026
FlowBudget: Training-Free Sampling-Budget Estimation for Flow Matching Models
Zahra Dehghani, Pablo Piantanida, Mohammadhadi Shateri
Sampling from continuous flow-based generators such as flow matching models requires numerical ODE or SDE integration, making inference cost depend on the number of function evalua…
cs.LG2026
Source-Free Class Relearning: Diagnosing Forgetting in Class Unlearning
Zahra Dehghani, Pablo Piantanida, Mohammadhadi Shateri
Class unlearning aims to remove a model's ability to recognize designated forget classes while preserving performance on retain classes. However, low forget accuracy after unlearni…
cs.LG2026
Continuous Adversarial MeanFlow Transfer
Yara Bahram, Zahra Dehghani, Mélodie Desbos +3
Training fast generators on new domains with limited data remains challenging for two reasons. First, adapting a pretrained diffusion or flow model to a new domain leaves its costl…