English

VICE: Variational Interpretable Concept Embeddings

Machine Learning 2022-10-07 v8 Applications Machine Learning

Abstract

A central goal in the cognitive sciences is the development of numerical models for mental representations of object concepts. This paper introduces Variational Interpretable Concept Embeddings (VICE), an approximate Bayesian method for embedding object concepts in a vector space using data collected from humans in a triplet odd-one-out task. VICE uses variational inference to obtain sparse, non-negative representations of object concepts with uncertainty estimates for the embedding values. These estimates are used to automatically select the dimensions that best explain the data. We derive a PAC learning bound for VICE that can be used to estimate generalization performance or determine a sufficient sample size for experimental design. VICE rivals or outperforms its predecessor, SPoSE, at predicting human behavior in the triplet odd-one-out task. Furthermore, VICE's object representations are more reproducible and consistent across random initializations, highlighting the unique advantage of using VICE for deriving interpretable embeddings from human behavior.

Keywords

Cite

@article{arxiv.2205.00756,
  title  = {VICE: Variational Interpretable Concept Embeddings},
  author = {Lukas Muttenthaler and Charles Y. Zheng and Patrick McClure and Robert A. Vandermeulen and Martin N. Hebart and Francisco Pereira},
  journal= {arXiv preprint arXiv:2205.00756},
  year   = {2022}
}

Comments

Accepted at NeurIPS 2022

R2 v1 2026-06-24T11:04:28.616Z