TwinBooster: Synergising Large Language Models with Barlow Twins and Gradient Boosting for Enhanced Molecular Property Prediction

Maximilian G. Schuh, Davide Boldini, Stephan A. Sieber
1/9/2024
q-bio.BMcs.AIcs.CL

No official code found. Best implementation: maxischuh/TwinBooster (6 stars, MIT, last commit 2026-02-02).

Abstract

The success of drug discovery and development relies on the precise prediction of molecular activities and properties. While in silico molecular property prediction has shown remarkable potential, its use has been limited so far to assays for which large amounts of data are available. In this study, we use a fine-tuned large language model to integrate biological assays based on their textual information, coupled with Barlow Twins, a Siamese neural network using a novel self-supervised learning approach. This architecture uses both assay information and molecular fingerprints to extract the true molecular information. TwinBooster enables the prediction of properties of unseen bioassays and molecules by providing state-of-the-art zero-shot learning tasks. Remarkably, our artificial intelligence pipeline shows excellent performance on the FS-Mol benchmark. This breakthrough demonstrates the application of deep learning to critical property prediction tasks where data is typically scarce. By accelerating the early identification of active molecules in drug discovery and development, this method has the potential to help streamline the identification of novel therapeutics.

View on arXivView PDF

Code Implementations(1)

MIT

Package for TwinBooster. Enables fast and powerful zero-shot molecular property prediction.

61Python, Jupyter NotebookJan 9, 20248 months agoMIT
cheminformaticschemistrylarge-language-modelsmachine-learningmolecular-property-prediction+1 more

Similar Papers

Cite this paper

@article{schuh2024twinbooster,
  title  = {TwinBooster: Synergising Large Language Models with Barlow Twins and Gradient Boosting for Enhanced Molecular Property Prediction},
  author = {Maximilian G. Schuh and Davide Boldini and Stephan A. Sieber},
  year   = {2024},
  eprint = {2401.04478},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.04478v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2401.04478.svg)](https://codeofpaper.com/paper/2401.04478)

Discussion