LLM2D
GraphPINE: 基于图重要性传播的可解释药物响应预测
GraphPINE: Graph Importance Propagation for Interpretable Drug Response Prediction
作者: Yoshitaka Inoue, Tianfan Fu, Augustin Luna
发布日期: 4/9/2025
arXiv ID: oai:arXiv.org:2504.05454v1

摘要

arXiv:2504.05454v1 交叉类型 摘要:可解释性对于生物医学研究中的许多任务都是必要的。最近的可解释性方法主要集中在注意力、梯度和Shapley值上。这些方法不适用于具有强烈相关先验知识的数据,并且无法根据已知的预测特征之间的关系来约束可解释性结果。 我们提出了GraphPINE,这是一种利用领域特定先验知识初始化节点重要性优化的图神经网络(GNN)架构,用于药物反应预测。通常,一个手动的预测后步骤会查阅文献(即先验知识),以了解返回的预测特征。虽然在预测后可以获得梯度和注意力的节点重要性,但这些方法缺乏互补的先验知识;GraphPINE旨在克服这一限制。GraphPINE与其它GNN门控方法的不同之处在于,它利用了一种类似LSTM的序列格式。我们引入了一个重要性传播层,该层统一了1) 特征矩阵和节点重要性的更新,并且2) 利用了基于GNN的图传播特征值。这种初始化和更新机制使得特征学习更加有见地,从而提高了图的表示能力。 我们使用药物筛选和基因数据在包含超过5,000个基因节点的基因-基因图中应用GraphPINE,初始重要性基于药物-靶点相互作用(DTI)图。基因-基因图和DTIs是从已整理的来源获取的,并根据讨论药物和基因之间关系的文章数量加权。GraphPINE在952种药物上的PR-AUC值为0.894,ROC-AUC值为0.796。代码可在https://anonymous.4open.science/r/GraphPINE-40DE 获取。