ChemXTree: A Feature-Enhanced Graph Neural Network-Neural Decision Tree Framework for ADMET Prediction.

Journal: Journal of chemical information and modeling
Published Date:

Abstract

The rapid progression of machine learning, especially deep learning (DL), has catalyzed a new era in drug discovery, introducing innovative approaches for predicting molecular properties. Despite the many methods available for feature representation, efficiently utilizing rich, high-dimensional information remains a significant challenge. Our work introduces ChemXTree, a novel graph-based model that integrates a Gate Modulation Feature Unit (GMFU) and neural decision tree (NDT) in the output layer to address this challenge. Extensive evaluations on benchmark data sets, including MoleculeNet and eight additional drug databases, have demonstrated ChemXTree's superior performance, surpassing or matching the current state-of-the-art models. Visualization techniques clearly demonstrate that ChemXTree significantly improves the separation between substrates and nonsubstrates in the latent space. In summary, ChemXTree demonstrates a promising approach for integrating advanced feature extraction with neural decision trees, offering significant improvements in predictive accuracy for drug discovery tasks and opening new avenues for optimizing molecular properties.

Authors

  • Yuzhi Xu
    Shanghai Frontiers Science Center of Artificial Intelligence and Deep Learning and NYU-ECNU Center for Computational Chemistry, NYU Shanghai, Shanghai 200062, China.
  • Xinxin Liu
    Department of Nutrition and food hygiene, College of Public Health of Zhengzhou University, Zhengzhou, China, 450001. Electronic address: 852284114@qq.com.
  • Wei Xia
    Key Laboratory of Environment and Health, Ministry of Education & Ministry of Environmental Protection, State Key Laboratory of Environmental Health (Incubation), School of Public Health, Tongji Medical College, Huazhong University of Science and Technology, Wuhan, China.
  • Jiankai Ge
    Chemical and Biomolecular Engineering, University of Illinois at Urbana-Champaign, Urbana, Illinois 61801, United States.
  • Cheng-Wei Ju
    College of Chemistry, Nankai University, Tianjin 300071, China.
  • Haiping Zhang
    Oil Crops Research Institute, Chinese Academy of Agricultural Sciences, Key Laboratory of Oilseeds processing, Ministry of Agriculture, Oil Crops and Lipids Process Technology National & Local Joint Engineering Laboratory, Hubei Key Laboratory of Lipid Chemistry and Nutrition, Wuhan 430062, China.
  • John Z H Zhang
    Shanghai Engineering Research Center of Molecular Therapeutics & New Drug Development, School of Chemistry and Molecular Engineering, East China Normal University, Shanghai 200062, China.