Interpretable-by-Design Text Understanding with Iteratively Generated Concept Bottleneck

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ludan, Josh Magnus, Lyu, Qing, Yang, Yue, Dugan, Liam, Yatskar, Mark, Callison-Burch, Chris
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929300990590976
author Ludan, Josh Magnus
Lyu, Qing
Yang, Yue
Dugan, Liam
Yatskar, Mark
Callison-Burch, Chris
author_facet Ludan, Josh Magnus
Lyu, Qing
Yang, Yue
Dugan, Liam
Yatskar, Mark
Callison-Burch, Chris
contents Black-box deep neural networks excel in text classification, yet their application in high-stakes domains is hindered by their lack of interpretability. To address this, we propose Text Bottleneck Models (TBM), an intrinsically interpretable text classification framework that offers both global and local explanations. Rather than directly predicting the output label, TBM predicts categorical values for a sparse set of salient concepts and uses a linear layer over those concept values to produce the final prediction. These concepts can be automatically discovered and measured by a Large Language Model (LLM) without the need for human curation. Experiments on 12 diverse text understanding datasets demonstrate that TBM can rival the performance of black-box baselines such as few-shot GPT-4 and finetuned DeBERTa while falling short against finetuned GPT-3.5. Comprehensive human evaluation validates that TBM can generate high-quality concepts relevant to the task, and the concept measurement aligns well with human judgments, suggesting that the predictions made by TBMs are interpretable. Overall, our findings suggest that TBM is a promising new framework that enhances interpretability with minimal performance tradeoffs.
format Preprint
id arxiv_https___arxiv_org_abs_2310_19660
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Interpretable-by-Design Text Understanding with Iteratively Generated Concept Bottleneck
Ludan, Josh Magnus
Lyu, Qing
Yang, Yue
Dugan, Liam
Yatskar, Mark
Callison-Burch, Chris
Computation and Language
Black-box deep neural networks excel in text classification, yet their application in high-stakes domains is hindered by their lack of interpretability. To address this, we propose Text Bottleneck Models (TBM), an intrinsically interpretable text classification framework that offers both global and local explanations. Rather than directly predicting the output label, TBM predicts categorical values for a sparse set of salient concepts and uses a linear layer over those concept values to produce the final prediction. These concepts can be automatically discovered and measured by a Large Language Model (LLM) without the need for human curation. Experiments on 12 diverse text understanding datasets demonstrate that TBM can rival the performance of black-box baselines such as few-shot GPT-4 and finetuned DeBERTa while falling short against finetuned GPT-3.5. Comprehensive human evaluation validates that TBM can generate high-quality concepts relevant to the task, and the concept measurement aligns well with human judgments, suggesting that the predictions made by TBMs are interpretable. Overall, our findings suggest that TBM is a promising new framework that enhances interpretability with minimal performance tradeoffs.
title Interpretable-by-Design Text Understanding with Iteratively Generated Concept Bottleneck
topic Computation and Language
url https://arxiv.org/abs/2310.19660