CTAGE: Curvature-Based Topology-Aware Graph Embedding for Learning Molecular Representations

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chen, Yili, Li, Zhengyu, Wan, Zheng, Yu, Hui, Wei, Xian
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916095769706496
author Chen, Yili
Li, Zhengyu
Wan, Zheng
Yu, Hui
Wei, Xian
author_facet Chen, Yili
Li, Zhengyu
Wan, Zheng
Yu, Hui
Wei, Xian
contents AI-driven drug design relies significantly on predicting molecular properties, which is a complex task. In current approaches, the most commonly used feature representations for training deep neural network models are based on SMILES and molecular graphs. While these methods are concise and efficient, they have limitations in capturing complex spatial information. Recently, researchers have recognized the importance of incorporating three-dimensional information of molecular structures into models. However, capturing spatial information requires the introduction of additional units in the generator, bringing additional design and computational costs. Therefore, it is necessary to develop a method for predicting molecular properties that effectively combines spatial structural information while maintaining the simplicity and efficiency of graph neural networks. In this work, we propose an embedding approach CTAGE, utilizing $k$-hop discrete Ricci curvature to extract structural insights from molecular graph data. This effectively integrates spatial structural information while preserving the training complexity of the network. Experimental results indicate that introducing node curvature significantly improves the performance of current graph neural network frameworks, validating that the information from k-hop node curvature effectively reflects the relationship between molecular structure and function.
format Preprint
id arxiv_https___arxiv_org_abs_2307_13275
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle CTAGE: Curvature-Based Topology-Aware Graph Embedding for Learning Molecular Representations
Chen, Yili
Li, Zhengyu
Wan, Zheng
Yu, Hui
Wei, Xian
Machine Learning
Artificial Intelligence
Quantitative Methods
AI-driven drug design relies significantly on predicting molecular properties, which is a complex task. In current approaches, the most commonly used feature representations for training deep neural network models are based on SMILES and molecular graphs. While these methods are concise and efficient, they have limitations in capturing complex spatial information. Recently, researchers have recognized the importance of incorporating three-dimensional information of molecular structures into models. However, capturing spatial information requires the introduction of additional units in the generator, bringing additional design and computational costs. Therefore, it is necessary to develop a method for predicting molecular properties that effectively combines spatial structural information while maintaining the simplicity and efficiency of graph neural networks. In this work, we propose an embedding approach CTAGE, utilizing $k$-hop discrete Ricci curvature to extract structural insights from molecular graph data. This effectively integrates spatial structural information while preserving the training complexity of the network. Experimental results indicate that introducing node curvature significantly improves the performance of current graph neural network frameworks, validating that the information from k-hop node curvature effectively reflects the relationship between molecular structure and function.
title CTAGE: Curvature-Based Topology-Aware Graph Embedding for Learning Molecular Representations
topic Machine Learning
Artificial Intelligence
Quantitative Methods
url https://arxiv.org/abs/2307.13275