arXiv:2410.08024cs.LGcs.AI2024-10被引 13

用原子量子特性预训练图变压器,提升药物ADMET预测性能

Pretraining Graph Transformers with Atom-in-a-Molecule Quantum Properties for Improved ADMET Modeling

  • 以原子级量子性质预训练图变压器模型
  • 预训练后在公开数据集上显著提升ADMET预测准确率
  • 适合药物研发中的分子性质建模与跨数据集泛化研究

我们评估了在原子级量子力学特征上预训练图变压器架构对药物类化合物吸收、分布、代谢、排泄和毒性(ADMET)性质建模的影响。对比了三种策略:基于分子量子性质(特别是HOMO-LUMO能隙)的预训练,以及自监督原子掩码技术。在治疗数据共同体(Therapeutic Data Commons)的ADMET数据集上微调后,发现基于原子量子力学性质的预训练模型整体表现更优。进一步分析潜在表示发现,监督式预训练在微调后仍保留预训练信息,不同预训练方式导致各层隐含表达能力呈现不同趋势。此外,原子级量子性质预训练模型通过注意力权重捕捉到更多输入图的低频拉普拉斯特征值模式,并更好表征分子中原子环境。该分析应用于更大规模的非公开微粒体清除率数据集,验证了所提指标的泛化能力。在此情况下,模型表现与表征分析一致,尤其凸显了在公共基准上表现相近的模型在大规模制药数据上可能存在差异。

原文摘要 · Abstract (English)

We evaluate the impact of pretraining Graph Transformer architectures on atom-level quantum-mechanical features for the modeling of absorption, distribution, metabolism, excretion, and toxicity (ADMET) properties of drug-like compounds. We compare this pretraining strategy with two others: one based on molecular quantum properties (specifically the HOMO-LUMO gap) and one using a self-supervised atom masking technique. After fine-tuning on Therapeutic Data Commons ADMET datasets, we evaluate the performance improvement in the different models observing that models pretrained with atomic quantum mechanical properties produce in general better results. We then analyse the latent representations and observe that the supervised strategies preserve the pretraining information after finetuning and that different pretrainings produce different trends in latent expressivity across layers. Furthermore, we find that models pretrained on atomic quantum mechanical properties capture more low-frequency laplacian eigenmodes of the input graph via the attention weights and produce better representations of atomic environments within the molecule. Application of the analysis to a much larger non-public dataset for microsomal clearance illustrates generalizability of the studied indicators. In this case the performances of the models are in accordance with the representation analysis and highlight, especially for the case of masking pretraining and atom-level quantum property pretraining, how model types with similar performance on public benchmarks can have different performances on large scale pharmaceutical data.

图神经网络药物发现量子化学ADMET预测

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