mlsanityData to 7 Oct 2026

Transformers for molecular property prediction: Domain adaptation efficiently improves performance

arXiv 2503.03360 ↗First posted 5 Mar 2025cs.LG cs.AI cs.CLcited by 3

Not on the map. Why?

Abstract (arXiv)

Over the past six years, molecular transformer models have become key tools in drug discovery. Most existing models are pre-trained on large, unlabeled datasets such as ZINC or ChEMBL. However, the extent to which large-scale pre-training improves molecular property prediction remains unclear. This study evaluates transformer models for this task while addressing their limitations. We explore how pre-training dataset size and chemically informed objectives impact performance. Our results show that increasing the dataset beyond approximately 400K to 800K molecules from large-scale unlabeled databases does not enhance performance across seven datasets covering five ADME endpoints: lipophilicity, permeability, solubility (two datasets), microsomal stability (two datasets), and plasma protein binding. In contrast, domain adaptation on a small, domain-specific dataset (less than or equal 4K molecules) using multi-task regression of physicochemical properties significantly boosts performance (P-value less than 0.001). A model pre-trained on 400K molecules and adapted with domain-specific data outperforms larger models such as MolFormer and performs comparably to MolBERT. Benchmarks against Random Forest (RF) baselines using descriptors and Morgan fingerprints show that chemically and physically informed features consistently yield better performance across model types. While RF remains a strong baseline, we identify concrete practices to enhance transformer performance. Aligning pre-training and adaptation with chemically meaningful tasks and domain-relevant data presents a promising direction for molecular property prediction. Our models are available on HuggingFace for easy use and adaptation.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 28 of its 73 references that are in the map. First posted on arXiv 5 Mar 2025.

Builds on

28 of 73 references.

References · 28

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
  4. 2020Scaling Laws for Neural Language Modelscited by 6,411
  5. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  6. 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,201
  7. 2016Google's Neural Machine Translation System: Bridging the Gap between Human and Machine Translationcited by 1,471
  8. 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
  9. 2021Revisiting Deep Learning Models for Tabular Datacited by 725
  10. 2020Self-supervised Learning: Generative or Contrastivecited by 693
  11. 2021Tabular Data: Deep Learning is Not All You Needcited by 530
  12. 2019Analyzing Learned Molecular Representations for Property Predictioncited by 373
16 more
  1. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
  2. 2021Explaining Neural Scaling Lawscited by 353
  3. 2020ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Predictioncited by 332
  4. 2018GuacaMol: Benchmarking Models for De Novo Molecular Designcited by 294
  5. 2017Efficient Natural Language Response Suggestion for Smart Replycited by 228
  6. 2019A Study of BFLOAT16 for Deep Learning Trainingcited by 227
  7. 2023A Survey on Self-supervised Learning: Algorithms, Applications, and Future Trendscited by 205
  8. 2021Large-Scale Chemical Language Representations Capture Molecular Structure and Propertiescited by 203
  9. 2021Self-Supervised Learning of Graph Neural Networks: A Unified Reviewcited by 184
  10. 2022ChemBERTa-2: Towards Chemical Foundation Modelscited by 123
  11. 2020Molecule Attention Transformercited by 92
  12. 2020Molecular representation learning with language models and domain-relevant auxiliary taskscited by 91
  13. 2017On Sampling Strategies for Neural Network-based Collaborative Filteringcited by 71
  14. 2024From Words to Molecules: A Survey of Large Language Models in Chemistrycited by 14
  15. 2024Transformers for molecular property prediction: Lessons learned from the past five yearscited by 7
  16. 2024Enhancing Uncertainty Quantification in Drug Discovery with Censored Regression Labelscited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 1; 2026: 2.

3 papers.