Fast and Interpretable Protein Substructure Alignment via Optimal Transport
Zhiyu Wang, Bingxin Zhou, Weishu Zhao, Yang Tan, Jing Wang, Pietro Lio, Liang Hong
Abstract
Proteins are essential biological macromolecules that execute life functions. Local motifs within protein structures, such as active sites, are the most critical components for linking structure to function and are key to understanding protein evolution and enabling protein engineering. Existing computational methods struggle to identify and compare these local structures, which leaves a significant gap in understanding protein structures and harnessing their functions. This study presents PLASMA, the first deep learning framework for efficient and interpretable residue-level protein substructure alignment. We reformulate the problem as a regularized optimal transport task and leverage differentiable Sinkhorn iterations. For a pair of input protein structures, PLASMA outputs a clear alignment matrix with an interpretable overall similarity score. Through extensive quantitative evaluations and three biological case studies, we demonstrate that PLASMA achieves accurate, lightweight, and interpretable residue-level alignment. Additionally, we introduce PLASMA-PF, a training-free variant that provides a practical alternative when training data are unavailable. Our method addresses a critical gap in protein structure analysis tools and offers new opportunities for functional annotation, evolutionary studies, and structure-based drug design. Reproducibility is ensured via our official implementation at https://github.com/ZW471/PLASMA-Protein-Local-Alignment.git.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on4
- Deep Shells: Unsupervised Shape Correspondence with Optimal TransportMarvin Eisenberger, Aysim Toker, Laura Leal-Taixé, Daniel CremersNeurIPS 2020 · 107 citations
- Evaluating Representation Learning on the Protein Structure UniverseArian Rokkum Jamasb, Alex Morehead, Chaitanya K. Joshi, Zuobai Zhang et al.ICLR 2024 · 26 citations
- Iteratively Refined Early Interaction Alignment for Subgraph Matching based Graph RetrievalAshwin Ramachandran, Vaibhav Raj, Indradyumna Roy, Soumen Chakrabarti et al.NeurIPS 2024 · 7 citations
- Charting the Design Space of Neural Graph Representations for Subgraph MatchingVaibhav Raj, Indradyumna Roy, Ashwin Ramachandran, Soumen Chakrabarti et al.ICLR 2025
Related papers
- VenusX: Unlocking Fine-Grained Functional Understanding of ProteinsYang Tan, Wenrui Gou, Bozitao Zhong, Huiqun Yu et al.ICLR 2026 · 6 citations
- Greater than the Sum of Its Parts: Building Substructure into Protein Encoding ModelsRobert Calef, Arthur Liang, Manolis Kellis, Marinka ZitnikICLR 2026 · 2 citations
- Fast End-to-End Learning on Protein SurfacesFreyr Sverrisson, Jean Feydy, Bruno E. Correia, Michael M. BronsteinCVPR 2021
- Domain-Aware Multi-View Contrastive Representation Learning for Protein Subcellular Localization PredictionQiang Zhang, Feng Yang, Weihong Huang, Jing Feng et al.AAAI 2026
- Representational Alignment Across Model Layers and Brain Regions with Multi-Level Optimal TransportShaan Shah, Meenakshi KhoslaICLR 2026 · 4 citations
