arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.13645cs.CL
收藏

PEACH:跨上下文与层次结构的预训练嵌入解释

PEACH: Pretrained-embedding Explanation Across Contextual and Hierarchical Structure

University of Melbourne(墨尔本大学);University of Sydney(悉尼大学);FortifyEdge

Feiqi Cao, Caren Han, Hyunsuk Chung

AI总结:

本文提出基于树的解释技术PEACH,利用任意预训练上下文嵌入以人类可解释的方式解释文本分类,在9个NLP基准上验证其灵活性与实用性,并能识别模型错误协助数据集调试。

摘要:

在这项工作中,我们提出了一种新颖的基于树的解释技术,PEACH(Pretrained-embedding Explanation Across Contextual and Hierarchical Structure),它能够以基于树的人类可解释方式,使用任何预训练上下文嵌入来解释文本文档是如何被分类的。注意,PEACH可以将PLMs的任何上下文嵌入作为决策树的训练输入。使用提出的PEACH,我们在9个不同的NLP文本分类基准上对几种上下文嵌入进行了全面分析。该分析通过应用几种PLM上下文嵌入、其属性选择、缩放和聚类方法,展示了模型的灵活性。此外,我们通过基于人类可解释词云的树可视化特征选择和文本分类的重要趋势,展示了解释的实用性,这能清晰地识别模型错误并协助数据集调试。除可解释性外,PEACH的表现优于或类似于预训练模型。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑