arXiv:2504.02014q-bio.BMcs.AI2025-04被引 3

用超图神经网络融合药物分子结构特征,提升靶点结合亲和力预测精度。

HCAF-DTA: drug-target binding affinity prediction with cross-attention fused hypergraph neural networks

  • 构建药物分子超图,通过树分解捕捉子结构与全局化学特征。
  • 在Davis和KIBA数据集上MSE分别达0.198和0.122,较最优基线提升4%。
  • 适合药物设计、AI制药领域研究者关注,尤其关注分子互作建模者。

准确预测药物与靶标蛋白之间的结合亲和力是计算机辅助药物设计的核心任务。现有深度学习方法往往忽略药物分子内部子结构特征及药物-靶标相互作用信息,导致预测性能受限。本文提出基于交叉注意力融合超图神经网络的药物-靶标关联预测模型HCAF-DTA。模型创新性地在特征提取阶段引入超图表示:基于树分解算法构建药物分子超图,通过跳跃连接将超图神经网络与图神经网络融合,提取子结构与全局特征,其中超边可高效表征功能基团等关键化学特征;针对蛋白质特征提取,基于ESM模型预测的残基接触图构建加权图,并使用多层图神经网络捕捉空间依赖性。在预测阶段,设计双向多头交叉注意力机制,从原子与氨基酸双视角建模分子间相互作用,通过注意力融合跨模态相关特征。在Davis和KIBA等基准数据集上的实验表明,HCAF-DTA在所有三项性能评估指标上均优于现有最先进方法,其MSE分别达到0.198和0.122,相较最优基线最高提升4%。

原文摘要 · Abstract (English)

Accurate prediction of the binding affinity between drugs and target proteins is a core task in computer-aided drug design. Existing deep learning methods tend to ignore the information of internal sub-structural features of drug molecules and drug-target interactions, resulting in limited prediction performance. In this paper, we propose a drug-target association prediction model HCAF-DTA based on cross-attention fusion hypergraph neural network. The model innovatively introduces hypergraph representation in the feature extraction stage: drug molecule hypergraphs are constructed based on the tree decomposition algorithm, and the sub-structural and global features extracted by fusing the hypergraph neural network with the graphical neural network through hopping connections, in which the hyper edges can efficiently characterise the functional functional groups and other key chemical features; for the protein feature extraction, a weighted graph is constructed based on the residues predicted by the ESM model contact maps to construct weighted graphs, and multilayer graph neural networks were used to capture spatial dependencies. In the prediction stage, a bidirectional multi-head cross-attention mechanism is designed to model intermolecular interactions from the dual viewpoints of atoms and amino acids, and cross-modal features with correlated information are fused by attention. Experiments on benchmark datasets such as Davis and KIBA show that HCAF-DTA outperforms state of the arts in all three performance evaluation metrics, with the MSE metrics reaching 0.198 and 0.122, respectively, with an improvement of up to 4% from the optimal baseline.

药物设计超图神经网络亲和力预测交叉注意力

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