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

贝叶斯神经网络用于功能ANOVA模型

Bayesian Neural Networks for Functional ANOVA model

Department of Statistics, Seoul National University(统计系,首尔国立大学);Department of Applied Mathematics, University of Twente(应用数学系,特文特大学)

Seokhun Park, Choeun Kim, Jihu Lee, Yunseop Shin, Insung Kong, Yongdai Kim

AI总结:

本文提出贝叶斯-TPNN,通过贝叶斯推断方法改进功能ANOVA模型,以更高效地检测高阶成分。

摘要:

随着对机器学习可解释性需求的增加,功能ANOVA分解作为一种原理性工具,重新获得了关注,用于将高维函数分解为低维成分,以揭示不同变量组的贡献。最近,张量积神经网络(TPNN)已被开发并应用于功能ANOVA模型中,称为ANOVA-TPNN。然而,ANOVA-TPNN的一个缺点是需要事先指定要估计的成分,这使得难以将更高阶的TPNN纳入功能ANOVA模型,因为计算和内存限制。在本工作中,我们提出了贝叶斯-TPNN,一种用于具有TPNN基函数的功能ANOVA模型的贝叶斯推断程序,能够以比ANOVA-TPNN更低的计算成本检测更高阶的成分。我们开发了高效的MCMC算法,并通过分析多个基准数据集证明了贝叶斯-TPNN的性能。理论上,我们证明了贝叶斯-TPNN的后验是一致的。

英文摘要:

With the increasing demand for interpretability in machine learning, functional ANOVA decomposition has gained renewed attention as a principled tool for breaking down high-dimensional function into low-dimensional components that reveal the contributions of different variable groups. Recently, Tensor Product Neural Network (TPNN) has been developed and applied as basis functions in the functional ANOVA model, referred to as ANOVA-TPNN. A disadvantage of ANOVA-TPNN, however, is that the components to be estimated must be specified in advance, which makes it difficult to incorporate higher-order TPNNs into the functional ANOVA model due to computational and memory constraints. In this work, we propose Bayesian-TPNN, a Bayesian inference procedure for the functional ANOVA model with TPNN basis functions, enabling the detection of higher-order components with reduced computational cost compared to ANOVA-TPNN. We develop an efficient MCMC algorithm and demonstrate that Bayesian-TPNN performs well by analyzing multiple benchmark datasets. Theoretically, we prove that the posterior of Bayesian-TPNN is consistent.

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

↑