arXiv:2606.20906cs.LGcs.AI2026-06

将分子图按原子类型对拆分,提升性质预测精度。

MMGNN: Multi-level, multi-color graph neural networks for molecular property prediction

论文配图:MMGNN: Multi-level, multi-color graph neural networks for molecular property prediction
图 1 · 摘自论文原文
  • 按化学键和空间关系构建多层级、多颜色子图,保留原子级细节。
  • 在5个分类与3个回归任务上表现优异,2D/3D模型各有优势。
  • 适合需要精细分子相互作用建模的研究者使用。

分子消息传递神经网络通常通过单一图传播化学多样性的相互作用,可能混杂特定信号且需深层传播以捕捉长程效应。我们提出多层级、多颜色图神经网络(MMGNN),一种分层框架,将分子图分解为重叠的原子类型对特异性子图,同时保持原子级分辨率。MMGNN-2D从共价连接构建化学着色子图,而MMGNN-3D基于空间邻近性构建几何着色子图,并用距离、角度和扭转描述符增强边信息。两种变体对每个子图应用共享的消息传递主干,通过原子级聚合和分子读出组合表示。我们在MoleculeNet的五个分类和三个回归基准上评估,采用常见骨架划分及五次独立运行。MMGNN-2D在分类数据集上取得最高宏平均AUC-ROC 0.838,ESOL任务上最低RMSE 0.803。MMGNN-3D在BBBP任务上获得最高平均AUC-ROC 0.956,FreeSolv任务上最低RMSE 1.793,表明拓扑与几何表示具有互补优势。结构分析与留一分析进一步揭示子图分解如何影响学习表征与原子类型对敏感性。这些结果支持重叠的交互特异性图分解作为分子性质预测的有力策略。

原文摘要 · Abstract (English)

Molecular message-passing neural networks commonly propagate chemically diverse interactions through a single graph, which may mix interaction-specific signals and require deep propagation to capture long-range effects. We introduce the Multi-level, Multi-color Graph Neural Network (MMGNN), a hierarchical framework that decomposes a molecular graph into overlapping atom-type-pair-specific subgraphs while preserving atom-level resolution. MMGNN-2D constructs chemical-colored subgraphs from covalent connectivity, whereas MMGNN-3D constructs geometric-colored subgraphs from spatial proximity and augments their edges with distance, angular, and torsional descriptors. Both variants apply a shared communicative message-passing backbone to each subgraph and combine the resulting representations through atom-wise aggregation and molecular readout. We evaluated MMGNN on five classification and three regression benchmarks from MoleculeNet using common scaffold splits and five independent runs. MMGNN-2D achieved the highest macro-average AUC-ROC of 0.838 across the classification datasets and the lowest RMSE on ESOL (0.803). MMGNN-3D obtained the highest mean AUC-ROC on BBBP (0.956) and the lowest RMSE on FreeSolv (1.793), indicating complementary strengths of topological and geometric representations. Structural and leave-one-out analyses further illustrate how the subgraph decomposition affects learned representations and atom-type-pair sensitivities. These results support overlapping interaction-specific graph decomposition as a competitive strategy for molecular property prediction.

分子图神经网络属性预测多尺度建模化学信息学

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