arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.03342q-bio.QMcs.LG
收藏

基于图残差的分子性质预测方法

Graph Residual based Method for Molecular Property Prediction

IIT Bombay(印度理工学院孟买分校);Centre for Machine Intelligence and Data Science, IIT Bombay(印度理工学院孟买分校机器智能与数据科学中心)

Kanad Sen, Saksham Gupta, Abhishek Raj, Alankar Alankar

AI总结:

针对传统分子性质预测依赖难获取的分子特征的问题,提出基于GRU的边条件残差图神经网络ECRGNN,结合VAE和端到端学习实现多类多标签预测,在多数据集上验证了回归与分类性能。

摘要:

由机器学习驱动的性质预测方法一直是研究热点。然而,在关键应用场景中,提升模型的泛化能力、准确率和推理速度仍有大量工作要做。传统机器学习模型依赖从分子中提取的特征来预测性质,而这类特征通常难以获取。本研究采用了一种新型深度学习方法——边条件残差图神经网络(Edge Conditioned Residual Graph Neural Network,ECRGNN),仅需分子的图结构即可直接预测性质。本研究使用分子的SMILES(简化分子线性输入规范,Simplified Molecular Input Line Entry System)表示作为输入数据格式,并进一步将其转换为图数据库,构成训练数据。本文详细阐述了基于GRU(门控循环单元)的新型方法ECRGNN的映射输入机制,重点分析了该方法的回归性能和分类效果。此外,还详细介绍了用于多类多标签性质预测的变分自编码器(Variational Autoencoder,VAE)及端到端学习方法。研究将所得结果与标准基准数据集及部分新开发的数据集进行了对比,明确定义了所有使用的性能指标,并说明了选择这些指标的原因。

英文摘要:

Machine learning-driven methods for property prediction have been of deep interest. However, much work remains to be done to improve the generalization ability, accuracy, and inference time for critical applications. The traditional machine learning models predict properties based on the features extracted from the molecules, which are often not easily available. In this work, a novel Deep Learning method, the Edge Conditioned Residual Graph Neural Network (ECRGNN), has been applied, allowing us to predict properties directly only the Graph-based structures of the molecules. SMILES (Simplified Molecular Input Line Entry System) representation of the molecules has been used in the present study as input data format, which has been further converted into a graph database, which constitutes the training data. This manuscript highlights a detailed description of the novel GRU-based methodology, ECRGNN, to map the inputs that have been used. Emphasis is placed on highlighting both the regressive property and the classification efficacy of the same. A detailed description of the Variational Autoencoder (VAE) and the end-to-end learning method used for multi-class multi-label property prediction has been provided as well. The results have been compared with standard benchmark datasets as well as some newly developed datasets. All performance metrics that have been used have been clearly defined, and their reason for choice.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑