arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2605.20009cs.LGcs.AIcs.NE
收藏

用最优双贝叶斯学习训练神经网络

Training Neural Networks with Optimal Double-Bayesian Learning

Lister Hill National Center for Biomedical Communications, National Library of Medicine National Institutes of Health(利斯特希尔国家生物医学通讯中心、国家医学图书馆国家卫生研究院)

Vy Bui, Hang Yu, Karthik Kantipudi, Ziv Yaniv, Stefan Jaeger

AI总结:

本文提出了一种新的概率框架,用于学习率这一关键参数,通过双贝叶斯决策机制改进随机梯度下降,从而推导出理论上最优的学习率,并在多种任务中验证其有效性。

摘要:

反向传播与梯度下降是大多数机器学习神经网络架构中常用的优化策略。然而,找到指导训练的最优超参数已证明具有挑战性。尽管普遍认可选择合适参数对于避免过拟合和获得无偏结果至关重要,但这一选择仍主要基于经验实验和经验。本文提出了一种新的概率框架,用于学习率这一随机梯度下降中的关键参数。该框架将经典贝叶斯统计发展为一种涉及两个对抗性贝叶斯过程的双贝叶斯决策机制。从这两个过程可以推导出理论上最优的学习率,并用于随机梯度下降。在各种分类、分割和检测任务中的实验验证了理论上推导出的学习率的实践意义。本文还讨论了所提出的双贝叶斯框架对网络训练和模型性能的影响。

英文摘要:

Backpropagation with gradient descent is a common optimization strategy employed by most neural network architectures in machine learning. However, finding optimal hyperparameters to guide training has proven challenging. While it is widely acknowledged that selecting appropriate parameters is crucial for avoiding overfitting and achieving unbiased outcomes, this choice remains largely based on empirical experiments and experience. This paper presents a new probabilistic framework for the learning rate, a key parameter in stochastic gradient descent. The framework develops classic Bayesian statistics into a double-Bayesian decision mechanism involving two antagonistic Bayesian processes. A theoretically optimal learning rate can be derived from these two processes and used for stochastic gradient descent. Experiments across various classification, segmentation, and detection tasks corroborate the practical significance of the theoretically derived learning rate. The paper also discusses the ramifications of the proposed double-Bayesian framework for network training and model performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑