Proceedings

M^2LLM: Multi-view Molecular Representation Learning with Large Language Models

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 7437-7445.

https://doi.org/10.24963/ijcai.2025/827

Abstract

Accurate molecular property prediction is a critical challenge with wide-ranging applications in chemistry, materials science, and drug discovery. Molecular representation methods, including fingerprints and graph neural networks (GNNs), achieve state-of-the-art results by effectively deriving features from molecular structures. However, these methods often overlook decades of accumulated semantic and contextual knowledge. Recent advancements in large language models (LLMs) demonstrate remarkable reasoning abilities and prior knowledge across scientific domains, leading us to hypothesize that LLMs can generate rich molecular representations when guided to reason in multiple perspectives. To address these gaps, we propose M^2LLM, a multi-view framework that integrates three perspectives: the molecular structure view, the molecular task view, and the molecular rules view. These views are fused dynamically to adapt to task requirements, and experiments demonstrate that M^2LLM achieves state-of-the-art performance on multiple benchmarks across classification and regression tasks. Moreover, we demonstrate that representation derived from LLM achieves exceptional performance by leveraging two core functionalities: the generation of molecular embeddings through their encoding capabilities and the curation of molecular features through advanced reasoning processes.

  1. [Breiman, 2001] Leo Breiman. Random forests. Machine learning, 45:5-32, 2001.
  2. [Bu et al., 2024] Weixin Bu, Xiaofeng Cao, Yizhen Zheng, and Shirui Pan. Improving augmentation consistency for graph contrastive learning. Pattern Recognition, 148:110182, 2024.
  3. [Delaney, 2004] John S Delaney. Esol: estimating aqueous solubility directly from molecular structure. Journal of chemical information and computer sciences, 44(3):1000- 1005, 2004.
  4. [Drews, 2000] Jurgen Drews. Drug discovery: a historical perspective. science, 287(5460):1960-1964, 2000.
  5. [Du et al., 2024] Wenjie Du, Shuai Zhang, Jun Xia Di Wu, Ziyuan Zhao, Junfeng Fang, and Yang Wang. Mmgnn: A molecular merged graph neural network for explainable solvation free energy prediction. In Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence, pages 5808-5816, 2024. IJCAI
  6. [Dubey et al., 2024] Abhimanyu Dubey, Abhinav Jauhri, et al. The llama 3 herd of models. arXiv preprint arXiv:2407.21783, 2024.
  7. [Fabian et al., 2020] Benedek Fabian, Thomas Edlich, et al. Molecular representation learning with language mod- els and domain-relevant auxiliary tasks. arXiv preprint arXiv:2011.13230, 2020.
  8. [Gayvert et al., 2016] Kaitlyn M Gayvert, Neel S Madhukar, and Olivier Elemento. A data-driven approach to predict- ing successes and failures of clinical trials. Cell chemical biology, 23(10):1294-1301, 2016.
  9. [Harding et al., 2024] Simon D Harding, Jane F Armstrong, et al. The iuphar/bps guide to pharmacology in 2024. Nu- cleic Acids Research, 52(D1):D1438-D1449, 2024.
  10. [Hu et al., 2019] Weihua Hu, Bowen Liu, et al. Strategies for pre-training graph neural networks. arXiv preprint arXiv:1905.12265, 2019.
  11. [Jeon and Kim, 2019] Woosung Jeon and Dongsup Kim. Fp2vec: a new molecular featurizer for learning molecular properties. Bioinformatics, 35(23):4979-4985, 2019.
  12. [Kenton and Toutanova, 2019] Jacob Devlin Ming- Wei Chang Kenton and Lee Kristina Toutanova. Bert: Pre-training of deep bidirectional transformers for lan- guage understanding. In Proceedings of naacL-HLT, volume 1. Minneapolis, Minnesota, 2019.
  13. [Koh et al., 2024] Huan Yee Koh, Anh TN Nguyen, Shirui Pan, Lauren T May, and Geoffrey I Webb. Physicochemi- cal graph neural network for learning protein-ligand inter- action fingerprints from sequence data. Nature Machine Intelligence, pages 1-15, 2024.
  14. [Kojima et al., 2022] Takeshi Kojima, Shixiang Shane Gu, et al. Large language models are zero-shot reason- ers. Advances in neural information processing systems, 35:22199-22213, 2022.
  15. [Kuhn et al., 2016] Michael Kuhn, Ivica Letunic, Lars Juhl Jensen, and Peer Bork. The sider database of drugs and side effects. Nucleic acids research, 44(D1):D1075- D1079, 2016.
  16. [Landrum and Riniker, 2024] Gregory A Landrum and Sere- ina Riniker. Combining ic50 or k i values from different sources is a source of significant noise. Journal of Chemi- cal Information and Modeling, 64(5):1560-1567, 2024.
  17. [Liu et al., 2022] Shengchao Liu, Hanchen Wang, et al. Pre- training molecular graph representation with 3d geometry. In International Conference on Learning Representations, 2022.
  18. [Luo et al., 2024] Yizhen Luo, Kai Yang, et al. Learning multi-view molecular representations with structured and unstructured knowledge. In Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pages 2082-2093, 2024.
  19. [Martins et al., 2012] Ines Filipa Martins, Ana L Teixeira, et al. A bayesian approach to in silico blood-brain bar- rier penetration modeling. Journal of chemical informa- tion and modeling, 52(6):1686-1697, 2012.
  20. [Medsker et al., 2001] Larry R Medsker, Lakhmi Jain, et al. Recurrent neural networks. Design and Applications, 5(64-67):2, 2001.
  21. [Mirza et al., 2024] Adrian Mirza, Nawaf Alampara, et al. Are large language models superhuman chemists? arXiv preprint arXiv:2404.01475, 2024.
  22. [Mobley and Guthrie, 2014] David L Mobley and J Peter Guthrie. Freesolv: a database of experimental and cal- culated hydration free energies, with input files. Journal of computer-aided molecular design, 28:711-720, 2014.
  23. [OpenAI et al., 2023] Josh OpenAI, Achiam, Steven Adler, et al. Gpt-4 technical report. arXiv preprint arXiv:2303.08774, 2023.
  24. [OpenAI, 2024] OpenAI. New embedding models and api updates, 2024.
  25. [Rogers and Hahn, 2010] David Rogers and Mathew Hahn. Extended-connectivity fingerprints. Journal of chemical information and modeling, 50(5):742-754, 2010.
  26. [Rollins et al., 2024] Zachary A Rollins, Alan C Cheng, and Essam Metwally. Molprop: Molecular property prediction with multimodal language and graph fusion. Journal of Cheminformatics, 16(1):56, 2024.
  27. [Rong et al., 2020] Yu Rong, Yatao Bian, et al. Self- supervised graph transformer on large-scale molecular data. Advances in neural information processing systems, 33:12559-12571, 2020.
  28. [Ross et al., 2022] Jerret Ross, Brian Belgodere, et al. Large-scale chemical language representations capture molecular structure and properties. Nature Machine In- telligence, 4(12):1256-1264, 2022.
  29. [Sadeghi et al., 2024] Shaghayegh Sadeghi, Alan Bui, Ali Forooghi, Jianguo Lu, and Alioune Ngom. Comparative analysis of llama and chatgpt embeddings for molecule embedding. arXiv preprint arXiv:2402.00024, 2024.
  30. [Shirasuna et al., 2024] Victor Yukio Shirasuna, Eduardo Soares, et al. A multi-view mixture-of-experts based on language and graphs for molecular properties prediction. In ICML 2024 AI for Science Workshop, 2024.
  31. [Stärk et al., 2022] Hannes Stärk, Dominique Beaini, Gabriele Corso, Prudencio Tossou, Christian Dallago, Stephan Günnemann, and Pietro Liò. 3d infomax improves gnns for molecular property prediction. In International Conference on Machine Learning, pages 20479-20502.
  32. PMLR, 2022.
  33. [Subramanian et al., 2016] Govindan Subramanian, Bharath Ramsundar, Vijay Pande, and Rajiah Aldrin Denny. Com- putational modeling of β-secretase 1 (bace-1) inhibitors using ligand based approaches. Journal of chemical infor- mation and modeling, 56(10):1936-1949, 2016.
  34. [Taylor et al., 2022] Ross Taylor, Marcin Kardas, Guillem Cucurull, Thomas Scialom, Anthony Hartshorn, Elvis Sar- avia, Andrew Poulton, Viktor Kerkez, and Robert Stojnic. Galactica: A large language model for science. arXiv preprint arXiv:2211.09085, 2022.
  35. [Wang et al., 2019] Sheng Wang, Yuzhi Guo, et al. Smiles- bert: large scale unsupervised pre-training for molecular property prediction. In Proceedings of the 10th ACM in- ternational conference on bioinformatics, computational biology and health informatics, pages 429-436, 2019.
  36. [Wang et al., 2022] Yuyang Wang, Jianren Wang, Zhonglin Cao, and Amir Barati Farimani. Molecular contrastive learning of representations via graph neural networks. Na- ture Machine Intelligence, 4(3):279-287, 2022.
  37. [Wang et al., 2024a] Luzhi Wang, Dongxiao He, He Zhang, Yixin Liu, Wenjie Wang, Shirui Pan, Di Jin, and Tat-Seng Chua. Goodat: towards test-time graph out-of-distribution detection. In Proceedings of the AAAI Conference on Arti- ficial Intelligence, volume 38, pages 15537-15545, 2024.
  38. [Wang et al., 2024b] Luzhi Wang, Yizhen Zheng, Di Jin, Fuyi Li, Yongliang Qiao, and Shirui Pan. Contrastive graph similarity networks. ACM Transactions on the Web, 18(2):1-20, 2024.
  39. [Weininger, 1988] David Weininger. Smiles, a chemical lan- guage and information system. 1. introduction to method- ology and encoding rules. Journal of chemical information and computer sciences, 28(1):31-36, 1988.
  40. [Wu et al., 2018] Zhenqin Wu, Bharath Ramsundar, Evan N Feinberg, Joseph Gomes, Caleb Geniesse, Aneesh S Pappu, Karl Leswing, and Vijay Pande. Moleculenet: a benchmark for molecular machine learning. Chemical sci- ence, 9(2):513-530, 2018.
  41. [Wu et al., 2024] Man Wu, Xin Zheng, Qin Zhang, Xiao Shen, Xiong Luo, Xingquan Zhu, and Shirui Pan. Graph learning under distribution shifts: A comprehensive sur- vey on domain adaptation, out-of-distribution, and contin- ual learning. arXiv preprint arXiv:2402.16374, 2024.
  42. [Xia et al., 2022] Jun Xia, Chengshuai Zhao, Bozhen Hu, Zhangyang Gao, Cheng Tan, Yue Liu, Siyuan Li, and Stan Z Li. Mole-bert: Rethinking pre-training graph neu- ral networks for molecules. In The Eleventh International Conference on Learning Representations, 2022.
  43. [Yang et al., 2019] Kevin Yang, Kyle Swanson, Wengong Jin, et al. Analyzing learned molecular representations for property prediction. Journal of chemical information and modeling, 59(8):3370-3388, 2019.
  44. [You et al., 2020] Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, and Yang Shen. Graph con- trastive learning with augmentations. Advances in neural information processing systems, 33:5812-5823, 2020.
  45. [Yu et al., 2024] Jiajun Yu, Zhihao Wu, Jinyu Cai, Adele Lu Jia, and Jicong Fan. Kernel readout for graph neural networks. In Proceedings of the Thirty-Third Interna- tional Joint Conference on Artificial Intelligence, IJCAI- 24, pages 2505-2514, 2024. IJCAI
  46. [Yu et al., 2025] Jiajun Yu, Yizhen Zheng, Huan Yee Koh, Shirui Pan, Tianyue Wang, and Haishuai Wang. Collab- orative expert llms guided multi-objective molecular opti- mization. arXiv preprint arXiv:2503.03503, 2025.
  47. [Zhang et al., 2019] He Zhang, Hanlin Mo, You Hao, Qi Li, Shirui Li, and Hua Li. Fast and efficient calculations of structural invariants of chirality. Pattern Recognit. Lett., 128:270-277, 2019.
  48. [Zhang et al., 2025] He Zhang, Bang Wu, Xiangwen Yang, Xingliang Yuan, Xiaoning Liu, and Xun Yi. Dynamic graph unlearning: A general and efficient post-processing method via gradient transformation. In Proceedings of the ACM on Web Conference 2025, pages 931-944, 2025.
  49. [Zheng et al., 2024a] Xin Zheng, Dongjin Song, Qingsong Wen, Bo Du, and Shirui Pan. Online gnn evaluation under test-time graph distribution shifts. In The Twelfth Interna- tional Conference on Learning Representations, 2024.
  50. [Zheng et al., 2024b] Yan Zheng, Song Wu, Junyu Lin, Yazhou Ren, Jing He, Xiaorong Pu, and Lifang He. Cross- view contrastive fusion for enhanced molecular property prediction. Proccedings of the Thirty-Third International Joint Conference on Artificial Intelligence, 2024.
  51. [Zheng et al., 2024c] Yizhen Zheng, Huan Yee Koh, Mad- die Yang, Li Li, Lauren T May, Geoffrey I Webb, Shirui Pan, and George Church. Large language models in drug discovery and development: From disease mechanisms to clinical trials. arXiv preprint arXiv:2409.04481, 2024.
  52. [Zheng et al., 2025] Yizhen Zheng, Huan Yee Koh, Jiaxin Ju, Anh TN Nguyen, Lauren T May, Geoffrey I Webb, and Shirui Pan. Large language models for scientific dis- covery in molecular property prediction. Nature Machine Intelligence, pages 1-11, 2025.
  53. [Zhou et al., 2023] Gengmo Zhou, Zhifeng Gao, et al. Uni- mol: A universal 3d molecular representation learning framework. In The Eleventh International Conference on Learning Representations, 2023.

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

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