Proceedings

A Systematic Survey of Chemical Pre-trained Models

Download BibTeX Cite

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence. Survey Track. Pages 6787-6795.

https://doi.org/10.24963/ijcai.2023/760

Abstract

Deep learning has achieved remarkable success in learning representations for molecules, which is crucial for various biochemical applications, ranging from property prediction to drug design. However, training Deep Neural Networks (DNNs) from scratch often requires abundant labeled molecules, which are expensive to acquire in the real world. To alleviate this issue, tremendous efforts have been devoted to Chemical Pre-trained Models (CPMs), where DNNs are pre-trained using large-scale unlabeled molecular databases and then fine-tuned over specific downstream tasks. Despite the prosperity, there lacks a systematic review of this fast-growing field. In this paper, we present the first survey that summarizes the current progress of CPMs. We first highlight the limitations of training molecular representation models from scratch to motivate CPM studies. Next, we systematically review recent advances on this topic from several key perspectives, including molecular descriptors, encoder architectures, pre-training strategies, and applications. We also highlight the challenges and promising avenues for future research, providing a useful resource for both machine learning and scientific communities.

  1. [Axelrod and Gómez-Bombarelli, 2022] S. Axelrod and R. Gómez- Bombarelli. GEOM, Energy-Annotated Molecular Conforma- tions for Property Prediction and Molecular Generation. Sci. Data, 2022.
  2. [Bagal et al., 2021] V. Bagal, R. Aggarwal, et al. MolGPT: Molec- ular Generation Using a Transformer-Decoder Model. J. Chem. Inf. Model., 2021.
  3. [Berman et al., 2000] H. M. Berman, J. Westbrook, et al. The Pro- tein Data Bank. Nucleic Acids Res., 2000.
  4. [Brown et al., 2020] T. B. Brown, B. Mann, et al. Language Models are Few-Shot Learners. In NeurIPS, 2020.
  5. [Chithrananda et al., 2020] S. Chithrananda, G. Grand, et al. Chem- BERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Prediction. arXiv.org, October 2020.
  6. [Consonni and Todeschini, 2009] V. Consonni and R. Todeschini. Molecular Descriptors for Chemoinformatics. 2009.
  7. [Corso et al., 2020] G. Corso, L. Cavalleri, et al. Principal Neigh- bourhood Aggregation for Graph Nets. In NeurIPS, 2020.
  8. [Degen et al., 2008] J. Degen, C. Wegscheid-Gerlach, et al. On the Art of Compiling and Using 'Drug-Like' Chemical Fragment Spaces. ChemMedChem, 2008.
  9. [Devlin et al., 2019] J. Devlin, M. Chang, et al. BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding. In NAACL, 2019.
  10. [Du et al., 2022a] W. Du, H. Zhang, et al. SE(3) Equivariant Graph Neural Networks with Complete Local Frames. In ICML, 2022.
  11. [Du et al., 2022b] Y. Du, T. Fu, et al. MolGenSurvey: A System- atic Survey in Machine Learning Models for Molecule Design. arXiv.org, March 2022.
  12. [Edwards et al., 2022] C. Edwards, T. Lai, et al. Translation be- tween Molecules and Natural Language. In EMNLP, 2022.
  13. [Fang et al., 2022a] X. Fang, L. Liu, et al. Geometry-Enhanced Molecular Representation Learning for Property Prediction. Nat. Mach. Intell., 2022.
  14. [Fang et al., 2022b] Y. Fang, Q. Zhang, et al. Molecular Contrastive Learning with Chemical Element Knowledge Graph. In AAAI, 2022.
  15. [Feng et al., 2022] J. Feng, Z. Wang, et al. MGMAE: Molecu- lar Representation Learning by Reconstructing Heterogeneous Graphs with A High Mask Ratio. In CIKM, 2022.
  16. [Gaulton et al., 2012] A. Gaulton, L. J. Bellis, et al. ChEMBL: A Large-Scale Bioactivity Database for Drug Discovery. Nucleic Acids Res., 2012.
  17. [Guo et al., 2022] Z. Guo, P. K. Sharma, et al. Multilingual Molec- ular Representation Learning via Contrastive Pre-training. In ACL, 2022.
  18. [Hassani and Ahmadi, 2020] K. Hassani and A. H. K. Ahmadi. Contrastive Multi-View Representation Learning on Graphs. In ICML, 2020.
  19. [Hawkins, 2017] P. C. D. Hawkins. Conformation Generation: The State of the Art. J. Chem. Inf. Model., 2017.
  20. [He et al., 2016] K. He, X. Zhang, et al. Deep Residual Learning for Image Recognition. In CVPR, 2016.
  21. [Hjelm et al., 2019] R. D. Hjelm, A. Fedorov, et al. Learning Deep Representations by Mutual Information Estimation and Maxi- mization. In ICLR, 2019.
  22. [Ho et al., 2020] J. Ho, A. Jain, et al. Denoising Diffusion Proba- bilistic Models. In NeurIPS, 2020.
  23. [Honda et al., 2019] S. Honda, S. Shi, et al. SMILES Transformer: Pre-trained Molecular Fingerprint for Low Data Drug Discovery. arXiv.org, November 2019.
  24. [Hou et al., 2022] Z. Hou, X. Liu, et al. GraphMAE: Self- Supervised Masked Graph Autoencoders. In KDD, 2022.
  25. [Hu et al., 2020a] W. Hu, B. Liu, et al. Strategies for Pre-training Graph Neural Networks. In ICLR, 2020.
  26. [Hu et al., 2020b] Z. Hu, Y. Dong, et al. GPT-GNN: Generative Pre-Training of Graph Neural Networks. In KDD, 2020.
  27. [Hu et al., 2021] W. Hu, M. Fey, et al. OGB-LSC: A large-scale challenge for machine learning on graphs. In NeurIPS Datasets and Benchmarks, 2021.
  28. [Hu et al., 2022] B. Hu, J. Xia, J. Zheng, C. Tan, Y. Huang, Y. Xu, and S. Z. Li. Protein language models and structure prediction: Connection and progression. arXiv preprint arXiv:2211.16742, 2022.
  29. [Huang et al., 2021] K. Huang, T. Fu, et al. Therapeutics Data Com- mons: Machine Learning Datasets and Tasks for Drug Discovery and Development. In NeurIPS Datasets and Benchmarks, 2021.
  30. [Jiao et al., 2023] R. Jiao, J. Han, et al. Energy-Motivated Equiv- ariant Pretraining for 3D Molecular Graphs. In AAAI, 2023.
  31. [Kearnes et al., 2016] S. Kearnes, K. McCloskey, et al. Molecular Graph Convolutions: Moving Beyond Fingerprints. J. Comput. Aided Mol. Des., 2016.
  32. [Kim et al., 2022] D. Kim, J. Baek, et al. Graph Self-supervised Learning with Accurate Discrepancy Learning. In NeurIPS, 2022.
  33. [Kipf and Welling, 2017] N. T. Kipf and M. Welling. Semi- Supervised Classification with Graph Convolutional Networks. In ICLR, 2017.
  34. [Krenn et al., 2020] M. Krenn, F. Häse, et al. Self-Referencing Embedded Strings (SELFIES): A 100% Robust Molecular String Representation. Mach. Learn. Sci. Technol., 2020.
  35. [Li et al., 2021] P. Li, J. Wang, et al. An Effective Self-Supervised Framework for Learning Expressive Molecular Global Represen- tations to Drug Discovery. Briefings Bioinform., 2021.
  36. [Li et al., 2022a] H. Li, D. Zhao, et al. KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Pre- diction. In KDD, 2022.
  37. [Li et al., 2022b] S. Li, J. Zhou, et al. GeomGCL: Geometric Graph Contrastive Learning for Molecular Property Prediction. In AAAI, 2022.
  38. [Lin et al., 2022] X. Lin, C. Xu, et al. PanGu Drug Model: Learn a Molecule Like a Human. bioRxiv.org, April 2022.
  39. [Liu et al., 2022] S. Liu, H. Wang, et al. Pre-training Molecular Graph Representation with 3D Geometry. In ICLR, 2022.
  40. [Liu et al., 2023] S. Liu, H. Guo, et al. Molecular Geometry Pre- training with SE(3)-Invariant Denoising Distance Matching. In ICLR, 2023.
  41. [Nguyen et al., 2021] T. Nguyen, H. Le, et al. GraphDTA: Predict- ing Drug-Target Binding Affinity with Graph Neural Networks. Bioinform., 2021.
  42. [Pinheiro et al., 2022] G. A. Pinheiro, J. L. Da Silva, et al. SMI- CLR: Contrastive Learning on Multiple Molecular Representa- tions for Semisupervised and Unsupervised Representation Learn- ing. J. Chem. Inf. Model., 2022.
  43. [Raffel et al., 2020] C. Raffel, N. Shazeer, et al. Exploring the Lim- its of Transfer Learning with a Unified Text-to-Text Transformer.
  44. J. Mach. Learn. Res., 2020.
  45. [Rong et al., 2020] Y. Rong, Y. Bian, et al. Self-Supervised Graph Transformer on Large-Scale Molecular Data. In NeurIPS, 2020.
  46. [Ross et al., 2022] J. Ross, B. Belgodere, et al. Molformer: Large Scale Chemical Language Representations Capture Molecular Structure and Properties. Nat. Mach. Intell., 2022.
  47. [Sanchez-Gonzalez et al., 2020] A. Sanchez-Gonzalez, J. Godwin, et al. Learning to Simulate Complex Physics with Graph Net- works. In ICML, 2020.
  48. [Satorras et al., 2021] V. G. Satorras, E. Hoogeboom, et al. E(n) Equivariant Graph Neural Networks. In ICML, 2021.
  49. [Schütt et al., 2017] K. Schütt, P.-J. Kindermans, et al. SchNet: A Continuous-Filter Convolutional Neural Network for Modeling Quantum Interactions. In NIPS, 2017.
  50. [Schütt et al., 2021] K. Schütt, O. T. Unke, et al. Equivariant mes- sage passing for the prediction of tensorial properties and molec- ular spectra. In ICML, 2021.
  51. [Stärk et al., 2022] H. Stärk, D. Beaini, et al. 3D Infomax Improves GNNs for Molecular Property Prediction. In ICML, 2022.
  52. [Sterling and Irwin, 2015] T. Sterling and J. J. Irwin. ZINC 15- Ligand Discovery for Everyone. J. Chem. Inf. Model., 2015.
  53. [Stumpfe et al., 2019] D. Stumpfe, H. Hu, et al. Evolving Concept of Activity Cliffs. ACS Omega, 2019.
  54. [Sun et al., 2020] F. Sun, J. Hoffmann, et al. InfoGraph: Unsuper- vised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization. In ICLR, 2020.
  55. [Sun et al., 2021] M. Sun, J. Xing, et al. MoCL: Contrastive Learn- ing on Molecular Graphs with Multi-level Domain Knowledge. In KDD, 2021.
  56. [Sun et al., 2022] R. Sun, H. Dai, et al. Does GNN Pretraining Help Molecular Representation? In NeurIPS, 2022.
  57. [Suresh et al., 2021] S. Suresh, P. Li, et al. Adversarial Graph Aug- mentation to Improve Graph Contrastive Learning. In NeurIPS, 2021.
  58. [Vaswani et al., 2017] A. Vaswani, N. Shazeer, et al. Attention is All You Need. In NIPS, 2017.
  59. [Velickovic et al., 2018] P. Velickovic, G. Cucurull, et al. Graph Attention Networks. In ICLR, 2018.
  60. [Velickovic et al., 2019] P. Velickovic, W. Fedus, et al. Deep Graph Infomax. In ICLR, 2019.
  61. [Wang et al., 2017] Y. Wang, S. H. Bryant, et al. Pubchem Bioas- say: 2017 Update. Nucleic Acids Res., 2017.
  62. [Wang et al., 2019] S. Wang, Y. Guo, et al. SMILES-BERT: Large Scale Unsupervised Pre-Training for Molecular Property Predic- tion. In BCB, 2019.
  63. [Wang et al., 2021] Y. Wang, Y. Min, et al. Molecular Graph Con- trastive Learning with Parameterized Explainable Augmentations. In BIBM, 2021.
  64. [Wang et al., 2022a] Y. Wang, R. Magar, et al. Improving Molec- ular Contrastive Learning via Faulty Negative Mitigation and Decomposed Fragment Contrast. J. Chem. Inf. Model., 2022.
  65. [Wang et al., 2022b] Y. Wang, J. Wang, et al. MolCLR: Molecu- lar Contrastive Learning of Representations via Graph Neural Networks. Nat. Mach. Intell., 2022.
  66. [Weininger, 1988] D. Weininger. SMILES, A Chemical Language and Information System. 1. Introduction to Methodology and Encoding Rules. J. Chem. Inf. Comput. Sci., 1988.
  67. [Wu et al., 2018] Z. Wu, B. Ramsundar, et al. MoleculeNet: A Benchmark for Molecular Machine Learning. Chem. Sci., 2018.
  68. [Xia et al., 2022a] J. Xia, L. Wu, et al. ProGCL: Rethinking Hard Negative Mining in Graph Contrastive Learning. In ICML, 2022.
  69. [Xia et al., 2022b] J. Xia, L. Wu, et al. SimGRACE: A Simple Framework for Graph Contrastive Learning without Data Aug- mentation. In WWW, 2022.
  70. [Xia et al., 2023] J. Xia, C. Zhao, et al. Mole-BERT: Rethinking Pre-training Graph Neural Networks for Molecules. In ICLR, 2023.
  71. [Xu et al., 2019] K. Xu, W. Hu, et al. How Powerful are Graph Neural Networks? In ICLR, 2019.
  72. [Xu et al., 2021a] D. Xu, W. Cheng, et al. InfoGCL: Information- Aware Graph Contrastive Learning. In NeurIPS, 2021.
  73. [Xu et al., 2021b] M. Xu, H. Wang, et al. Self-supervised Graph- level Representation Learning with Local and Global Structure. In ICML, 2021.
  74. [Xu et al., 2021c] Z. Xu, Y. Luo, et al. Molecule3D: A Benchmark for Predicting 3D Geometries from Molecular Graphs. arXiv.org, September 2021.
  75. [Yi et al., 2022] J. Yi, C. Wu, et al. MICER: A Pre-Trained Encoder-Decoder Architecture for Molecular Image Caption- ing. Bioinform., 2022.
  76. [Ying et al., 2021] C. Ying, T. Cai, et al. Do Transformers Really Perform Badly for Graph Representation? In NeurIPS, 2021.
  77. [You et al., 2020] Y. You, T. Chen, et al. Graph Contrastive Learn- ing with Augmentations. In NeurIPS, 2020.
  78. [You et al., 2021] Y. You, T. Chen, et al. Graph Contrastive Learn- ing Automated. In ICML, 2021.
  79. [You et al., 2022] Y. You, T. Chen, et al. Bringing Your Own View: Graph Contrastive Learning without Prefabricated Data Augmen- tations. In WSDM, 2022.
  80. [Zaidi et al., 2023] S. Zaidi, M. Schaarschmidt, et al. Pre-training via Denoising for Molecular Property Prediction. In ICLR, 2023.
  81. [Zeng et al., 2022a] X. Zeng, H. Xiang, et al. Accurate Predic- tion of Molecular Properties and Drug Targets Using a Self- Supervised Image Representation Learning Framework. Nat. Mach. Intell., 2022.
  82. [Zeng et al., 2022b] Z. Zeng, Y. Yao, et al. A Deep-Learning Sys- tem Bridging Molecule Structure and Biomedical Text with Com- prehension Comparable to Human Professionals. Nat. Commun., 2022.
  83. [Zhang et al., 2021] Z. Zhang, Q. Liu, et al. Motif-based Graph Self-Supervised Learning for Molecular Property Prediction. In NeurIPS, 2021.
  84. [Zheng et al., 2022] J. Zheng, Y. Wang, G. Wang, J. Xia, Y. Huang, G. Zhao, Y. Zhang, and S. Z. Li. Using context-to-vector with graph retrofitting to improve word embeddings. ACL, 2022.
  85. [Zhou et al., 2023] G. Zhou, Z. Gao, et al. Uni-Mol: A Universal 3D Molecular Representation Learning Framework. In ICLR, 2023.
  86. [Zhu et al., 2021] J. Zhu, Y. Xia, et al. Dual-view Molecule Pre- training. arXiv.org, June 2021.
  87. [Zhu et al., 2022a] J. Zhu, Y. Xia, et al. Unified 2D and 3D Pre- Training of Molecular Representations. In KDD, 2022.
  88. [Zhu et al., 2022b] Y. Zhu, D. Chen, et al. Featurizations Matter: A Multiview Contrastive Learning Approach to Molecular Pre- training. In AI4Science@ICML, 2022.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards