arXiv:2605.19291stat.MLcs.LG2026-05

提出FSGD,让SGD在高维数据中边流式处理边学习潜在因子。

Factor Augmented High-Dimensional SGD

论文配图:Factor Augmented High-Dimensional SGD
图 1 · 摘自论文原文
  • 用潜在因子增强的SGD,直接处理流式数据,无需预存全部数据。
  • 首次将因子估计误差纳入理论分析,证明了小步长下的收敛性。
  • 适合大规模高维机器学习系统,尤其对实时更新场景友好。

随机梯度下降(SGD)是现代机器学习中的基础优化算法。本文提出因子增强型SGD(FSGD),一种在高维学习任务中利用潜在因子表示的新优化方法。与依赖离线表示学习和全量数据存储的标准两阶段降维方法不同,FSGD的核心创新在于仅需处理流式数据,可扩展至大规模高维问题。此外,我们建立了首个将潜在因子估计误差显式纳入分析的理论框架,在衰减步长与小批量更新下,证明了ℓ^s范数下的矩收敛性。该结果为在高维机器学习系统中可靠、可扩展地使用SGD提供了新基础。

原文摘要 · Abstract (English)

Stochastic gradient descent (SGD) is a fundamental optimization algorithm widely used in modern machine learning. In this paper, we propose Factor-Augmented SGD (FSGD), a new optimization method that leverages latent factor representations in high-dimensional learning tasks. Unlike standard two-stage dimension reduction approaches that rely on offline representation learning and full data storage, a key novelty of FSGD is that it operates purely on streaming data, making it scalable to large-scale and high-dimensional problems. Furthermore, we establish the first theoretical framework that explicitly incorporates latent factor estimation error into the analysis of SGD, and provide moment convergence in $\ell^s$ norm under decaying step sizes and mini-batch updates. Our results provide a new foundation for employing SGD reliably and scalably in high-dimensional machine learning systems.

优化算法高维学习流式处理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。