2024

LM Transparency Tool: Interactive Tool for Analyzing Transformer Language Models

Tufanov, Igor, Hambardzumyan, Karen, Ferrando, Javier et al.

Understand

We present the LM Transparency Tool (LM-TT), an open-source interactive toolkit for analyzing the internal workings of Transformer-based language models.

  • Differently from previously existing tools that focus on isolated parts of the decision-making process, our framework is designed to make the entire prediction process transparent, and allows tracing back model behavior from the top-layer representation to very fine-grained parts of the model.
  • Specifically, it (1) shows the important part of the whole input-to-output information flow, (2) allows attributing any changes done by a model block to individual attention heads and feed-forward neurons, (3) allows interpreting the functions of those heads or neurons.
  • A crucial part of this pipeline is showing the importance of specific model components at each step.

Reading the bibliography…