BAM：用于可解释表格预测的双变量加性模型
BAM: Bivariate Additive Models for Interpretable Tabular Prediction
AI 导读
研究者提出双变量加性模型（BAM），将 GA²M 中每个成对形状函数约束为非对称双变量高斯的简单参数形式，特征对在训练中自动选择，每个组件可可视化为单峰等高线图。在六个表格基准上的实验显示，BAM 的预测精度与标准 GAM 相当，接近 GA²M。论文还展示了精度与需检查组件数量之间的权衡。
正文 · 原文
Abstract:Generalized Additive Models with pairwise interactions (GA$^2$Ms) are regarded as glass-box models, since each pairwise shape function can be visualized as a contour plot. In practice, the shape functions learned by methods such as EBM or NODE-GAM are so complex (irregular contours and sharp local fluctuations) that a practitioner cannot meaningfully interpret them or check them against domain knowledge. Auditability in this stronger sense requires substantially simpler shape functions. We argue that such simplification can match the accuracy of standard GAMs if the right feature pairs are captured. We propose Bivariate Additive Models (BAMs), a class of GA$^2$Ms in which every pairwise shape function is constrained to a simple parametric form, instantiated as an asymmetric bivariate Gaussian. Each component is visualizable as a unimodal contour map, and its feature pair is selected automatically during training. Experiments on six tabular benchmarks show that BAM attains predictive accuracy comparable to standard GAMs and approaches that of GA$^2$Ms. We also show the trade-off between accuracy and the number of components to inspect.
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2611.05940 [cs.LG] |
| (or arXiv:2611.05940v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2611.05940 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Atsushi Koike Dr. [view email]
[v1]
Wed, 4 Nov 2026 04:28:59 UTC (810 KB)
来源：arXiv cs.AI · arxiv.org