用领域知识提升传染病模型的预测能力和科学合理性
SEANN: A Domain-Informed Neural Network for Epidemiological Insights
- 将元分析中的效应量作为先验知识融入神经网络训练
- 在数据少且噪声多的情况下,预测性能和可解释性显著提升
- 适合关注公共卫生建模与可解释AI的研究者
在流行病学中,传统统计方法如逻辑回归、线性回归等常用于分析预测因子与健康结果的关系。然而,非参数机器学习技术(如深度神经网络)结合可解释AI工具,为该任务带来新机遇。尽管潜力巨大,这些方法受限于高质量、高数量数据的缺乏。为此,本文提出SEANN,一种利用领域知识——合并效应量(Pooled Effect Sizes, PES)的新型神经网络。PES常见于发表的元分析研究中,代表科学共识的量化形式。通过自定义损失函数直接整合到学习过程中,实验表明,在数据稀缺且含噪的场景下,SEANN相比无领域知识的神经网络,在预测泛化能力及提取关系的科学合理性方面均有显著提升。
原文摘要 · Abstract (English)
In epidemiology, traditional statistical methods such as logistic regression, linear regression, and other parametric models are commonly employed to investigate associations between predictors and health outcomes. However, non-parametric machine learning techniques, such as deep neural networks (DNNs), coupled with explainable AI (XAI) tools, offer new opportunities for this task. Despite their potential, these methods face challenges due to the limited availability of high-quality, high-quantity data in this field. To address these challenges, we introduce SEANN, a novel approach for informed DNNs that leverages a prevalent form of domain-specific knowledge: Pooled Effect Sizes (PES). PESs are commonly found in published Meta-Analysis studies, in different forms, and represent a quantitative form of a scientific consensus. By direct integration within the learning procedure using a custom loss, we experimentally demonstrate significant improvements in the generalizability of predictive performances and the scientific plausibility of extracted relationships compared to a domain-knowledge agnostic neural network in a scarce and noisy data setting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。