PEACH: Pretrained-embedding Explanation Across Contextual and Hierarchical Structure

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cao, Feiqi, Han, Caren, Chung, Hyunsuk
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910417325916160
author Cao, Feiqi
Han, Caren
Chung, Hyunsuk
author_facet Cao, Feiqi
Han, Caren
Chung, Hyunsuk
contents In this work, we propose a novel tree-based explanation technique, PEACH (Pretrained-embedding Explanation Across Contextual and Hierarchical Structure), that can explain how text-based documents are classified by using any pretrained contextual embeddings in a tree-based human-interpretable manner. Note that PEACH can adopt any contextual embeddings of the PLMs as a training input for the decision tree. Using the proposed PEACH, we perform a comprehensive analysis of several contextual embeddings on nine different NLP text classification benchmarks. This analysis demonstrates the flexibility of the model by applying several PLM contextual embeddings, its attribute selections, scaling, and clustering methods. Furthermore, we show the utility of explanations by visualising the feature selection and important trend of text classification via human-interpretable word-cloud-based trees, which clearly identify model mistakes and assist in dataset debugging. Besides interpretability, PEACH outperforms or is similar to those from pretrained models.
format Preprint
id arxiv_https___arxiv_org_abs_2404_13645
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PEACH: Pretrained-embedding Explanation Across Contextual and Hierarchical Structure
Cao, Feiqi
Han, Caren
Chung, Hyunsuk
Computation and Language
In this work, we propose a novel tree-based explanation technique, PEACH (Pretrained-embedding Explanation Across Contextual and Hierarchical Structure), that can explain how text-based documents are classified by using any pretrained contextual embeddings in a tree-based human-interpretable manner. Note that PEACH can adopt any contextual embeddings of the PLMs as a training input for the decision tree. Using the proposed PEACH, we perform a comprehensive analysis of several contextual embeddings on nine different NLP text classification benchmarks. This analysis demonstrates the flexibility of the model by applying several PLM contextual embeddings, its attribute selections, scaling, and clustering methods. Furthermore, we show the utility of explanations by visualising the feature selection and important trend of text classification via human-interpretable word-cloud-based trees, which clearly identify model mistakes and assist in dataset debugging. Besides interpretability, PEACH outperforms or is similar to those from pretrained models.
title PEACH: Pretrained-embedding Explanation Across Contextual and Hierarchical Structure
topic Computation and Language
url https://arxiv.org/abs/2404.13645