arXiv cs.AI· Giovanni Briglia, Marco Lippi, Stefano Mariani, Franco Zambonelli·· 8 小时前AI 评分27
神经贝叶斯网络 NBN：面向模块化可扩展不确定性决策的 GPU 原生方案
Neural Bayesian Networks for Modular and Scalable Decision-Making Under Uncertainty
AI 导读
研究者提出神经贝叶斯网络（NBN），将贝叶斯网络中每个条件分布替换为可微分的张量机制，同时保持原有图结构与分解方式不变，从而可表示任意可学习的条件分布并原生运行在 GPU 上。团队同时发布基于 PyTorch 的 NBN 库及公开基准，在大规模网络和大批量查询上凭借 GPU 并行实现两个数量级的加速。NBN 库已公开可用。
来源：arXiv cs.AI · arxiv.org