about
Advancing Interpretability in Text Classification Through Prototype Learning (arxiv.org)
2 points by PaulHoule on Nov 6, 2024 | hide | past | pdf | discuss on HN

In plain words: ProtoLens is a text classifier that learns example-like patterns and points to the exact short phrases in a text that match each one, so you can see why it decided. It beat both pattern-based and black-box rivals on several text benchmarks while staying accurate.

Abstract · Advancing Interpretability in Text Classification through Prototype Learning

Deep neural networks have achieved remarkable performance in various text-based tasks but often lack interpretability, making them less suitable for applications where transparency is critical. To address this, we propose ProtoLens, a novel prototype-based model that provides fine-grained, sub-sentence level interpretability for text classification. ProtoLens uses a Prototype-aware Span Extraction module to identify relevant text spans associated with learned prototypes and a Prototype Alignment mechanism to ensure prototypes are semantically meaningful throughout training. By aligning the prototype embeddings with human-understandable examples, ProtoLens provides interpretable predictions while maintaining competitive accuracy. Extensive experiments demonstrate that ProtoLens outperforms both prototype-based and non-interpretable baselines on multiple text classification benchmarks. Code and data are available at \url{https://anonymous.4open.science/r/ProtoLens-CE0B/}.

Bowen Wei, Ziwei Zhu
arXiv:2410.17546 · cs.CL, cs.AI · submitted Oct 23, 2024 · updated Oct 24, 2024
abstract · pdf · html

add comment on HN