2022/06/27 by Erik Nijkamp, Jeffrey A. Ruffolo, Nijkamp, Erik +7 · 24 citations
Biochemistry, Genetics and Molecular Biology · #FOS: Biological sciences #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning in Bioinformatics #Quantitative Methods (q-bio.QM) #RNA and protein synthesis mechanisms #vaccines and immunoinformatics approaches
paper · pdf · doi:10.48550/arxiv.2206.13517
openalex publication_date 2022/06/27 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28
Attention-based models trained on protein sequences have demonstrated incredible success at classification and generation tasks relevant for artificial intelligence-driven protein design. However, we lack a sufficient understanding of how very large-scale models and data play a role in effective protein model development. We introduce a suite of protein language models, named ProGen2, that are scaled up to 6.4B parameters and trained on different sequence datasets drawn from over a billion proteins from genomic, metagenomic, and immune repertoire databases. ProGen2 models show state-of-the-art performance in capturing the distribution of observed evolutionary sequences, generating novel viable sequences, and predicting protein fitness without additional finetuning. As large model sizes and raw numbers of protein sequences continue to become more widely accessible, our results suggest that a growing emphasis needs to be placed on the data distribution provided to a protein sequence model. We release the ProGen2 models and code at https://github.com/salesforce/progen.