Biological structure and function emerge from scaling unsupervised learning to 250 million protein sequences.

Journal: Proceedings of the National Academy of Sciences of the United States of America
Published Date:

Abstract

In the field of artificial intelligence, a combination of scale in data and model capacity enabled by unsupervised learning has led to major advances in representation learning and statistical generation. In the life sciences, the anticipated growth of sequencing promises unprecedented data on natural sequence diversity. Protein language modeling at the scale of evolution is a logical step toward predictive and generative artificial intelligence for biology. To this end, we use unsupervised learning to train a deep contextual language model on 86 billion amino acids across 250 million protein sequences spanning evolutionary diversity. The resulting model contains information about biological properties in its representations. The representations are learned from sequence data alone. The learned representation space has a multiscale organization reflecting structure from the level of biochemical properties of amino acids to remote homology of proteins. Information about secondary and tertiary structure is encoded in the representations and can be identified by linear projections. Representation learning produces features that generalize across a range of applications, enabling state-of-the-art supervised prediction of mutational effect and secondary structure and improving state-of-the-art features for long-range contact prediction.

Authors

  • Alexander Rives
    Facebook AI Research, New York, NY 10003; arives@cs.nyu.edu.
  • Joshua Meier
    Facebook AI Research, New York, NY 10003.
  • Tom Sercu
    IBM Thomas J. Watson Research Center, Yorktown Heights, NY, USA.
  • Siddharth Goyal
    Facebook AI Research, New York, NY 10003.
  • Zeming Lin
    Department of Computer Science, New York University, New York, NY 10012.
  • Jason Liu
    Program in Computational Biology and Bioinformatics, Yale University, New Haven, Connecticut, United States of America.
  • Demi Guo
    Department of Computer Science, Harvard University.
  • Myle Ott
    Facebook AI Research, New York, NY 10003.
  • C Lawrence Zitnick
    Facebook Artificial Intelligence Research, Menlo Park, CA.
  • Jerry Ma
    Booth School of Business, University of Chicago, Chicago, IL 60637.
  • Rob Fergus
    Department of Computer Science, New York University, New York, NY 10012.