Tx-LLM: A Large Language Model for Therapeutics

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chaves, Juan Manuel Zambrano, Wang, Eric, Tu, Tao, Vaishnav, Eeshit Dhaval, Lee, Byron, Mahdavi, S. Sara, Semturs, Christopher, Fleet, David, Natarajan, Vivek, Azizi, Shekoofeh
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911911993409536
author Chaves, Juan Manuel Zambrano
Wang, Eric
Tu, Tao
Vaishnav, Eeshit Dhaval
Lee, Byron
Mahdavi, S. Sara
Semturs, Christopher
Fleet, David
Natarajan, Vivek
Azizi, Shekoofeh
author_facet Chaves, Juan Manuel Zambrano
Wang, Eric
Tu, Tao
Vaishnav, Eeshit Dhaval
Lee, Byron
Mahdavi, S. Sara
Semturs, Christopher
Fleet, David
Natarajan, Vivek
Azizi, Shekoofeh
contents Developing therapeutics is a lengthy and expensive process that requires the satisfaction of many different criteria, and AI models capable of expediting the process would be invaluable. However, the majority of current AI approaches address only a narrowly defined set of tasks, often circumscribed within a particular domain. To bridge this gap, we introduce Tx-LLM, a generalist large language model (LLM) fine-tuned from PaLM-2 which encodes knowledge about diverse therapeutic modalities. Tx-LLM is trained using a collection of 709 datasets that target 66 tasks spanning various stages of the drug discovery pipeline. Using a single set of weights, Tx-LLM simultaneously processes a wide variety of chemical or biological entities(small molecules, proteins, nucleic acids, cell lines, diseases) interleaved with free-text, allowing it to predict a broad range of associated properties, achieving competitive with state-of-the-art (SOTA) performance on 43 out of 66 tasks and exceeding SOTA on 22. Among these, Tx-LLM is particularly powerful and exceeds best-in-class performance on average for tasks combining molecular SMILES representations with text such as cell line names or disease names, likely due to context learned during pretraining. We observe evidence of positive transfer between tasks with diverse drug types (e.g.,tasks involving small molecules and tasks involving proteins), and we study the impact of model size, domain finetuning, and prompting strategies on performance. We believe Tx-LLM represents an important step towards LLMs encoding biochemical knowledge and could have a future role as an end-to-end tool across the drug discovery development pipeline.
format Preprint
id arxiv_https___arxiv_org_abs_2406_06316
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Tx-LLM: A Large Language Model for Therapeutics
Chaves, Juan Manuel Zambrano
Wang, Eric
Tu, Tao
Vaishnav, Eeshit Dhaval
Lee, Byron
Mahdavi, S. Sara
Semturs, Christopher
Fleet, David
Natarajan, Vivek
Azizi, Shekoofeh
Computation and Language
Artificial Intelligence
Computational Engineering, Finance, and Science
Machine Learning
Developing therapeutics is a lengthy and expensive process that requires the satisfaction of many different criteria, and AI models capable of expediting the process would be invaluable. However, the majority of current AI approaches address only a narrowly defined set of tasks, often circumscribed within a particular domain. To bridge this gap, we introduce Tx-LLM, a generalist large language model (LLM) fine-tuned from PaLM-2 which encodes knowledge about diverse therapeutic modalities. Tx-LLM is trained using a collection of 709 datasets that target 66 tasks spanning various stages of the drug discovery pipeline. Using a single set of weights, Tx-LLM simultaneously processes a wide variety of chemical or biological entities(small molecules, proteins, nucleic acids, cell lines, diseases) interleaved with free-text, allowing it to predict a broad range of associated properties, achieving competitive with state-of-the-art (SOTA) performance on 43 out of 66 tasks and exceeding SOTA on 22. Among these, Tx-LLM is particularly powerful and exceeds best-in-class performance on average for tasks combining molecular SMILES representations with text such as cell line names or disease names, likely due to context learned during pretraining. We observe evidence of positive transfer between tasks with diverse drug types (e.g.,tasks involving small molecules and tasks involving proteins), and we study the impact of model size, domain finetuning, and prompting strategies on performance. We believe Tx-LLM represents an important step towards LLMs encoding biochemical knowledge and could have a future role as an end-to-end tool across the drug discovery development pipeline.
title Tx-LLM: A Large Language Model for Therapeutics
topic Computation and Language
Artificial Intelligence
Computational Engineering, Finance, and Science
Machine Learning
url https://arxiv.org/abs/2406.06316