arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.23947cs.LGcs.AI

层次化概念基于的可解释模型

Hierarchical Concept-based Interpretable Models

University of Cambridge(剑桥大学);University of Oxford(牛津大学)

Oscar Hill, Mateo Espinosa Zarlenga, Mateja Jamnik

AI总结:

本文提出层次化概念嵌入模型HiCEMs,通过概念分裂方法自动发现细粒度子概念,提升模型可解释性和任务准确性。

摘要:

现代深度神经网络由于其潜在表示的不透明性,仍然难以解释,这阻碍了模型的理解、调试和去偏。概念嵌入模型(CEMs)通过将输入映射到人类可解释的概念表示来解决这一问题,从而可以预测任务。然而,CEMs无法表示概念之间的关系,并且在训练过程中需要不同粒度的概念注释,限制了其应用。在本文中,我们引入了层次化概念嵌入模型(HiCEMs),一种新的CEMs家族,通过层次结构显式建模概念关系。为了使HiCEMs在现实环境中应用,我们提出了概念分裂,一种方法,用于从预训练CEM的嵌入空间自动发现更细粒度的子概念,而无需额外注释。这使HiCEMs能够在有限的概念标签下生成细粒度的解释,减少注释负担。我们的评估涵盖了多个数据集,包括用户研究和在PseudoKitchens上的实验,这是一个新提出的基于概念的数据集,包含3D厨房渲染。结果表明:(1)概念分裂发现了训练期间不存在的人类可解释的子概念,可用于训练高准确度的HiCEMs;(2)HiCEMs能够在不同粒度上实现强大的测试时间概念干预,从而提高任务准确性。

英文摘要:

Modern deep neural networks remain challenging to interpret due to the opacity of their latent representations, impeding model understanding, debugging, and debiasing. Concept Embedding Models (CEMs) address this by mapping inputs to human-interpretable concept representations from which tasks can be predicted. Yet, CEMs fail to represent inter-concept relationships and require concept annotations at different granularities during training, limiting their applicability. In this paper, we introduce Hierarchical Concept Embedding Models (HiCEMs), a new family of CEMs that explicitly model concept relationships through hierarchical structures. To enable HiCEMs in real-world settings, we propose Concept Splitting, a method for automatically discovering finer-grained sub-concepts from a pretrained CEM's embedding space without requiring additional annotations. This allows HiCEMs to generate fine-grained explanations from limited concept labels, reducing annotation burdens. Our evaluation across multiple datasets, including a user study and experiments on PseudoKitchens, a newly proposed concept-based dataset of 3D kitchen renders, demonstrates that (1) Concept Splitting discovers human-interpretable sub-concepts absent during training that can be used to train highly accurate HiCEMs, and (2) HiCEMs enable powerful test-time concept interventions at different granularities, leading to improved task accuracy.

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

↑