Cited by 137Machine Learning in BioinformaticsProtein Structure and DynamicsGenomics and Phylogenetic Studies

Convolutions are competitive with transformers for protein sequence pretraining

Open access
Yang, Kevin K., Fusi, Nicolo, Lu, Alex X.·2024·Cell Systems

Published version of a preprint

Abstract

Pretrained protein sequence language models have been shown to improve the performance of many prediction tasks and are now routinely integrated into bioinformatics tools. However, these models largely rely on the transformer architecture, which scales quadratically with sequence length in both run-time and memory. Therefore, state-of-the-art models have limitations on sequence length. To address this limitation, we investigated whether convolutional neural network (CNN) architectures, which scale linearly with sequence length, could be as effective as transformers in protein language models. With masked language model pretraining, CNNs are competitive with, and occasionally superior to, transformers across downstream applications while maintaining strong performance on sequences longer than those allowed in the current state-of-the-art transformer models. Our work suggests that computational efficiency can be improved without sacrificing performance, simply by using a CNN architecture instead of a transformer, and emphasizes the importance of disentangling pretraining task and model architecture. A record of this paper's transparent peer review process is included in the supplemental information.

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.1016/j.cels.2024.01.008

Cites

48 of 48 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?

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: 6

  1. Rao, R.M., Liu, J., Verkuil, R., Meier, J., Canny, J., Abbeel, P., Sercu, T., and Rives, A. (2021). MSA transformer. In Proceedings of the 38th International Conference on Machine Learning PMLR, 139, M. Meila and T. Zhang, eds., pp. 8844-8856. https://proceedings.mlr.press/v139/ rao21a.html.
  2. Dallago, C., Mou, J., Johnston, KadinaE., Wittmann, B., Bhattacharya, N., Goldman, S., Madani, A., and Yang, K.K. (2021). FLIP: benchmark tasks in fitness landscape inference for proteins. In Thirty-fifth Conference on Neural Information Processing Systems Datasets and Benchmarks Track (Round 2).
  3. Tay, Y., Bahri, D., Yang, L., Metzler, D., and Juan, D.-C. (2020a). Sparse sinkhorn attention. In International Conference on Machine Learning PMLR, pp. 9438-9447.
  4. Zaheer, M., Guruganesh, G., Dubey, K.A., Ainslie, J., Alberti, C., Ontanon, S., Pham, P., Ravula, A., Wang, Q., Yang, L., et al. (2020). Big bird: trans- formers for longer sequences. Adv. Neural Inf. Process. Syst. 33, 17283-17297.
  5. Zhang, Zuobai, Xu, M., Jamasb, A.R., Chenthamarakshan, V., Lozano, A., Das, P., and Tang, J. (2023). Protein representation learning by geometric structure pretraining. In The Eleventh International Conference on Learning Representations https://openreview.net/forum?id=to3qCB3tOh9.
  6. Hsu, C., Verkuil, R., Liu, J., Lin, Z., Hie, B., Sercu, T., Lerer, A., and Rives, A. (2022). Learning inverse folding from millions of predicted structures. In International Conference on Machine Learning PMLR, pp. 8946-8970.

Cited by

Sorted by citations

Authors
Yang, K. K., Fusi, N., & Lu, A. X.
Affiliations
Microsoft Research New England (United States) (US)
Corresponding author
Yang, Kevin K.
Venue
Cell Systems
Publisher
Elsevier BV
Published
2024-03
Volume
15
Issue
3
Pages
286-294.e2
ISSN
2405-4712 (ISSN-L)
OpenAlex
W4392302569
Cited by
137
References
48
Citations by year
2026: 28 · 2025: 46 · 2024: 36 · 2023: 22 · 2022: 5
FWCI
15.00
Type
journal article
Language
en
Open access
Locations
  • Cell Systems (publishedVersion)
  • bioRxiv (Cold Spring Harbor Laboratory) (acceptedVersion, public-domain)
  • PubMed (publishedVersion)
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
convolutional neural networks, transformer models, computational efficiency, protein sequence prediction
MeSH terms
Amino Acid Sequence, Peer Review, Neural Networks, Computer, Computational Biology

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