arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.06963stat.MLcs.LG

隐马尔可夫神经网络

Hidden Markov Neural Networks

ESOMAS, University of Torino(都灵大学ESOMAS学院);Collegio Carlo Alberto(卡洛阿尔贝托学院);School of Mathematics, University of Bristol(布里斯托大学数学学院)

Lorenzo Rimella, Nick Whiteley

AI总结:

该研究提出隐马尔可夫神经网络,将网络权重建模为隐马尔可夫模型隐状态,结合序列反向传播贝叶斯与变分DropConnect,在多任务上实现了优异预测性能与有效不确定性量化。

摘要:

我们定义了一种随时间演化的贝叶斯神经网络,名为隐马尔可夫神经网络(Hidden Markov Neural Network),它解决了时间序列预测和持续学习中的关键挑战:在适应新数据与适度遗忘过时信息之间取得平衡。该方法通过将神经网络的权重建模为隐马尔可夫模型(Hidden Markov model)的隐状态来实现,观测过程由可用数据定义。我们采用滤波算法来学习权重上随时间演化的后验分布的变分近似。通过利用反向传播贝叶斯(Bayes by Backprop)的序列变体,并辅以一种名为变分DropConnect(variational DropConnect)的更强正则化技术,隐马尔可夫神经网络实现了鲁棒的正则化和可扩展的推理。在MNIST、动态分类任务以及视频下一帧预测上的实验表明,隐马尔可夫神经网络在提供强大预测性能的同时,还能实现有效的不确定性量化。

英文摘要:

We define an evolving in-time Bayesian neural network called a Hidden Markov Neural Network, which addresses the crucial challenge in time-series forecasting and continual learning: striking a balance between adapting to new data and appropriately forgetting outdated information. This is achieved by modelling the weights of a neural network as the hidden states of a Hidden Markov model, with the observed process defined by the available data. A filtering algorithm is employed to learn a variational approximation of the evolving-in-time posterior distribution over the weights. By leveraging a sequential variant of Bayes by Backprop, enriched with a stronger regularization technique called variational DropConnect, Hidden Markov Neural Networks achieve robust regularization and scalable inference. Experiments on MNIST, dynamic classification tasks, and next-frame forecasting in videos demonstrate that Hidden Markov Neural Networks provide strong predictive performance while enabling effective uncertainty quantification.

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

↑