arXiv:2602.20573cs.LG2026-02

对比四种GNN模型在分子回归任务中的表现,发现GCN和GIN最优。

MolGraphBench: A Benchmark of GNN Architectures for Molecular Regression Tasks

  • 构建分子图数据集基准,测试四类主流GNN架构性能。
  • GCN与GIN在三个数据集上均达最优或接近最优结果,如B3DB上RMSE为0.518。
  • 提醒应将图层类型作为可调超参数,而非固定设计选择。

分子通常以SMILES字符串表示,可转换为手工特征或指纹用于性质预测。研究表明SMILES可转化为分子图$G = (V, E)$,其中原子为节点$(V)$,化学键为边$(E)$,进而用于训练图神经网络(GNN)模型。尽管近年来已有大量新旧GNN架构应用于分子性质预测,但缺乏系统性基准评估。本文提出MolGraphBench,一个针对分子回归任务的全面基准,涵盖四种常用GNN模型。实验表明,图卷积网络(GCN)与图同构网络(GIN)在绝对性能、训练效率、迁移学习及预测质量方面表现最佳。研究还发现分子指纹在融合框架(GNN-FP)中不具互补性。此外,所提GNN模型在三个数据集上表现优于或相当当前最优基线:如在B3DB上GCN的RMSE为0.518,在FreeSolv上GIN-FP的RMSE为1.022,在RT数据集上GIN的MAE为63.783。研究建议,图层类型应作为可调超参数而非固定设计,以实现更优性能。

原文摘要 · Abstract (English)

Molecules are often represented as SMILES strings, which can be readily converted to hand-crafted descriptors or fingerprints (FP) for molecular property prediction. Research has demonstrated that SMILES can be converted to molecular graphs $G = (V, E)$, with atoms as nodes $(V)$ and bonds as edges $(E)$. These molecular graphs can subsequently be used to train graph neural networks (GNN) models. Despite the recent surge in application of GNN (existing and novel architectures) for molecular property prediction, a rigorous benchmark is still lacking. We propose MolGraphBench, a comprehensive benchmark of four commonly used GNN models for molecular property prediction. Benchmarking results demonstrate graph convolutional network (GCN) and graph isomorphism networks (GIN) as the optimal GNN architectures for molecular graph regression tasks, based on absolute performance, training efficiency, transfer learning and prediction quality. The study also indicates the non-complementary nature of molecular fingerprints in the fusion (GNN-FP) framework. Furthermore, our GNN models achieved performance superior or comparable performance to current state-of-the-art GNN baselines across three datasets (GCN with RMSE of $0.518$ on B3DB, GIN-FP with RMSE of $1.022$ on FreeSolv and GIN with MAE of $63.783$ on RT datasets). Findings from this study indicate that type of GNN-layer, should be treated as a tunable hyperparameter rather than a fixed design choice to achieve superior performance.

分子图GNN基准测试回归任务

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