Background Gradients Shape Memorization in Flow Matching
On the map: Diffusion models → Image generation, text-to-image and editing also fits: Video generation
Abstract (arXiv)
Repetition is closely associated with memorization in generative models, but how other training images affect the retention and copying of targets remains unclear. We study this question in class-conditioned flow matching, where images outside the target set form the background. At fixed target repetition and same-class background row count, replacing repeated same-class images with distinct images reduces the target extraction rate from 80.7% to 18.0%. To explain this effect, we develop a paired-trajectory framework that isolates target-induced parameter displacement and the background gradient response to it. This response has an exact path-integrated curvature representation, connecting background loss geometry to target learning. Reciprocal response transfer between repeated and distinct backgrounds changes target retention and copying in both directions, establishing the response's causal role. After target removal, the response correction parallel to the target-induced displacement preserves approximately 90% of the copying effects of full response transfer. Directly scaling the displacement also changes copying without further training. The post-removal copying effects of reciprocal transfer are reproduced across datasets and architectures. Together, these results identify the background gradient response as a mechanism through which same-class training data shape the retention of target learning and the reproduction of target images.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 60 of its 71 references that are in the map. First posted on arXiv 27 Sep 2026.
Builds on
60 of 71 references.
Closest ancestors · 1
Same direction, newest first
Other references · 59
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
38 more
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2018Demystifying MMD GANscited by 1,268
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2018The Secret Sharer: Evaluating and Testing Unintended Memorization in Neural Networkscited by 847
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 835
- 2022Quantifying Memorization Across Neural Language Modelscited by 755
- 2023Extracting Training Data from Diffusion Modelscited by 723
- 2021Deduplicating Training Data Makes Language Models Bettercited by 705
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 565
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 547
- 2020What Neural Networks Memorize and Why: Discovering the Long Tail via Influence Estimationcited by 492
- 2017Sharp Minima Can Generalize For Deep Netscited by 462
- 2019Does Learning Require Memorization? A Short Tale about a Long Tailcited by 452
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 357
- 2022Deduplicating Training Data Mitigates Privacy Risks in Language Modelscited by 309
- 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 289
- 2023TRAK: Attributing Model Behavior at Scalecited by 273
- 2022Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Modelscited by 218
- 2023Understanding and Mitigating Copying in Diffusion Modelscited by 205
- 2023Generalization in diffusion models arises from geometry-adaptive harmonic representationscited by 179
- 2023On Memorization in Diffusion Modelscited by 122
- 2022Measuring Forgetting of Memorized Training Examplescited by 118
- 2024Detecting, Explaining, and Mitigating Memorization in Diffusion Modelscited by 94
- 2025Why Diffusion Models Don't Memorize: The Role of Implicit Dynamical Regularization in Trainingcited by 78
- 2021On Memorization in Probabilistic Deep Generative Modelscited by 60
- 2023A Reproducible Extraction of Training Images from Diffusion Modelscited by 60
- 2025On the Closed-Form of Flow Matching: Generalization Does Not Arise from Target Stochasticitycited by 48
- 2023Intriguing Properties of Data Attribution on Diffusion Modelscited by 47
- 2024A Geometric Framework for Understanding Memorization in Generative Modelscited by 40
- 2024Finding NeMo: Localizing Neurons Responsible For Memorization in Diffusion Modelscited by 36
- 2024Influence Functions for Scalable Data Attribution in Diffusion Modelscited by 31
- 2025On the Edge of Memorization in Diffusion Modelscited by 22
- 2024Understanding and Mitigating Memorization in Generative Models via Sharpness of Probability Landscapescited by 20
50 most cited of 59 All 59 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Tracing Generated Samples to Training-Data Clusters in Flow-Matching Models
- 2025From Navigation to Refinement: Revealing the Two-Stage Nature of Flow-based Diffusion Models through Oracle Velocity
- 2026The Geometry of Memorization: Finite-Time Spectral Sensitivity as a Diagnostic for Flow Matching Models
- 2026Posterior Augmented Flow Matching
- 2026Rethinking Memorization Mitigation in Diffusion Models: Reinforcing Text Conditioning
- 2025How Diffusion Models Memorize
- 2025Demystifying Foreground-Background Memorization in Diffusion Models
- 2026LeapAlign: Post-Training Flow Matching Models at Any Generation Step by Building Two-Step Trajectories