用可学习的斜向规则提升模型简洁性与准确性
Sparse Oblique Rule Boosting for Simpler Additive Rule Ensembles

- 引入可学习的稀疏线性变换规则,突破轴对齐限制
- 14个任务中模型复杂度更低,精度相当或更优
- 适合需要高可解释性且避免人工特征工程的场景
小规模符号规则集成模型具有良好的可解释性。传统方法使用单变量阈值条件(如 $x \geq t$)构成轴对齐的决策区域,虽易学习但依赖高质量输入特征。若缺乏此类特征,需增加规则数量和复杂度,削弱可解释性。本文提出基于梯度提升的稀疏斜向规则增强方法,采用可学习的稀疏线性组合规则(如 $\mathbf{x}^T\mathbf{w} \geq t$),使决策区域为一般多面体。在14个回归与分类任务上的实验表明,该方法在保持相似或更优预测性能的同时,显著降低模型复杂度,提升了可解释性与准确性之间的平衡,并减少对手工特征工程的依赖。
原文摘要 · Abstract (English)
Small additive ensembles of symbolic rules offer interpretable prediction models. Traditionally, these ensembles use rule conditions based on conjunctions of simple threshold propositions $x \geq t$ on a single input variable $x$ and threshold $t$, resulting geometrically in axis-parallel polytopes as decision regions. While this form ensures a high degree of interpretability for individual rules and can be learned efficiently using the gradient boosting approach, it relies on having access to a curated set of expressive input features so that a small ensemble of axis-parallel regions can describe the target variable well. Absent such features, reaching sufficient accuracy requires increasing the number and complexity of individual rules, which diminishes the interpretability of the model. Here, we extend classical rule ensembles by introducing logical propositions with learnable sparse linear transformations of input variables, i.e., propositions of the form $\mathbf{x}^T\mathbf{w} \geq t$, where $\mathbf{w}$ is a learnable sparse weight vector, enabling decision regions as general polyhedrons with oblique faces. We propose a learning method using gradient boosting based on a weighted logistic regression. Empirical results across 14 regression and classification tasks demonstrate that the proposed method achieves lower model complexity than competitive baselines while maintaining similar or better predictive accuracy. Hence, the approach provides a favorable trade-off between interpretability and accuracy and reduces the reliance on manual feature engineering.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。