English

Understanding the Inner Workings of Language Models Through Representation Dissimilarity

Machine Learning 2023-10-24 v1 Artificial Intelligence Computation and Language

Abstract

As language models are applied to an increasing number of real-world applications, understanding their inner workings has become an important issue in model trust, interpretability, and transparency. In this work we show that representation dissimilarity measures, which are functions that measure the extent to which two model's internal representations differ, can be a valuable tool for gaining insight into the mechanics of language models. Among our insights are: (i) an apparent asymmetry in the internal representations of model using SoLU and GeLU activation functions, (ii) evidence that dissimilarity measures can identify and locate generalization properties of models that are invisible via in-distribution test set performance, and (iii) new evaluations of how language model features vary as width and depth are increased. Our results suggest that dissimilarity measures are a promising set of tools for shedding light on the inner workings of language models.

Keywords

Cite

@article{arxiv.2310.14993,
  title  = {Understanding the Inner Workings of Language Models Through Representation Dissimilarity},
  author = {Davis Brown and Charles Godfrey and Nicholas Konz and Jonathan Tu and Henry Kvinge},
  journal= {arXiv preprint arXiv:2310.14993},
  year   = {2023}
}

Comments

EMNLP 2023 (main)