arXiv:2410.05289cs.AIcs.LG2024-10被引 8

提出可解释药物发现新方法MARS,识别并修正模型推理偏差。

MARS: A neurosymbolic approach for interpretable drug discovery

  • 结合逻辑规则与学习权重,构建神经符号模型
  • 发现模型依赖节点度数而非生物学规则进行预测
  • 确保预测结果符合已知药物作用机制,适合医药研究

神经符号(NeSy)人工智能融合逻辑规则与神经网络,在生物医学应用中具备更强可解释性。为评估药物发现中的可解释性,我们设计了一项新预测任务——药物作用机制(MoA)去卷积,并构建专用知识图谱MoA-net。在此基础上,提出摩阿检索系统(MARS),一种基于逻辑规则与学习规则权重的神经符号方法。实验发现,现有NeSy模型在知识图谱上易受‘度数偏差’影响,即预测由节点连接数主导而非生物学规则。我们提出识别并缓解该问题的方法。最终,MARS性能媲美当前最先进模型,且生成的解释与已知药物作用机制一致,表明其预测具有生物学意义,可为下游药物研发提供可靠依据。

原文摘要 · Abstract (English)

Background: Neurosymbolic (NeSy) artificial intelligence describes the combination of logic or rule-based techniques with neural networks. Compared to neural approaches, NeSy methods often possess enhanced interpretability, which is particularly promising for biomedical applications like drug discovery. However, no clear guidelines exist to assess the biological plausibility of model interpretations. Methods: To assess interpretability in the context of drug discovery, we devise a novel prediction task, called drug mechanism-of-action (MoA) deconvolution, with an associated, tailored knowledge graph (KG), MoA-net. We then develop the MoA Retrieval System (MARS), a NeSy approach for drug discovery which leverages logical rules with learned rule weights. Results: Using MARS' interpretable features alongside domain knowledge, we find that MARS and other NeSy approaches on KGs are susceptible to reasoning shortcuts, in which the prediction of true labels is driven by ``degree-bias'' rather than the domain-based rules. Subsequently, we demonstrate ways to identify and mitigate this. Thereafter, MARS achieves performance on par with current state-of-the-art models while producing model interpretations aligned with known MoAs. Conclusion: Through MARS, we showcase the novel task of computational MoA deconvolution. Our results emphasize the importance of using interpretable models, like NeSy ones, for applications in drug discovery. Specifically, by identifying and mitigating reasoning shortcuts, MARS MoA predictions which are biologically meaningful and, therefore, more reliable for downstream drug discovery research.

可解释AI药物发现知识图谱神经符号

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