菜单

关于 🐙 GitHub
arXiv 提交日期: 2026-07-06
📄 Abstract - Non-asymptotic Convergence of Stochastic Gradient Descent in Score-based Generative Models

Score-based Generative Models (SGMs) have achieved impressive performance in data generation across a wide range of applications. While the statistical properties of their sampling procedures are increasingly well understood, the optimization dynamics underlying their training remain less explored. SGMs are typically trained by minimizing a weighted denoising scorematching objective, yet optimization guarantees with stochastic gradients remain limited. In this work, we study Stochastic Gradient Descent (SGD) for SGMs, contributing results in two complementary regimes. First, for general score parameterizations, we establish a non-convex convergence rate for SGD on the weighted denoising score-matching objective, with explicit dependence on the schedule-dependent weighting factors. Second, for overparameterized two-layer ReLU networks, we develop a Neural Tangent Kernel analysis tailored to diffusion training with stochastic gradients, yielding score-approximation error bounds along the SGD trajectory. Finally, our analysis quantifies the role of the reweighting factor in the score approximation error, providing theoretical guidance for weighting choices used in practice.

顶级标签: machine learning theory
详细标签: stochastic gradient descent score-based generative models non-convex optimization neural tangent kernel convergence analysis 或 搜索:

基于分数的生成模型中随机梯度下降的非渐近收敛性分析 / Non-asymptotic Convergence of Stochastic Gradient Descent in Score-based Generative Models


1️⃣ 一句话总结

该论文首次从理论上证明了在基于分数的生成模型中,使用随机梯度下降(SGD)训练去噪分数匹配目标时,即使目标函数非凸,SGD也能以可控的速率收敛,并且揭示了训练过程中权重因子对最终模型精度的影响规律,为实践中如何选择权重提供了理论依据。

源自 arXiv: 2607.04775