In-Context Learning can distort the relationship between sequence likelihoods and biological fitness

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kantroo, Pranav, Wagner, Günter P., Machta, Benjamin B.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909591315415040
author Kantroo, Pranav
Wagner, Günter P.
Machta, Benjamin B.
author_facet Kantroo, Pranav
Wagner, Günter P.
Machta, Benjamin B.
contents Language models have emerged as powerful predictors of the viability of biological sequences. During training these models learn the rules of the grammar obeyed by sequences of amino acids or nucleotides. Once trained, these models can take a sequence as input and produce a likelihood score as an output; a higher likelihood implies adherence to the learned grammar and correlates with experimental fitness measurements. Here we show that in-context learning can distort the relationship between fitness and likelihood scores of sequences. This phenomenon most prominently manifests as anomalously high likelihood scores for sequences that contain repeated motifs. We use protein language models with different architectures trained on the masked language modeling objective for our experiments, and find transformer-based models to be particularly vulnerable to this effect. This behavior is mediated by a look-up operation where the model seeks the identity of the masked position by using the other copy of the repeated motif as a reference. This retrieval behavior can override the model's learned priors. This phenomenon persists for imperfectly repeated sequences, and extends to other kinds of biologically relevant features such as reversed complement motifs in RNA sequences that fold into hairpin structures.
format Preprint
id arxiv_https___arxiv_org_abs_2504_17068
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle In-Context Learning can distort the relationship between sequence likelihoods and biological fitness
Kantroo, Pranav
Wagner, Günter P.
Machta, Benjamin B.
Machine Learning
Biomolecules
Language models have emerged as powerful predictors of the viability of biological sequences. During training these models learn the rules of the grammar obeyed by sequences of amino acids or nucleotides. Once trained, these models can take a sequence as input and produce a likelihood score as an output; a higher likelihood implies adherence to the learned grammar and correlates with experimental fitness measurements. Here we show that in-context learning can distort the relationship between fitness and likelihood scores of sequences. This phenomenon most prominently manifests as anomalously high likelihood scores for sequences that contain repeated motifs. We use protein language models with different architectures trained on the masked language modeling objective for our experiments, and find transformer-based models to be particularly vulnerable to this effect. This behavior is mediated by a look-up operation where the model seeks the identity of the masked position by using the other copy of the repeated motif as a reference. This retrieval behavior can override the model's learned priors. This phenomenon persists for imperfectly repeated sequences, and extends to other kinds of biologically relevant features such as reversed complement motifs in RNA sequences that fold into hairpin structures.
title In-Context Learning can distort the relationship between sequence likelihoods and biological fitness
topic Machine Learning
Biomolecules
url https://arxiv.org/abs/2504.17068