All talks

Talk Université Aix-Marseille · Marseille, France

Journées de Statistique 2025

Oral presentation of Taking a Big Step: Large Learning Rates in Denoising Score Matching Prevent Memorization.

Slides

slides page 1 of 21
1 / 21 Download PDF

The paper behind this talk

Taking a Big Step: Large Learning Rates in Denoising Score Matching Prevent Memorization

COLT 2025

Yu-Han Wu, Pierre Marion, Gérard Biau, Claire Boyer

  • Problem. The exact minimizer of denoising score matching memorizes the training data, yet trained models memorize only mildly.
  • Approach. Analyze gradient descent with large steps, which can only converge to minima whose sharpness is bounded by the inverse learning rate.
  • Result. The memorizing solution is too sharp to be reached; large learning rates prevent memorization.
Key figure from Taking a Big Step: Large Learning Rates in Denoising Score Matching Prevent Memorization
Graphs of the learned model with different learning rates and of the empirical optimal score, for two pairs of (μ, σ). As the learning rate decreases, the learned model approaches the empirical optimal score. When σ is smaller (right), that score is more irregular and a smaller learning rate is needed to reach it.

Read more about the paper

Also presented at

All talks