English

ContinualFlow: Learning and Unlearning with Neural Flow Matching

Machine Learning 2025-06-24 v1 Artificial Intelligence

Abstract

We introduce ContinualFlow, a principled framework for targeted unlearning in generative models via Flow Matching. Our method leverages an energy-based reweighting loss to softly subtract undesired regions of the data distribution without retraining from scratch or requiring direct access to the samples to be unlearned. Instead, it relies on energy-based proxies to guide the unlearning process. We prove that this induces gradients equivalent to Flow Matching toward a soft mass-subtracted target, and validate the framework through experiments on 2D and image domains, supported by interpretable visualizations and quantitative evaluations.

Keywords

Cite

@article{arxiv.2506.18747,
  title  = {ContinualFlow: Learning and Unlearning with Neural Flow Matching},
  author = {Lorenzo Simone and Davide Bacciu and Shuangge Ma},
  journal= {arXiv preprint arXiv:2506.18747},
  year   = {2025}
}

Comments

Accepted at the ICML 2025 Workshop on Machine Unlearning for Generative AI (MUGen @ ICML25, Vancouver, July 2025)

R2 v1 2026-07-01T03:29:40.841Z