arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.09159cs.LGcs.AIcs.CV
收藏

VAE-CE:使用解纠缠VAE的视觉对比解释

VAE-CE: Visual Contrastive Explanation using Disentangled VAEs

Eindhoven University of Technology(埃因霍温理工大学)

Yoeri Poels, Vlado Menkovski

摘要:

分类模型的目标是为数据分配正确的标签。在大多数情况下,这些数据并没有被给定的标签集完全描述。领域中通常存在一组丰富的有意义的概念,可以更精确地描述每个数据点。这些概念对于解释模型的分类也非常有用。在本文中,我们提出了一个模型,表示为变分自编码器为基础的对比解释(VAE-CE),表示数据与高层次的概念,并使用这种表示的分类和生成的解释。解释以对比的方式产生,传达为什么数据点被分配给一个类而不是另一个类。解释被指定为输入数据点的一组转换,每一步都描述了一个概念向对比类的变化。我们使用一个解开的VAE,扩展与一个新的监督方法解开个别尺寸的模型。对合成数据和MNIST的分析表明,解纠缠和解释的方法提供了优于其他方法的优点。

英文摘要:

The goal of a classification model is to assign the correct labels to data. In most cases, this data is not fully described by the given set of labels. Often a rich set of meaningful concepts exist in the domain that can much more precisely describe each datapoint. Such concepts can also be highly useful for interpreting the model's classifications. In this paper we propose a model, denoted as Variational Autoencoder-based Contrastive Explanation (VAE-CE), that represents data with high-level concepts and uses this representation for both classification and generating explanations. The explanations are produced in a contrastive manner, conveying why a datapoint is assigned to one class rather than an alternative class. An explanation is specified as a set of transformations of the input datapoint, with each step depicting a concept changing towards the contrastive class. We build the model using a disentangled VAE, extended with a new supervised method for disentangling individual dimensions. An analysis on synthetic data and MNIST shows that the approaches to both disentanglement and explanation provide benefits over other methods.

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

↑