Structural Interpretations of Protein Language Model Representations via Differentiable Graph Partitioning

Framework using SoftBlobGIN with differentiable Gumbel-softmax pooling to project ESM-2 protein language model representations onto contact graphs for structure-aware interpretation of dense residue embeddings.