bioRxiv · 10.1101/2023.08.04.551626
Efficient and accurate sequence generation with small-scale protein language models
Abstract
Large Language Models (LLMs) have demonstrated exceptional capabilities in understanding contextual relationships, outperforming traditional methodologies in downstream tasks such as text generation and sentence classification. This success has been mirrored in the realm of protein language models (pLMs), where proteins are encoded as text via their amino acid sequences. However, the training of pLMs, which involves tens to hundreds of millions of sequences and hundreds of millions to billions of parameters, poses a significant computational challenge. In this study, we introduce a Small-Scale Protein Language Model (SS-pLM), a more accessible approach that requires training on merely millions of representative sequences, reducing the number of trainable parameters to 14.8M. This model significantly reduces the computational load, thereby democratizing the use of foundational models in protein studies. We demonstrate that the performance of our model, when fine-tuned to a specific set of sequences for generation, is comparable to that of larger, more computationally demanding pLM.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Serrano, Y., Roda, S., Guallar, V., Molina, A.. 2023-08-06. Efficient and accurate sequence generation with small-scale protein language models. https://doi.org/10.1101/2023.08.04.551626
Cite the original work for its findings. Save a collection to share your selection of sources.