Demystifying Embedding Spaces Using Large Language Models

Abstract

Embeddings have become a pivotal means to represent complex, multi-faceted information about entities, concepts, and relationships in a condensed and useful format. Nevertheless, they often preclude direct interpretation. While downstream tasks make use of these compressed representations, meaningful interpretation usually requires visualization using dimensionality reduction or specialized machine learning interpretability methods. This paper addresses the challenge of making such embeddings more interpretable and broadly useful, by employing large language models (LLMs) to directly interact with embeddings -- transforming abstract vectors into understandable narratives. By injecting embeddings into LLMs, we enable querying and exploration of complex embedding data. We demonstrate our approach on a variety of diverse tasks, including: enhancing concept activation vectors (CAVs), communicating novel embedded entities, and decoding user preferences in recommender systems. Our work couples the immense information potential of embeddings with the interpretative power of LLMs.

Cite

Text

Tennenholtz et al. "Demystifying Embedding Spaces Using Large Language Models." International Conference on Learning Representations, 2024.

Markdown

[Tennenholtz et al. "Demystifying Embedding Spaces Using Large Language Models." International Conference on Learning Representations, 2024.](https://mlanthology.org/iclr/2024/tennenholtz2024iclr-demystifying/)

BibTeX

@inproceedings{tennenholtz2024iclr-demystifying,
  title     = {{Demystifying Embedding Spaces Using Large Language Models}},
  author    = {Tennenholtz, Guy and Chow, Yinlam and Hsu, ChihWei and Jeong, Jihwan and Shani, Lior and Tulepbergenov, Azamat and Ramachandran, Deepak and Mladenov, Martin and Boutilier, Craig},
  booktitle = {International Conference on Learning Representations},
  year      = {2024},
  url       = {https://mlanthology.org/iclr/2024/tennenholtz2024iclr-demystifying/}
}