通过特征图提升规则模型可解释性,助力医疗决策洞察
Enhancing interpretability of rule-based classifiers through feature graphs
- 构建基于图的特征可视化方法,解析规则间特征交互
- 提出与模型无关的特征重要性度量,在4种方法上验证有效
- 可发现新风险因子和生物标志物,适合医疗领域研究者使用
在医疗等对透明性要求高的领域,规则系统因可解释性强常优于黑箱模型。但随着规则复杂度上升,关键特征识别、交互理解及跨规则集比较日益困难。为此,本文提出一套完整框架:引入基于图的特征可视化策略,设计不依赖规则预测器的新特征重要性度量,以及基于特征贡献的规则集距离度量。在两个临床数据集和四种规则模型(决策树、逻辑学习机、关联规则、带规则提取的神经网络)上实验,揭示了临床特征组合预测价值的新见解,可在数据集和类别层面发现新风险因素、标志性基因及潜在生物标志物,并确定应优先采集的患者信息以提升诊断准确率。在15个公开基准上的对比分析显示,该方法性能竞争力强且鲁棒性更优。代码已开源:https://github.com/ChristelSirocchi/rule-graph。
原文摘要 · Abstract (English)
In domains where transparency and trustworthiness are crucial, such as healthcare, rule-based systems are widely used and often preferred over black-box models for decision support systems due to their inherent interpretability. However, as rule-based models grow complex, discerning crucial features, understanding their interactions, and comparing feature contributions across different rule sets becomes challenging. To address this, we propose a comprehensive framework for estimating feature contributions in rule-based systems, introducing a graph-based feature visualisation strategy, a novel feature importance metric agnostic to rule-based predictors, and a distance metric for comparing rule sets based on feature contributions. By experimenting on two clinical datasets and four rule-based methods (decision trees, logic learning machines, association rules, and neural networks with rule extraction), we showcase our method's capability to uncover novel insights on the combined predictive value of clinical features, both at the dataset and class-specific levels. These insights can aid in identifying new risk factors, signature genes, and potential biomarkers, and determining the subset of patient information that should be prioritised to enhance diagnostic accuracy. Comparative analysis of the proposed feature importance score with state-of-the-art methods on 15 public benchmarks demonstrates competitive performance and superior robustness. The method implementation is available on GitHub: https://github.com/ChristelSirocchi/rule-graph.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。