|国家科技期刊平台
首页|期刊导航|南京理工大学学报|基于集成重要性采样的随机梯度下降算法

基于集成重要性采样的随机梯度下降算法OA北大核心CSTPCD

中文摘要

许多机器学习和深度学习问题都可以使用随机梯度优化算法求解,目前流行的算法大多通过均匀采样从样本集中抽取样本计算梯度估计。然而,随机采样的梯度估计会带来较大的方差,这个方差会随着优化的进行而累积,降低算法收敛速度。为缓解这一现象,可以为每个样本赋予不同的采样概率。该文基于集成学习的思想,提出了一种新的选取非均匀采样分布的算法。算法的主要目的是选取一个采样器权重,使梯度估计的方差尽可能小。所提算法由多个简单采样器组成,采样权重为每个简单采样器分配贡献权重,从而得到最终的采样分布。集成重要性采样算法可以和以往的随机梯度优化方法任意结合,该文给出了使用集成重要性采样的随机梯度下降算法。在试验中,可以直观地看到算法起效的原因。在真实数据集中,展示了所提算法减小方差的效果,与其他算法相比具有一定优势。

张浩;鲁淑霞;

河北大学数学与信息科学学院,河北保定071002 河北省机器学习与计算智能重点实验室,河北保定071002

计算机与自动化

集成学习重要性采样采样器随机梯度下降方差减少

《南京理工大学学报》 2024 (003)

P.342-350 / 9

河北省创新能力提升计划科技研发平台建设专项(22567623H)。

10.14177/j.cnki.32-1397n.2024.48.03.012

评论