中文

一种核扩展随机神经网络

机器学习 2022-01-17 v1 机器学习

摘要

深度神经网络在机器学习中存在诸多基本问题。例如,其训练常陷入局部极小,且预测不确定性难以评估。为解决这些问题,我们提出所谓的核扩展随机神经网络(K-StoNet)模型,该模型将支持向量回归(SVR)作为首个隐藏层,并将神经网络重新表述为潜变量模型。前者通过径向基函数(RBF)核将输入向量映射到无限维特征空间,确保其训练损失面上无局部极小。后者将高维非凸的神经网络训练问题分解为一系列低维凸优化问题,并使其预测不确定性易于评估。K-StoNet可使用插补正则化优化(IRO)算法轻松训练。相比传统深度神经网络,K-StoNet具有渐近收敛至全局最优的理论保证,且使预测不确定性易于评估。新模型在训练、预测与不确定性量化上的表现通过模拟与真实数据示例加以说明。

关键词

引用

@article{arxiv.2201.05319,
  title  = {A Kernel-Expanded Stochastic Neural Network},
  author = {Yan Sun and Faming Liang},
  journal= {arXiv preprint arXiv:2201.05319},
  year   = {2022}
}

备注

Accepted by JRSSB