Robust Text Classification: Analyzing Prototype-Based Networks

Zhivar Sourati, Darshan Girish Deshpande, Filip Ilievski, Kiril Gashteovski, Sascha Saralajew


Abstract
Downstream applications often require text classification models to be accurate and robust. While the accuracy of state-of-the-art Language Models (LMs) approximates human performance, they often exhibit a drop in performance on real-world noisy data. This lack of robustness can be concerning, as even small perturbations in text, irrelevant to the target task, can cause classifiers to incorrectly change their predictions. A potential solution can be the family of Prototype-Based Networks (PBNs) that classifies examples based on their similarity to prototypical examples of a class (prototypes) and has been shown to be robust to noise for computer vision tasks. In this paper, we study whether the robustness properties of PBNs transfer to text classification tasks under both targeted and static adversarial attack settings. Our results show that PBNs, as a mere architectural variation of vanilla LMs, offer more robustness compared to vanilla LMs under both targeted and static settings. We showcase how PBNs’ interpretability can help us understand PBNs’ robustness properties. Finally, our ablation studies reveal the sensitivity of PBNs’ robustness to the strictness of clustering and the number of prototypes in the training phase, as tighter clustering and a low number of prototypes result in less robust PBNs.
Anthology ID:
2024.findings-emnlp.745
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2024
Month:
November
Year:
2024
Address:
Miami, Florida, USA
Editors:
Yaser Al-Onaizan, Mohit Bansal, Yun-Nung Chen
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
12736–12757
Language:
URL:
https://aclanthology.org/2024.findings-emnlp.745/
DOI:
10.18653/v1/2024.findings-emnlp.745
Bibkey:
Cite (ACL):
Zhivar Sourati, Darshan Girish Deshpande, Filip Ilievski, Kiril Gashteovski, and Sascha Saralajew. 2024. Robust Text Classification: Analyzing Prototype-Based Networks. In Findings of the Association for Computational Linguistics: EMNLP 2024, pages 12736–12757, Miami, Florida, USA. Association for Computational Linguistics.
Cite (Informal):
Robust Text Classification: Analyzing Prototype-Based Networks (Sourati et al., Findings 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.findings-emnlp.745.pdf
Software:
 2024.findings-emnlp.745.software.zip