Transformer Graph Neural Network Molecular Property Prediction

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Current computational methods for predicting molecular properties, such as Density Functional Theory (DFT), are time-consuming and computationally intensive, making it impractical for complex systems, thus limiting the prediction of molecular properties in fields like drug design and quantum chemistry.

Innovation Solution

A transformer-based graph neural network is trained using deep learning techniques to predict molecular properties by encoding structural information from molecular graphs, including centrality, spatial, and edge encodings, allowing it to attend to all nodes in the graph and adaptively process structural information.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Measurement precision

If conventional computational methods like Density Functional Theory (DFT) are used to predict molecular properties, then accuracy is maintained, but computational time and resource requirements become excessively high

Engineering Contradiction:
Improveprediction accuracyVSAvoidcomputational time
Core Design Contradiction:
Measurement precisionVSLoss of time

Solution Approach 1:

The patent creates a computational model (transformer-based GNN) that copies and learns from quantum mechanical calculations during training, then uses this learned model to predict molecular properties without performing actual quantum calculations. This allows the system to maintain high prediction accuracy while dramatically reducing computational time by using a simplified neural network representation instead of full DFT computations.

Inventive Principle:
Principle #26Copying

Solution Approach 2:

The patent replaces the mechanical quantum mechanical calculation system (DFT) with a neural network-based computational system. The transformer-based GNN substitutes the physics-based computational mechanics with a data-driven model that has learned the underlying patterns, achieving similar accuracy with fraction of the computational cost.

Inventive Principle:
Principle #28Mechanics substitution (Replace mechanical system)

2Productivity

If conventional graph neural networks are used to process molecular graphs, then computational efficiency is improved, but the ability to capture global structural information is limited

Engineering Contradiction:
Improvecomputational efficiencyVSAvoidglobal structural information
Core Design Contradiction:
ProductivityVSLoss of information

Solution Approach 1:

The patent makes the transformer-based GNN universally applicable to both local and global information processing. The self-attention mechanism enables the model to dynamically adapt its receptive field, allowing it to capture both local atomic interactions and global molecular structure simultaneously, making the model versatile for various molecular property prediction tasks.

Inventive Principle:
Principle #6Universality (Multi-functionality)

Solution Approach 2:

The patent adds the attention mechanism dimension to the traditional GNN architecture. Instead of only processing information through fixed sequential layers, the self-attention mechanism introduces a new computational dimension that allows nodes to directly interact with all other nodes in the graph, capturing global structural information while maintaining computational efficiency.

Inventive Principle:
Principle #17Another dimension (Dimensionality change)

Data Source

PatentUS20230402136A1Transformer-based graph neural network trained with structural information encoding
Publication Date: 2023.12.14 MICROSOFT TECHNOLOGY LICENSING LLC
  • US20230402136A1 patent drawing
  • US20230402136A1 patent drawing
  • US20230402136A1 patent drawing

AI summary

A computing system is provided, including a processor configured to, during a training phase, provide a training data set, including a pre-transformation molecular graph and post-transformation energy parameter value representing an energy change in a molecular system following an energy transformation. The pre-transformation graph includes a plurality of normal nodes connected by edges representing a distance and a bond between a pair of the normal nodes. The processor is further configured to encode structural information in each pre-transformation molecular graph as learnable embeddings, the structural information describing the relative positions of the atoms represented by the normal nodes. The structural information includes an edge encoding representing a type of bond between a pair of normal nodes in each pre-transformation molecular graph, and a spatial encoding representing a shortest path distance along the edges between a pair of normal nodes in each pre-transformation molecular graph.