Chemical Language Meets Geometric Graphs: A Multimodal Fusion Approach for Molecular Properties

Collin Francel, Massimiliano Lupo Pasini, Zachary R Fox

Submission Track: Short Paper

Submission Category: AI-Guided Design

Keywords: Language Models, Chemical Language Models, Geometric Deep Learning, Multimodal Learning, Graph Neural Networks

TL;DR: We evaluate fusing geometric graph neural networks with pretrained chemical language models for molecular property prediction.

Abstract: Rapid and accurate prediction of the physicochemical properties of molecules given their structures remains a key challenge in cheminformatics. Machine learning approaches offer high-throughput options, but the optimality of inductive biases and data representations are up for debate. For example, BERT-based masked language models (MLMs) can be trained in a self-supervised way on hundreds of millions to billions of readily available SMILES strings. Another option is graph neural networks (GNNs), which can operate directly on molecular structures. Yet, generating accurate molecular geometry is computationally expensive, leading to a relative scarcity in data compared to SMILES strings. It is attractive to combine these two paradigms by pre-training an LM on a large corpus of SMILES strings and embedding these representation into a geometric graph neural network. Despite the promise of such an approach, and contrary to previous studies, we find mixed results with the combination of the LMs and GNNs on several molecule datasets. In particular, we found evidence for improvement on the FreeSolv and QM7 benchmarks, but degraded performance on the ESOL, LIPO and QM9 datasets compared to a GNN baseline.

Submission Number: 51