Hybrid Quantum Graph Neural Network for Molecular Property Prediction

Michael Vitz,Hamed Mohammadbagherpoor,Samarth Sandeep,Andrew Vlasic,Richard Padbury,Anh Pham
2024-05-09
Abstract:To accelerate the process of materials design, materials science has increasingly used data driven techniques to extract information from collected data. Specially, machine learning (ML) algorithms, which span the ML discipline, have demonstrated ability to predict various properties of materials with the level of accuracy similar to explicit calculation of quantum mechanical theories, but with significantly reduced run time and computational resources. Within ML, graph neural networks have emerged as an important algorithm within the field of machine learning, since they are capable of predicting accurately a wide range of important physical, chemical and electronic properties due to their higher learning ability based on the graph representation of material and molecular descriptors through the aggregation of information embedded within the graph. In parallel with the development of state of the art classical machine learning applications, the fusion of quantum computing and machine learning have created a new paradigm where classical machine learning model can be augmented with quantum layers which are able to encode high dimensional data more efficiently. Leveraging the structure of existing algorithms, we developed a unique and novel gradient free hybrid quantum classical convoluted graph neural network (HyQCGNN) to predict formation energies of perovskite materials. The performance of our hybrid statistical model is competitive with the results obtained purely from a classical convoluted graph neural network, and other classical machine learning algorithms, such as XGBoost. Consequently, our study suggests a new pathway to explore how quantum feature encoding and parametric quantum circuits can yield drastic improvements of complex ML algorithm like graph neural network.
Quantum Physics,Materials Science,Machine Learning
What problem does this paper attempt to address?
The paper proposes a Hybrid Quantum Graph Neural Network (HyQCGNN) for the prediction of molecular properties, particularly the formation energy of perovskite materials. Traditional machine learning algorithms, especially Graph Neural Networks (GNNs), have demonstrated high efficiency in materials science but still suffer from limited computing resources and insufficient training data. In this paper, the researchers combine quantum computing to create a gradient-free hybrid quantum-classical convolutional graph neural network, aiming to improve the efficiency of predicting complex material properties. The main contributions of this paper include: 1. Development of the HyQCGNN model, which utilizes quantum layers for feature encoding and avoids the gradient vanishing problem in traditional quantum circuits through gradient-free optimization methods. 2. In the task of predicting the formation energy of perovskite materials, HyQCGNN performs comparably to purely classical GNNs and other machine learning algorithms such as XGBoost. 3. The study shows that quantum feature encoding and parameterized quantum circuits can significantly enhance the performance of complex machine learning algorithms like GNNs. The paper also explores the potential application of quantum machine learning in materials science, indicating that quantum computing can offer advantages in encoding high-dimensional data and simulating long-range correlations. Although the performance of HyQCGNN has not surpassed classical algorithms, it opens up new avenues for exploring how to enhance the performance of quantum graph neural networks through improved quantum feature mapping.