B-cos LM: Efficiently Transforming Pre-trained Language Models for Improved Explainability

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Yifan, Rao, Sukrut, Lee, Ji-Ung, Jobanputra, Mayank, Demberg, Vera
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912755219431424
author Wang, Yifan
Rao, Sukrut
Lee, Ji-Ung
Jobanputra, Mayank
Demberg, Vera
author_facet Wang, Yifan
Rao, Sukrut
Lee, Ji-Ung
Jobanputra, Mayank
Demberg, Vera
contents Post-hoc explanation methods for black-box models often struggle with faithfulness and human interpretability due to the lack of explainability in current neural architectures. Meanwhile, B-cos networks have been introduced to improve model explainability by proposing an architecture that removes bias terms and promotes input-weight alignment. Although B-cos networks have shown success in building explainable systems, their application has so far been limited to computer vision models and their associated training pipelines. In this work, we introduce B-cos LMs, i.e., B-cos Language Models (LMs) empowered for natural language processing (NLP) tasks. Our approach directly transforms pre-trained language models into B-cos LMs by combining B-cos conversion and task fine-tuning, improving efficiency compared to previous methods. Automatic and human evaluation results demonstrate that B-cos LMs produce more faithful and human interpretable explanations than post-hoc methods, while maintaining task performance comparable to conventional fine-tuning. Our in-depth analysis explores how B-cos LMs differ from conventionally fine-tuned models in their learning processes and explanation patterns. Finally, we present a first exploration of transforming decoder-only models to B-cos LMs for generation tasks. Our code is available at https://github.com/Ewanwong/bcos_lm.
format Preprint
id arxiv_https___arxiv_org_abs_2502_12992
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle B-cos LM: Efficiently Transforming Pre-trained Language Models for Improved Explainability
Wang, Yifan
Rao, Sukrut
Lee, Ji-Ung
Jobanputra, Mayank
Demberg, Vera
Computation and Language
Artificial Intelligence
Post-hoc explanation methods for black-box models often struggle with faithfulness and human interpretability due to the lack of explainability in current neural architectures. Meanwhile, B-cos networks have been introduced to improve model explainability by proposing an architecture that removes bias terms and promotes input-weight alignment. Although B-cos networks have shown success in building explainable systems, their application has so far been limited to computer vision models and their associated training pipelines. In this work, we introduce B-cos LMs, i.e., B-cos Language Models (LMs) empowered for natural language processing (NLP) tasks. Our approach directly transforms pre-trained language models into B-cos LMs by combining B-cos conversion and task fine-tuning, improving efficiency compared to previous methods. Automatic and human evaluation results demonstrate that B-cos LMs produce more faithful and human interpretable explanations than post-hoc methods, while maintaining task performance comparable to conventional fine-tuning. Our in-depth analysis explores how B-cos LMs differ from conventionally fine-tuned models in their learning processes and explanation patterns. Finally, we present a first exploration of transforming decoder-only models to B-cos LMs for generation tasks. Our code is available at https://github.com/Ewanwong/bcos_lm.
title B-cos LM: Efficiently Transforming Pre-trained Language Models for Improved Explainability
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2502.12992