构建可定制的分子手性数据集,用于评测GNN预测能力。
Chi-Geometry: A Library for Benchmarking Chirality Prediction of GNNs
- 生成带指定几何拓扑特征的合成图,隔离手性判断难点。
- 新架构实现全连接精准预测,另一架构引入虚拟节点保持线性计算开销。
- 适合研究分子图神经网络架构与手性预测的科研人员使用。
我们提出Chi-Geometry——一个用于测试和基准评估GNN手性预测能力的图数据生成库。该库可生成具有(i)用户指定几何与拓扑特性的合成图样本,以分离特定类型样本;(ii)随机化节点位置与种类,减少非相关关联。每个生成图仅含一个标记为R或S的手性中心,其余节点均标记为N/A(非手性)。这些样本被整合成连贯数据集,可用于评估GNN在节点分类任务中预测手性的能力。该方法支持更可解释、干扰更少的基准测试,指导新GNN架构设计。我们通过生成合成数据集,对多种先进GNN架构进行基准测试,结果引导我们设计出两种新架构:第一种采用全连接结构,在所有挑战性配置下准确预测手性,但训练与推理开销随节点数呈二次增长;第二种通过在原图结构中引入虚拟节点,避免全连接,使训练与推理开销保持线性增长,同时仍达到与现有最先进模型相当的手性检测精度。
原文摘要 · Abstract (English)
We introduce Chi-Geometry - a library that generates graph data for testing and benchmarking GNNs' ability to predict chirality. Chi-Geometry generates synthetic graph samples with (i) user-specified geometric and topological traits to isolate certain types of samples and (ii) randomized node positions and species to minimize extraneous correlations. Each generated graph contains exactly one chiral center labeled either R or S, while all other nodes are labeled N/A (non-chiral). The generated samples are then combined into a cohesive dataset that can be used to assess a GNN's ability to predict chirality as a node classification task. Chi-Geometry allows more interpretable and less confounding benchmarking of GNNs for prediction of chirality in the graph samples which can guide the design of new GNN architectures with improved predictive performance. We illustrate Chi-Geometry's efficacy by using it to generate synthetic datasets for benchmarking various state-of-the-art (SOTA) GNN architectures. The conclusions of these benchmarking results guided our design of two new GNN architectures. The first GNN architecture established all-to-all connections in the graph to accurately predict chirality across all challenging configurations where previously tested SOTA models failed, but at a computational cost (both for training and inference) that grows quadratically with the number of graph nodes. The second GNN architecture avoids all-to-all connections by introducing a virtual node in the original graph structure of the data, which restores the linear scaling of training and inference computational cost with respect to the number of nodes in the graph, while still ensuring competitive accuracy in detecting chirality with respect to SOTA GNN architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。