Hierarchical Graph Representation Learning From a Statistical Perspective for Generalizable and Interpretable...

Protein-ligand binding affinity (PLA) prediction aims to guide rational drug design by estimating the strength of interaction. The effectiveness of the representation learning of protein and ligand is key to successful PLA prediction. To this end, attention mechanism, as a powerful architectural paradigm, has been introduced and gradually emerged as the prevailing approach. However, intuitively, the classical attention paradigm based on similarity does not fit the biological mechanisms relevant for binding. Worse still, the cooperative and antagonistic effects among multiple atoms are deliberately disregarded in the classical formulation of attention mechanisms. Consequently, the rigid transplantation of classical architectures substantially undermines the PLA prediction performance. To address these challenges, we employ a hierarchical statistical attention model (HISA). Specifically, HISA employs a statistical attention mechanism (SAM) based on non-similarity computation to fit the biological prior and perceive the relationship of multiple atoms. In addition, we optimize HISA by employing clustering, enabling hierarchical representations of biomolecules. Extensive experiments demonstrate that HISA achieves state-of-the-art performance on multiple PLA benchmarks while simultaneously exhibiting generalizability and interpretability.