arXiv Machine Learning

On the spectral properties of generative denoiser Jacobians

arXiv Computer Vision
Sep 18

Training Flow Matching: The Role of Weighting and Parameterization

The paper investigates training objectives for denoising-based generative models, focusing on loss weighting and output parameterization such as noise-, clean image-, and velocity-based formulations. It conducts a systematic numerical study across synthetic datasets with controlled geometry and real image data, evaluating denoising accuracy via PSNR and generative quality via FID. The goal is to disentangle how training choices interact with data manifold dimensionality, model architecture, and dataset size, offering practical design insights rather than proposing a new method.

By Anne Gagneux, S\'egol\`ene Martin, R\'emi Gribonval, Mathurin Massias
arXiv Machine Learning
Sep 14

Deep learning methods for inverse problems using connections between proximal operators and Hamilton-Jacobi equations

The paper proposes a deep learning framework that learns priors for inverse problems by exploiting the relationship between proximal operators and Hamilton–Jacobi partial differential equations. Unlike existing methods that require inverting the prior after training, this approach learns the prior directly, enabling efficient evaluation in a single forward pass. Numerical experiments demonstrate the method’s effectiveness in dimensions up to 64.

By Oluwatosin Akande, Gabriel P. Langlois, Akwum Onwunta
arXiv Machine Learning
Aug 26

Generalization, memorization, and overfitting for diffusion models trained in the lazy high-dimensional regime

The paper investigates diffusion models trained in a lazy high‑dimensional regime, extending benign overfitting theory to generative settings. By analyzing denoising score matching in a vector‑valued RKHS with an inner‑product kernel, the authors derive exact risk trajectories under gradient flow when the number of samples scales proportionally with dimensionality. These trajectories reveal three distinct phases—spectral generalization, noise‑dominated interpolation, and empirical Bayes memorization—whose interplay shapes the distribution of generated samples.

By Hugo Latourelle-Vigeant, Sinho Chewi, Aram-Alexandre Pooladian, John Sous, Theodor Misiakiewicz