arXiv:2507.18926cs.LG2025-07被引 3

融合三维几何信息的图神经网络,提升脑屏障渗透性预测准确率

Geometric Multi-color Message Passing Graph Neural Networks for Blood-brain Barrier Permeability Prediction

  • 基于原子类型构建加权彩色子图,显式建模空间关系与化学上下文
  • 在三个基准数据集上实现AUC-ROC达0.947,回归任务RMSE为0.5628
  • 适用于新药研发中血脑屏障穿透性评估,尤其适合含罕见功能基团分子

准确预测血脑屏障渗透性(BBBP)对中枢神经系统药物开发至关重要。尽管图神经网络(GNN)已推动分子性质预测进展,但其通常依赖分子拓扑结构,忽视了影响运输机制的三维几何信息。本文提出几何多色消息传递图神经网络(GMC-MPNN),通过显式引入原子级几何特征和长程相互作用,增强标准消息传递架构。模型基于原子类型构建加权彩色子图,捕捉调控BBBP的空间关系与化学背景。在三个基准数据集上进行分类与回归任务评估,采用严格的骨架划分策略确保泛化能力验证。结果表明,GMC-MPNN持续优于现有先进模型,在分类任务中取得AUC-ROC 0.947和0.9212,在回归任务中实现RMSE 0.5628、皮尔逊相关系数0.6947。消融实验量化了特定原子对相互作用的影响,揭示模型性能源于对常见及罕见但化学重要的功能基团的联合学习。通过整合空间几何信息,GMC-MPNN确立了新性能标杆,为药物发现流程提供更精准通用的工具。

原文摘要 · Abstract (English)

Accurate prediction of blood-brain barrier permeability (BBBP) is essential for central nervous system (CNS) drug development. While graph neural networks (GNNs) have advanced molecular property prediction, they often rely on molecular topology and neglect the three-dimensional geometric information crucial for modeling transport mechanisms. This paper introduces the geometric multi-color message-passing graph neural network (GMC-MPNN), a novel framework that enhances standard message-passing architectures by explicitly incorporating atomic-level geometric features and long-range interactions. Our model constructs weighted colored subgraphs based on atom types to capture the spatial relationships and chemical context that govern BBB permeability. We evaluated GMC-MPNN on three benchmark datasets for both classification and regression tasks, using rigorous scaffold-based splitting to ensure a robust assessment of generalization. The results demonstrate that GMC-MPNN consistently outperforms existing state-of-the-art models, achieving superior performance in both classifying compounds as permeable/non-permeable (AUC-ROC of 0.947 and 0.9212) and in regressing continuous permeability values (RMSE of 0.5628, Pearson correlation of 0.6947). An ablation study further quantified the impact of specific atom-pair interactions, revealing that the model's predictive power derives from its ability to learn from both common and rare, but chemically significant, functional motifs. By integrating spatial geometry into the graph representation, GMC-MPNN sets a new performance benchmark and offers a more accurate and generalizable tool for drug discovery pipelines.

图神经网络药物发现几何建模渗透性预测

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