Tuukka Ruotsalo
2025
As easy as PIE: understanding when pruning causes language models to disagree
Pietro Tropeano
|
Maria Maistro
|
Tuukka Ruotsalo
|
Christina Lioma
Findings of the Association for Computational Linguistics: NAACL 2025
Language Model (LM) pruning compresses the model by removing weights, nodes, or other parts of its architecture. Typically, pruning focuses on the resulting efficiency gains at the cost of effectiveness.However, when looking at how individual data pointsare affected by pruning, it turns out that a particular subset of data points always bears most of the brunt (in terms of reduced accuracy) when pruning,but this effect goes unnoticed when reporting the mean accuracy of all data points. These data points are called PIEs and have been studied in image processing, but not in NLP.In a study of various NLP datasets, pruning methods, and levels of compression, we find that PIEs impact inference quality considerably, regardless of class frequency, andthat BERT is more prone to this than BiLSTM. We also find that PIEs contain a high amount of data points that have the largest influence on how well the model generalises to unseen data. This means that when pruning, with seemingly moderate loss to accuracy across all data points, we in fact hurt tremendously those data points that matter the most. We trace what makes PIEs both hard and impactful to inference to their overall longer and more semantically complex text. These findings are novel and contribute to understanding how LMs are affected by pruning. The code is available at: https://github.com/pietrotrope/AsEasyAsPIE
2024
An Unsupervised Approach to Achieve Supervised-Level Explainability in Healthcare Records
Joakim Edin
|
Maria Maistro
|
Lars Maaløe
|
Lasse Borgholt
|
Jakob Drachmann Havtorn
|
Tuukka Ruotsalo
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Electronic healthcare records are vital for patient safety as they document conditions, plans, and procedures in both free text and medical codes. Language models have significantly enhanced the processing of such records, streamlining workflows and reducing manual data entry, thereby saving healthcare providers significant resources. However, the black-box nature of these models often leaves healthcare professionals hesitant to trust them. State-of-the-art explainability methods increase model transparency but rely on human-annotated evidence spans, which are costly. In this study, we propose an approach to produce plausible and faithful explanations without needing such annotations. We demonstrate on the automated medical coding task that adversarial robustness training improves explanation plausibility and introduce AttInGrad, a new explanation method superior to previous ones. By combining both contributions in a fully unsupervised setup, we produce explanations of comparable quality, or better, to that of a supervised approach. We release our code and model weights.
Search
Fix data
Co-authors
- Maria Maistro 2
- Lasse Borgholt 1
- Joakim Edin 1
- Jakob Drachmann Havtorn 1
- Christina Lioma 1
- show all...