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.

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

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

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

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

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

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

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

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

MolRoPE-BERT: an Enhanced Molecular Representation with Rotary Position Embedding for Molecular Property Prediction

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

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

Ensemble Model With Bert,Roberta and Xlnet For Molecular property prediction

ASGN: An Active Semi-supervised Graph Neural Network for Molecular Property Prediction

DIG-Mol: A Contrastive Dual-Interaction Graph Neural Network for Molecular Property Prediction

MetaGIN: A Lightweight Framework for Molecular Property Prediction

Contrastive Dual-Interaction Graph Neural Network for Molecular Property Prediction

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

Learn molecular representations from large-scale unlabeled molecules for drug discovery

ResGAT: Residual Graph Attention Networks for molecular property prediction

An effective self-supervised framework for learning expressive molecular global representations to drug discovery

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