Abstract:Accurate prediction of pharmacological properties of small molecules is becoming increasingly important in drug discovery. Traditional feature-engineering approaches heavily rely on handcrafted descriptors and/or fingerprints, which need extensive human expert knowledge. With the rapid progress of artificial intelligence technology, data-driven deep learning methods have shown unparalleled advantages over feature-engineering-based methods. However, existing deep learning methods usually suffer from the scarcity of labeled data and the inability to share information between different tasks when applied to predicting molecular properties, thus resulting in poor generalization capability. Here, we proposed a novel multitask learning BERT (Bidirectional Encoder Representations from Transformer) framework, named MTL-BERT, which leverages large-scale pre-training, multitask learning, and SMILES (simplified molecular input line entry specification) enumeration to alleviate the data scarcity problem. MTL-BERT first exploits a large amount of unlabeled data through self-supervised pretraining to mine the rich contextual information in SMILES strings and then fine-tunes the pretrained model for multiple downstream tasks simultaneously by leveraging their shared information. Meanwhile, SMILES enumeration is used as a data enhancement strategy during the pretraining, fine-tuning, and test phases to substantially increase data diversity and help to learn the key relevant patterns from complex SMILES strings. The experimental results showed that the pretrained MTL-BERT model with few additional fine-tuning can achieve much better performance than the state-of-the-art methods on most of the 60 practical molecular datasets. Additionally, the MTL-BERT model leverages attention mechanisms to focus on SMILES character features essential to target properties for model interpretability.

CIMG-BERT: Pretraining Bidirectional Transformers with Chemistry Knowledge for Molecular Property Prediction

MG-BERT: leveraging unsupervised atomic representation learning for molecular property prediction

SMG-BERT: integrating stereoscopic information and chemical representation for molecular property prediction

Knowledge-based BERT: a Method to Extract Molecular Features Like Computational Chemists

Knowledge-based BERT: a method to extract molecular features such as computational chemists

ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Prediction

Mole-BERT: Rethinking Pre-training Graph Neural Networks for Molecules

KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Prediction

FG-BERT: a Generalized and Self-Supervised Functional Group-Based Molecular Representation Learning Framework for Properties Prediction.

Pushing the Boundaries of Molecular Property Prediction for Drug Discovery with Multitask Learning BERT Enhanced by SMILES Enumeration

ChemBERTa-2: Towards Chemical Foundation Models

A Chemical Domain Knowledge-Aware Framework for Multi-view Molecular Property Prediction

MultiModal-Learning for Predicting Molecular Properties: A Framework Based on Image and Graph Structures

DMPNN-Bert: a deep learning architecture for molecular property prediction.

CasANGCL: pre-training and fine-tuning model based on cascaded attention network and graph contrastive learning for molecular property prediction

A knowledge-guided pre-training framework for improving molecular representation learning

Contrastive Dual-Interaction Graph Neural Network for Molecular Property Prediction

Algebraic graph-assisted bidirectional transformers for molecular property prediction

Chemical structure-aware molecular image representation learning

Geometry-based BERT: an experimentally validated deep learning model for molecular property prediction in drug discovery

Triple Generative Self-Supervised Learning Method for Molecular Property Prediction