LLM2D
BARNN:一个贝叶斯自回归和循环神经网络
BARNN: A Bayesian Autoregressive and Recurrent Neural Network
作者: Dario Coscia, Max Welling, Nicola Demo, Gianluigi Rozza
发布日期: 2/3/2025
arXiv ID: oai:arXiv.org:2501.18665v1

摘要

arXiv:2501.18665v1公告类型: 交叉 摘要:自回归和递归网络在各个领域都取得了显著进展,从天气预报到分子生成和大型语言模型。尽管这些模型具有强大的预测能力,但在科学应用如偏微分方程求解、分子生成和机器学习力场中,它们缺乏一个严谨的不确定性处理框架,这是关键。为了弥补这一不足,我们提出了一种变分贝叶斯自回归和递归神经网络——BARNN (Bayesian Autoregressive and Recurrent Neural Network)。BARNNs 的目标是提供一种原则性的方式,将任何自回归或递归模型转化为其贝叶斯版本。BARNN 以变分丢弃方法为基础,允许将其应用于大型递归神经网络。此外,我们引入了“变分后验混合物”的时间版本(tVAMP-prior)以提高贝叶斯推断的效率和准确性。在偏微分方程建模和分子生成的广泛实验中,BARNN 不仅在准确度方面实现了与现有方法相似或更优的结果,还在此基础上在不确定性量化和建模长程依赖方面表现出色。