Quantum-Embedded Graph Neural Network Architecture for Molecular Property Prediction.

Journal: Journal of chemical information and modeling
Published Date:

Abstract

Accurate prediction of molecular properties is crucial for accelerating the development of new drugs, and quantum machine learning (QML) holds great promise in this domain. A typical QML pipeline comprises two core stages: encoding classical data into quantum representations followed by training and prediction using quantum computing-based machine learning (ML) models. In this article, we focus on the initial encoding stage and propose an effective quantum feature extraction approach for molecular graph data, introducing quantum node embedding and quantum edge embedding methods. We developed a hybrid quantum-classical ML framework and implemented several quantum-embedded graph neural network (QEGNN) models to evaluate the proposed method. Experiments conducted on three benchmark data sets with diverse molecular property prediction tasks demonstrate that QEGNN models consistently achieve higher accuracy, improved stability, and significantly reduced parameter complexity─hallmarks of quantum advantage. Furthermore, we validate the reliability of the quantum embedding approach on the superconducting quantum processor "Wukong," showing that the models retain stable performance even under the constraints of current noisy quantum hardware. This work highlights the potential of QML and paves the way for the development of universal QML models.

Authors

  • Min Lu
    Hangzhou Science and Technology Information Institute, Hangzhou, Zhejiang, China.
  • Lei Du
    School of Mathematical Sciences, Dalian University of Technology, Chuangxinyuan Building, No.2 Linggong Road, Ganjingzi District, Dalian, 116024, Liaoning, China.
  • Ziwei Cui
    School of Electronic Information and Communications, Huazhong University of Science and Technology, Wuhan 430074, China.
  • Yiming Zhao
    School of Software, Taiyuan University of Technology, Taiyuan, China.
  • Qipeng Yan
    Origin Quantum Computing Company Limited, Hefei, Anhui 230026, China.
  • Jianyu Zhao
    Origin Quantum Computing Company Limited, Hefei, Anhui 230026, China.
  • Ye Li
    Environment and Plant Protection Institute, Chinese Academy of Tropical Agricultural Science, Haikou 571010, People's Republic of China; Key Laboratory of Monitoring and Control of Tropical Agricultural and Forest Invasive Alien Pests, Ministry of Agriculture, Haikou 571010, People's Republic of China.
  • Menghan Dou
    Origin Quantum Computing Company Limited, Hefei, Anhui 230026, China.
  • Qingchun Wang
    Institute of Artificial Intelligence, Hefei Comprehensive National Science Center, Hefei, Anhui 230088, China.
  • Yu-Chun Wu
    Value Lab, Acer Inc., Taipei, Taiwan.
  • Guo-Ping Guo
    Origin Quantum Computing Company Limited, Hefei, Anhui 230026, China.