arXiv:2507.17775q-bio.QMcs.AI2025-07被引 1

对比三种图神经网络在不同数据量毒理学数据上的表现,发现数据多时用GIN,少时用GAT更优。

Comparison of Optimised Geometric Deep Learning Architectures, over Varying Toxicological Assay Data Environments

  • 针对7个不同数据量的毒理学数据集,优化比较GCN、GAT、GIN三类图神经网络性能。
  • 数据丰富时GIN平均AUC达0.849,优于GCN和GAT;数据稀少时GAT显著胜出。
  • 揭示GIN与GAT在超参数空间中行为差异,说明其算法特性独特,适合不同数据场景。

几何深度学习是人工智能驱动化学生物信息学的新兴技术,但不同图神经网络(GNN)架构在此领域的独特影响尚未深入探索。本研究比较了图卷积网络(GCNs)、图注意力网络(GATs)和图同构网络(GINs)在7个不同数据量与终点的毒理学检测数据集上的表现,用于二分类预测检测激活。经分子图预处理、类别平衡及5折分层后,对每种GNN在每个数据集上执行贝叶斯优化(共21次独立优化)。优化后的模型在各折平均AUC介于0.728至0.849之间,随数据集和模型而异。在数据最丰富的前5个数据集中,GIN始终优于GCN和GAT;而在数据最稀缺的2个数据集中,GAT显著占优。这表明GIN在数据充足环境下更优,而GAT在数据稀缺时更具优势。进一步分析高维超参数空间及最优状态发现,GCN与GAT的优化结果更接近,而GIN则表现出明显差异,凸显其作为GNN算法的独特性。

原文摘要 · Abstract (English)

Geometric deep learning is an emerging technique in Artificial Intelligence (AI) driven cheminformatics, however the unique implications of different Graph Neural Network (GNN) architectures are poorly explored, for this space. This study compared performances of Graph Convolutional Networks (GCNs), Graph Attention Networks (GATs) and Graph Isomorphism Networks (GINs), applied to 7 different toxicological assay datasets of varying data abundance and endpoint, to perform binary classification of assay activation. Following pre-processing of molecular graphs, enforcement of class-balance and stratification of all datasets across 5 folds, Bayesian optimisations were carried out, for each GNN applied to each assay dataset (resulting in 21 unique Bayesian optimisations). Optimised GNNs performed at Area Under the Curve (AUC) scores ranging from 0.728-0.849 (averaged across all folds), naturally varying between specific assays and GNNs. GINs were found to consistently outperform GCNs and GATs, for the top 5 of 7 most data-abundant toxicological assays. GATs however significantly outperformed over the remaining 2 most data-scarce assays. This indicates that GINs are a more optimal architecture for data-abundant environments, whereas GATs are a more optimal architecture for data-scarce environments. Subsequent analysis of the explored higher-dimensional hyperparameter spaces, as well as optimised hyperparameter states, found that GCNs and GATs reached measurably closer optimised states with each other, compared to GINs, further indicating the unique nature of GINs as a GNN algorithm.

图神经网络毒理学模型比较数据稀缺

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