arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.05945cs.LGcs.AI
收藏

消除硬概念瓶颈模型中的信息泄漏:基于监督、分层概念学习

Eliminating Information Leakage in Hard Concept Bottleneck Models with Supervised, Hierarchical Concept Learning

The Hong Kong University of Science and Technology(香港科技大学)

Ao Sun, Yuanyuan Yuan, Pingchuan Ma, Shuai Wang

AI总结:

针对概念瓶颈模型的信息泄漏问题,提出SupCBM,通过标签监督和分层概念集,利用干预矩阵实现标签预测,显著减少信息泄漏并提升性能。

摘要:

概念瓶颈模型(CBMs)旨在通过以人类可理解的概念连接特征和标签,提供可解释和可干预的预测。虽然最近的CBMs显示出有前景的潜力,但它们存在信息泄漏问题,即超出概念之外的意外信息(无论概念是以概率还是二元状态表示)会泄漏到后续的标签预测中。因此,不同的类别通过难以区分的概念被错误分类,破坏了CBMs的解释和干预能力。本文通过在概念预测中引入标签监督并构建分层概念集来缓解信息泄漏问题。据此,我们提出了一种新的CBMs范式,即SupCBM,它通过预测的概念和精心设计的干预矩阵来实现标签预测。SupCBM关注与预测标签最相关的概念,并且仅在呈现不同概念时区分类别。我们的评估表明,SupCBM在多种数据集上优于最先进的CBMs。它还在不同的骨干模型上表现出更好的泛化性。通过对不同CBMs中信息泄漏的适当量化,我们证明SupCBM显著减少了信息泄漏。

英文摘要:

Concept Bottleneck Models (CBMs) aim to deliver interpretable and interventionable predictions by bridging features and labels with human-understandable concepts. While recent CBMs show promising potential, they suffer from information leakage, where unintended information beyond the concepts (either when concepts are represented with probabilities or binary states) are leaked to the subsequent label prediction. Consequently, distinct classes are falsely classified via indistinguishable concepts, undermining the interpretation and intervention of CBMs. This paper alleviates the information leakage issue by introducing label supervision in concept predication and constructing a hierarchical concept set. Accordingly, we propose a new paradigm of CBMs, namely SupCBM, which achieves label predication via predicted concepts and a deliberately-designed intervention matrix. SupCBM focuses on concepts that are mostly relevant to the predicted label and only distinguishes classes when different concepts are presented. Our evaluations show that SupCBM outperforms SOTA CBMs over diverse datasets. It also manifests better generality across different backbone models. With proper quantification of information leakage in different CBMs, we demonstrate that SupCBM significantly reduces the information leakage.

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

↑