arXiv:2606.13295stat.MLcs.LG2026-06

提出可处理分层变量的可解释分类树,提升医学数据中的差异分析能力。

Simultaneous Latent Budget Trees for Stratified Classification

论文配图:Simultaneous Latent Budget Trees for Stratified Classification
图 1 · 摘自论文原文
  • 基于联合潜变量模型构建条件分裂规则,支持分层变量控制
  • 通过潜预算参数动态调整不同群体的类别分布特征
  • 适用于性别、时间等分层因素下的疾病进展分析,适合医学研究者

在可解释人工智能时代,单棵决策树因易于理解而再度受到关注。本文提出同时潜预算树(Simultaneous Latent Budget Trees, SLBT),一种针对存在分层因子(如时间、空间或人口统计变量)时的分类树概率建模框架,该因子作为控制变量或潜在混杂因素。标准树生长方法无法优化条件分裂规则,本文提出基于模型的分裂机制:子节点被解释为父节点上拟合的联合混合模型(如同时潜预算模型及其约束版本)的潜成分。混合参数根据各组别差异引导样本分配至子节点,而潜预算参数则更新各控制变量水平下响应类别的分布特征。参数通过最小二乘法估计,并从神经网络视角建模。可通过节点与路径的可视化辅助(包括视觉剪枝和决策树选择)实现交互式树结构解读。针对响应类别不平衡问题,提出合适度量。方法应用于肌萎缩侧索硬化症患者中性别相关的疾病进展差异研究。相关SLBT库及各类树算法已在关联GitHub仓库发布。

原文摘要 · Abstract (English)

In the era of Explainable Artificial Intelligence, there is a renewed focus on single trees for their ease of interpretation. This paper introduces Simultaneous Latent Budget Trees, a probabilistic machine learning framework for classification trees in the presence of a stratification factor such as a temporal, spatial, or demographic variable, acting as a control variable or potential confounder. Standard tree growth procedures are not designed to optimize a conditional split rule. A model-based split rule is proposed in which child nodes are interpreted as latent components of a simultaneous mixture model, such as the Simultaneous Latent Budget Model and its constrained versions, fitted to the parent node. Mixing parameters drive the observations, differently for each group, to the child nodes whereas latent budgets parameters update the response classes profile of each level of the control variable. Parameters are estimated by least squares considering a neural network perspective of the model. An informative tree structure can be interactively visualized with interpretation aids on the node and the paths, including visual pruning and decision tree selection procedure. Suitable measures are proposed to handle an unbalanced response class distribution. The proposed methodology is applied to investigate gender-related differences in disease progression of Amyotrophic Lateral Sclerosis. The SLBT library with the various tree-based algorithms is available in the linked GitHub repository.

可解释AI分类树分层分析医学数据

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。