System and method for molecular property prediction using edge conditioned identity mapping convolution neural network

Abstract

This disclosure relates generally to system and method for molecular property prediction. Typically, message-pooling mechanism employed in molecular property prediction using conventional message passing neural networks (MPNN) causes over smoothing of the node embeddings of the molecular graph. The disclosed system utilizes edge conditioned identity mapping convolution neural network for the message passing phase. In message passing phase, the system computes an incoming aggregated message vector for each node of the plurality of nodes of the molecular graph based on encoded message received from neighboring nodes such that encoded message vector is generated by fusing a node information and an connecting edge information of the set of neighboring nodes of the node. The incoming aggregated message vector is utilized for computing updated hidden state vector of each node. A discriminative graph-level vector representation is computed by pooling the updated hidden state vectors from all the nodes of the molecular graph.