PreprintCited by 356Machine Learning in BioinformaticsProtein Structure and DynamicsGenomics and Phylogenetic Studies

MSA Transformer

Open access
Rao, Roshan, Liu, Jason, Verkuil, Robert, Meier, Joshua, Canny, John F., Abbeel, Pieter, Sercu, Tom, Rives, Alexander·2021·bioRxiv (Cold Spring Harbor Laboratory)

Abstract

Abstract Unsupervised protein language models trained across millions of diverse sequences learn structure and function of proteins. Protein language models studied to date have been trained to perform inference from individual sequences. The longstanding approach in computational biology has been to make inferences from a family of evo lutionarily related sequences by fitting a model to each family independently. In this work we combine the two paradigms. We introduce a protein language model which takes as input a set of sequences in the form of a multiple sequence alignment. The model interleaves row and column attention across the input sequences and is trained with a variant of the masked language modeling objective across many protein families. The performance of the model surpasses current state-of-the-art unsupervised structure learning methods by a wide margin, with far greater parameter efficiency than prior state-of-the-art protein language models.

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Integrity signals

Registry

Crossref · 10.1101/2021.02.12.430858

Cites

52 of 52 references verified0 citation errors · 0 not checkable

Statistics

Claims the paper makes

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • The MSA clusters contained evolutionary couplings for both states, and the score based on contact maps correlated with the r.m.s.d. in AF2.

    “We used the same set of clusters to make predictions using the unsupervised deep learning model MSA Transformer36 and found that these clusters contained evolutionary couplings for both states, and the score based on contact maps correlated with the root mean squared deviation (r.m.s.d.) in AF2 (Methods and Extended Data Fig. 3).”
  • MSA Transformer is an unsupervised learning method; its contact predictions purely reflect evolutionary couplings learned in sequences, rather than being supervised on structure as is the case AF2.

    “MSA transformer is an unsupervised learning method, which signifies that its contact predictions purely reflect evolutionary couplings learned in sequences, rather than being supervised on structure as is the case AF2.”
  • Attention-based networks are extremely powerful for modeling complicated interactions between entities, making them suitable for modeling proteins whose functions are largely determined by interactions between amino acids.

    “Among many successful neural network architectures, attention-based networks are extremely powerful for modeling complicated interactions between entities, making them suitable for modeling proteins whose functions are largely determined by interactions between amino acids (34–37).”
  • The MSA Transformer derives embeddings via a tied row attention framework that exploits the equivalence of aligned positions in the MSA.

    “In the MSA Transformer (16), the neural network model exploits the equivalence of aligned positions in the MSA in order to derive embeddings via a tied row attention framework.”

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

The self-check needs the paper’s full text.

Cites

Flagged first, then by citations

Research works printed but not linked: 7

  1. Cuff, J. A. and Barton, G. J. Evaluation and improve- ment of multiple sequence methods for protein sec- ondary structure prediction. Proteins: Structure, Func- tion and Genetics, 34(4):508-519, 3 1999. ISSN 08873585. doi: 10.1002/(SICI)1097-0134(19990301) 34:4 508::AID-PROT10 3.0.CO;2-4. URL https:// pubmed.ncbi.nlm.nih.gov/10081963/. · DOI
  2. Ho, J., Kalchbrenner, N., Weissenborn, D., and Salimans, T. Axial Attention in Multidimensional Transformers. arXiv, 12 2019. URL http://arxiv.org/abs/ 1912.12180.
  3. Jumper, J., Evans, R., Pritzel, A., Green, T., Figurnov, M., Tunyasuvunakool, K., Ronneberger, O., Bates, R., Žídek, A., Bridgland, A., Meyer, C., Kohl, S. A. A., Potapenko, A., Ballard, A. J., Cowie, A., Romera-Paredes, B., Nikolov, S., Jain, R., Adler, J., Back, T., Petersen, S., Reiman, D., Steinegger, M., Pacholska, M., Silver, D., Vinyals, O., Senior, A. W., Kavukcuoglu, K., Kohli, P., and Hassabis, D. High Accuracy Protein Structure Prediction Using Deep Learning. In Fourteenth Critical Assessment of Techniques for Protein Structure Prediction (Abstract Book), pp. 22. 2020.
  4. Riesselman, A., Shin, J.-E., Kollasch, A., McMahon, C., Simon, E., Sander, C., Manglik, A., Kruse, A., and Marks, D. Accelerating Protein Design Using Autoregressive Generative Models. bioRxiv, pp. 757252, 2019. doi: 10. 1101/757252. URL https://www.biorxiv.org/ content/10.1101/757252v1. · DOI
  5. Sturmfels, P., Vig, J., Madani, A., and Rajani, N. F. Profile Prediction: An Alignment-Based Pre-Training Task for Protein Sequence Models. bioRxiv, 11 2020. URL http: //arxiv.org/abs/2012.00195.
  6. Thomas, J., Ramakrishnan, N., and Bailey-Kellogg, C. Graphical models of residue coupling in protein families, 4 2008. ISSN 15455963. URL https://pubmed. ncbi.nlm.nih.gov/18451428/.
  7. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Kaiser, Ł., and Polo- sukhin, I. Attention is All you Need. Advances in Neural Information Processing Systems 30, pp. 5998- 6008, 2017. URL http://papers.nips.cc/ paper/7181-attention-is-all-you-need. pdfhttps://papers.nips.cc/paper/ 7181-attention-is-all-you-need.pdf.

Cited by

Sorted by citations

Authors
Rao, R., Liu, J., Verkuil, R., Meier, J., Canny, J. F., Abbeel, P., Sercu, T., & Rives, A.
Affiliations
  • Facility for Antiproton and Ion Research (DE)
  • Meta (United States) (US)
  • University of California, Berkeley (US)
Corresponding author
  • Rao, Roshan
  • Rives, Alexander
Venue
bioRxiv (Cold Spring Harbor Laboratory)
Publisher
openRxiv
Published
2021-02-13
OpenAlex
W3133458480
Cited by
350
References
47
Citations by year
2026: 63 · 2025: 90 · 2024: 74 · 2023: 45 · 2022: 47 · 2021: 28 · 2020: 3
Type
posted content
Language
en
Open access
Locations
bioRxiv (Cold Spring Harbor Laboratory) (acceptedVersion, cc-by-nc-nd)
Funders
Defense Advanced Research Projects Agency
Topics
  • Life Sciences › Biochemistry, Genetics and Molecular Biology › Molecular Biology › Machine Learning in Bioinformatics
  • Life Sciences › Biochemistry, Genetics and Molecular Biology › Molecular Biology › Protein Structure and Dynamics
  • Life Sciences › Biochemistry, Genetics and Molecular Biology › Molecular Biology › Genomics and Phylogenetic Studies
Keywords
protein language models, multiple sequence alignment, masked language modeling, protein structure prediction, parameter efficiency

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review