Paying Attention to Attention: High Attention Sites as Indicators of Protein Family and Function in Language Models

Abstract

Protein Language Models (PLMs) use transformer architectures to capture patterns within protein sequences, providing a powerful computational representation of the protein sequence [1]. Through large-scale training on protein sequence data, PLMs generate vector representations that encapsulate the biochemical and structural properties of proteins [2]. At the core of PLMs is the attention mechanism, which facilitates the capture of long-range dependencies by computing pairwise importance scores across residues, thereby highlighting regions of biological interaction within the sequence [3]. The attention matrices offer an untapped opportunity to uncover specific biological properties of proteins, particularly their functions. In this work, we introduce a novel approach, using the Evolutionary Scale Model (ESM) [4], for identifying High Attention (HA) sites within protein sequences, corresponding to key residues that define protein families. By examining attention patterns across multiple layers, we pinpoint residues that contribute most to family classification and function prediction. Our contributions are as follows: (1) we propose a method for identifying HA sites at critical residues from the middle layers of the PLM; (2) we demonstrate that these HA sites provide interpretable links to biological functions; and (3) we show that HA sites improve active site predictions for functions of unannotated proteins. We make available the HA sites for the human proteome. This work offers a broadly applicable approach to protein classification and functional annotation and provides a biological interpretation of the PLMs representation. 1 Author SummaryUnderstanding how proteins work is critical to advancements in biology and medicine, and protein language models (PLMs) facilitate studying protein sequences at scale. These models identify patterns within protein sequences by focusing on key regions of the sequence that are important to distinguish the protein. Our work focuses on the Evolutionary Scale Model (ESM), a state-of-the-art PLM, and we analyze the models internal attention mechanism to identify the significant residues. We developed a new method to identify "High Attention (HA)" sites--specific parts of a protein sequence that are essential for classifying proteins into families and predicting their functions. By analyzing how the model prioritizes certain regions of protein sequences, we discovered that these HA sites often correspond to residues critical for biological activity, such as active sites where chemical reactions occur. Our approach helps interpret how PLMs understand protein data and enhances predictions for proteins whose functions are still unknown. As part of this work, we provide HA-site information for the entire human proteome, offering researchers a resource to further study the potential functional relevance of these residues.