GAProtoNet: A Multi-head Graph Attention-based Prototypical Network for Interpretable Text Classification

Ximing Wen, Wenjuan Tan, Rosina Weber


Abstract
Pretrained transformer-based Language Models (LMs) are well-known for their ability to achieve significant improvement on text classification tasks with their powerful word embeddings, but their black-box nature, which leads to a lack of interpretability, has been a major concern. In this work, we introduce GAProtoNet, a novel white-box Multi-head Graph Attention-based Prototypical Network designed to explain the decisions of text classification models built with LM encoders. In our approach, the input vector and prototypes are regarded as nodes within a graph, and we utilize multi-head graph attention to selectively construct edges between the input node and prototype nodes to learn an interpretable prototypical representation. During inference, the model makes decisions based on a linear combination of activated prototypes weighted by the attention score assigned for each prototype, allowing its choices to be transparently explained by the attention weights and the prototypes. Experiments on multiple public datasets show our approach achieves superior results without sacrificing the accuracy of the original black-box LMs. We also compare with four alternative prototypical network variations and our approach achieves the best accuracy and F1 among all. Our case study and visualization of prototype clusters also demonstrate the efficiency in explaining the decisions of black-box models built with LMs.
Anthology ID:
2025.coling-main.661
Volume:
Proceedings of the 31st International Conference on Computational Linguistics
Month:
January
Year:
2025
Address:
Abu Dhabi, UAE
Editors:
Owen Rambow, Leo Wanner, Marianna Apidianaki, Hend Al-Khalifa, Barbara Di Eugenio, Steven Schockaert
Venue:
COLING
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
9891–9901
Language:
URL:
https://aclanthology.org/2025.coling-main.661/
DOI:
Bibkey:
Cite (ACL):
Ximing Wen, Wenjuan Tan, and Rosina Weber. 2025. GAProtoNet: A Multi-head Graph Attention-based Prototypical Network for Interpretable Text Classification. In Proceedings of the 31st International Conference on Computational Linguistics, pages 9891–9901, Abu Dhabi, UAE. Association for Computational Linguistics.
Cite (Informal):
GAProtoNet: A Multi-head Graph Attention-based Prototypical Network for Interpretable Text Classification (Wen et al., COLING 2025)
Copy Citation:
PDF:
https://aclanthology.org/2025.coling-main.661.pdf