Bio-xLSTM: Generative modeling, representation and in-context learning of biological and chemical sequences

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Schmidinger, Niklas, Schneckenreiter, Lisa, Seidl, Philipp, Schimunek, Johannes, Hoedt, Pieter-Jan, Brandstetter, Johannes, Mayr, Andreas, Luukkonen, Sohvi, Hochreiter, Sepp, Klambauer, Günter
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913573124440064
author Schmidinger, Niklas
Schneckenreiter, Lisa
Seidl, Philipp
Schimunek, Johannes
Hoedt, Pieter-Jan
Brandstetter, Johannes
Mayr, Andreas
Luukkonen, Sohvi
Hochreiter, Sepp
Klambauer, Günter
author_facet Schmidinger, Niklas
Schneckenreiter, Lisa
Seidl, Philipp
Schimunek, Johannes
Hoedt, Pieter-Jan
Brandstetter, Johannes
Mayr, Andreas
Luukkonen, Sohvi
Hochreiter, Sepp
Klambauer, Günter
contents Language models for biological and chemical sequences enable crucial applications such as drug discovery, protein engineering, and precision medicine. Currently, these language models are predominantly based on Transformer architectures. While Transformers have yielded impressive results, their quadratic runtime dependency on the sequence length complicates their use for long genomic sequences and in-context learning on proteins and chemical sequences. Recently, the recurrent xLSTM architecture has been shown to perform favorably compared to Transformers and modern state-space model (SSM) architectures in the natural language domain. Similar to SSMs, xLSTMs have a linear runtime dependency on the sequence length and allow for constant-memory decoding at inference time, which makes them prime candidates for modeling long-range dependencies in biological and chemical sequences. In this work, we tailor xLSTM towards these domains and propose a suite of architectural variants called Bio-xLSTM. Extensive experiments in three large domains, genomics, proteins, and chemistry, were performed to assess xLSTM's ability to model biological and chemical sequences. The results show that models based on Bio-xLSTM a) can serve as proficient generative models for DNA, protein, and chemical sequences, b) learn rich representations for those modalities, and c) can perform in-context learning for proteins and small molecules.
format Preprint
id arxiv_https___arxiv_org_abs_2411_04165
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Bio-xLSTM: Generative modeling, representation and in-context learning of biological and chemical sequences
Schmidinger, Niklas
Schneckenreiter, Lisa
Seidl, Philipp
Schimunek, Johannes
Hoedt, Pieter-Jan
Brandstetter, Johannes
Mayr, Andreas
Luukkonen, Sohvi
Hochreiter, Sepp
Klambauer, Günter
Biomolecules
Artificial Intelligence
Machine Learning
Language models for biological and chemical sequences enable crucial applications such as drug discovery, protein engineering, and precision medicine. Currently, these language models are predominantly based on Transformer architectures. While Transformers have yielded impressive results, their quadratic runtime dependency on the sequence length complicates their use for long genomic sequences and in-context learning on proteins and chemical sequences. Recently, the recurrent xLSTM architecture has been shown to perform favorably compared to Transformers and modern state-space model (SSM) architectures in the natural language domain. Similar to SSMs, xLSTMs have a linear runtime dependency on the sequence length and allow for constant-memory decoding at inference time, which makes them prime candidates for modeling long-range dependencies in biological and chemical sequences. In this work, we tailor xLSTM towards these domains and propose a suite of architectural variants called Bio-xLSTM. Extensive experiments in three large domains, genomics, proteins, and chemistry, were performed to assess xLSTM's ability to model biological and chemical sequences. The results show that models based on Bio-xLSTM a) can serve as proficient generative models for DNA, protein, and chemical sequences, b) learn rich representations for those modalities, and c) can perform in-context learning for proteins and small molecules.
title Bio-xLSTM: Generative modeling, representation and in-context learning of biological and chemical sequences
topic Biomolecules
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2411.04165