用可读规则构建表格预测模型,确保数据缺失时仍稳定可靠。
FlagGAM: Rule-Basis Generalized Additive Models for Explainable Tabular Prediction

- 将数值和类别变量转为人类可读的规则基,分离规则与预测
- 在数据缺失或噪声下,平均AUROC下降最小,稳定性领先
- 适合需要透明规则和鲁棒性的金融、医疗等场景
表格应用常需可解释的预测规则及输入不完整时的稳定表现。我们提出FlagGAM,一种规则基框架,将特征级规则构建与预测分离。标志核心模块将数值和类别变量转化为稀疏、可读的一元基:阈值标志、类别标志、尾部偏差基和分类阶跃函数。默认加性头以受限GAM风格组合这些基,而保留的稀疏规则基矩阵支持混合类型分类与回归、特征特定加权,以及可选的灵活头。在干净基准上,加性FlagGAM在分类上接近现代加性和规则基基线,在回归上优于全局线性模型,但灵活性低于树基预测器。其最明显优势出现在部署时扰动下:在三个分类数据集上,FlagGAM在缺失值和数值噪声下的平均AUROC下降最小。灵活头提升绝对准确率,接近强树基基线,但应视为对学习规则基的非线性预测。结果表明,FlagGAM是一种约束型加性规则基模型,适用于需要可读规则和不完整输入下稳定行为的应用。
原文摘要 · Abstract (English)
Tabular applications often require inspectable prediction rules and stable behavior when records are incomplete. We propose FlagGAM, a rule-basis framework that separates feature-level rule construction from prediction. A Flag Core Module converts numerical and categorical variables into sparse, human-readable univariate bases: threshold flags, category-level flags, tail-deviation bases, and categorical step functions. A default additive head combines these bases as a restricted GAM-style predictor, while the retained sparse rule-basis matrix supports mixed-type classification and regression, feature-specific weighting, and optional flexible heads. On clean benchmarks, additive FlagGAM stays close to modern additive and rule-based baselines on classification and improves over global linear modeling on regression, while remaining less flexible than tree-based predictors. Its clearest advantage appears under deployment-time perturbations: across three classification datasets, FlagGAM has the smallest mean AUROC degradation under missingness and numerical noise. Flexible heads improve absolute accuracy and approach strong tree-based baselines, but should be interpreted as nonlinear predictors over learned rule bases. These results support FlagGAM as a constrained additive rule-basis model for applications that need readable rules and stable behavior with incomplete inputs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。