ElemedexA pocket encyclopedia of AI architectures

Glossary (TM) › Interpretability and evaluation

TM182

Probing

Also: diagnostic classifiers, probing classifiers

In Spanish: Sondeo (probing)

A technique that trains an auxiliary classifier on a model's internal activations to find out what information (syntactic, semantic, about the world) can be read off them.

A probe is a small model, almost always linear, trained to predict some property (a word's part of speech, its depth in the syntax tree, the state of a game board) from the vectors a model produces at some layer. If the probe does well above chance, the information is taken to be encoded there (Millière & Buckner, 2024b, sec. 2.3).

Using this technique, Tenney et al. (2019) found that BERT runs, layer by layer, through something like the classical natural-language-processing pipeline: parts of speech first, then syntax, named entities, semantic roles and, last, coreference. Hewitt & Manning (2019) went further with a "structural probe": they found a linear transformation under which the distance between two words' contextual embeddings approximates their distance in the parse tree, in both ELMo and BERT.

Two classic caveats apply. First, a powerful probe may learn the task by itself. Hewitt & Liang (2019) therefore proposed control tasks, which assign random labels to words: a good probe should be selective, succeeding on the linguistic task and failing on the control. Second, the fact that information can be decoded does not show that the model uses it. Millière & Buckner (2024b) illustrate this with a pond: throw in wooden letters spelling a word, and with the right apparatus you might recover the word from the ripples, yet nobody would say the pond represents it (2024b, sec. 2.3). Getting from correlation to cause requires interventions: modify the direction the probe found and check whether behavior changes as predicted. That is how the case for a world model in Othello-GPT was strengthened (Li et al., 2023; Millière & Buckner, 2024b, sec. 2.5.3).

The distinction matters for philosophy of mind. Theories of representation usually demand three things: that the state carry information, that it influence behavior, and that it be capable of misrepresenting (Millière & Buckner, 2024b, sec. 2.4). Probing alone establishes only the first.

Sources

  • Hewitt, J., & Liang, P. (2019). Designing and interpreting probes with control tasks. In K. Inui, J. Jiang, V. Ng, & X. Wan (Eds.), Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP) (pp. 2733–2743). Association for Computational Linguistics. https://doi.org/10.18653/v1/D19-1275 §
  • Hewitt, J., & Manning, C. D. (2019). A structural probe for finding syntax in word representations. In J. Burstein, C. Doran, & T. Solorio (Eds.), Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers) (pp. 4129–4138). Association for Computational Linguistics. https://doi.org/10.18653/v1/N19-1419 §
  • Li, K., Hopkins, A. K., Bau, D., Viégas, F., Pfister, H., & Wattenberg, M. (2023). Emergent world representations: Exploring a sequence model trained on a synthetic task. In 11th International Conference on Learning Representations (ICLR 2023). https://openreview.net/forum?id=DeG07_TcZvT §
  • Millière, R., & Buckner, C. (2024b). A philosophical introduction to language models—Part II: The way forward (arXiv:2405.03207). arXiv. https://doi.org/10.48550/arXiv.2405.03207 §
  • Tenney, I., Das, D., & Pavlick, E. (2019). BERT rediscovers the classical NLP pipeline. In A. Korhonen, D. Traum, & L. Màrquez (Eds.), Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics (pp. 4593–4601). Association for Computational Linguistics. https://doi.org/10.18653/v1/P19-1452 §