改进贝叶斯网络学习,实现快速可解释的临床分类。
Parallel Adaptive Multi-Objective Evolutionary Learning of Discretized Bayesian Network Classifiers for Clinical Data

- 并行自适应多目标优化,提升训练效率。
- 在16核CPU上提速超54倍,预测性能优于或等同于经典模型。
- 生成紧凑可读的临床可解释网络,适合医疗决策支持。
贝叶斯网络(BNs)因其透明的概率建模,在可解释人工智能中备受关注,适用于临床决策支持。近期提出的多目标进化算法Baymex能学习离散化贝叶斯网络,支持专家在似然性、模型复杂度和先验信念间权衡。尽管Baymex已优于现有方法,但仍存在计算耗时长、仅在合成数据上验证的问题。为此,本文引入并行策略与自适应防过拟合机制,并将Baymex重构为通过交叉熵损失与BIC复杂度项的多目标优化来训练贝叶斯网络分类器,以评估其在真实临床任务中的表现。实验在两个开源数据集(RADCURE、SUPPORT)及一个内部数据集上进行,结果显示:在16核CPU上加速超过54倍;相比决策树、逻辑回归、朴素贝叶斯和随机森林等常见基线模型,Baymex在预测性能上达到统计相当或更优,同时生成结构紧凑、临床可解读的贝叶斯网络;更重要的是,所发现的多个合理网络均包含与既定临床因素一致的预测变量。
原文摘要 · Abstract (English)
Bayesian Networks (BNs) are of interest from an explainable AI viewpoint, offering transparent probabilistic models for decision support. Baymex is a recently introduced multi-objective evolutionary algorithm for learning discretized BNs, enabling experts to trade-off different objectives of interest, such as likelihood, model complexity, and prior beliefs. While Baymex has been shown to outperform state-of-the-art BN learning approaches, Baymex still 1) requires a lot of computation time and 2) has only been evaluated on synthetic data. To improve scalability, we introduce a parallelization strategy as well as a mechanism that enables adaptively steering optimization toward networks that overfit less. We furthermore reconfigure Baymex to train a BN classifier through multi-objective optimization of cross-entropy loss and the BIC complexity term so as to evaluate its performance on real-world clinical classification tasks. Besides observing speedups up to over 54 times on a 16-core CPU, comparisons against clinically familiar baselines (decision trees, logistic regression, naive Bayes, and random forests) on two open-source (RADCURE and SUPPORT) and one in-house dataset, show that Baymex obtains statistically similar or better predictive performance while producing compact, clinically inspectable BNs. Importantly, Baymex finds multiple plausible BN classifiers that contain predictors consistent with established clinical factors.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。