E1: Retrieval-Augmented Protein Encoder Models
Sarthak Jain, Joel Beazer, Jeffrey A. RuffoloiD, Aadyot Bhatnagar, Ali MadaniiD
Abstract
Large language models trained on natural proteins learn powerful representations of protein sequences that are useful for downstream understanding and prediction tasks. Because they are only exposed to individual protein sequences during pretraining without any additional contextual information, conventional protein language models suffer from parameter inefficiencies in learning, baked-in phylogenetic biases, and functional performance issues at larger scales. To address these challenges, we have built Profluent-E1, a family of retrieval-augmented protein language models that explicitly condition on homologous sequences. By integrating retrieved evolutionary context through block-causal multi-sequence attention, E1 captures both general and family-specific constraints without fine-tuning. We train E1 models on four trillion tokens from the Profluent Protein Atlas and achieve state-of-the-art performance across zero-shot fitness and unsupervised contact-map prediction benchmarks - surpassing alternative sequence-only models. Performance scales with model size from 150M to 600M parameters, and E1 can be used flexibly in single-sequence or retrieval-augmented inference mode for fitness prediction, variant ranking, and embeddings for structural tasks. To encourage open science and further development in retrieval-augmented protein language models, we release three models for free research and commercial use at https://github.com/Profluent-AI/E1.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 7
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- RAG-ESM: Improving pretrained protein language models via sequence retrieval
Damiano Sgarbossa, Anne‐Florence Bitbol·PRX LifePeer-reviewed·Jul 30, 2025
- ProFam: Open-Source Protein Family Language Modelling for Fitness Prediction and Design
Jude Wells, Alex Hawkins‐Hooker, … Michael Heinzinger·bioRxiv·Dec 21, 2025
- Protein Language Model Fitness Is a Matter of Preference
Cade Gordon, Amy X. Lu, Pieter Abbeel·bioRxiv·Oct 3, 2024
- xTrimoPGLM: Unified 100B-Scale Pre-trained Transformer for Deciphering the Language of Protein
Bo Chen, Xingyi Cheng, … Le Song·bioRxiv·Jul 5, 2023
- Fine-tuning protein language models boosts predictions across diverse tasks
Robert Schmirler, Michael Heinzinger, Burkhard Rost·Nature CommunicationsPeer-reviewed·Aug 28, 2024
- Prot2Token: A multi-task framework for protein language processing using autoregressive language modeling
Mahdi Pourmirzaei, Farzaneh Esmaili, … Dong He Xu·bioRxiv·May 31, 2024