Backward Lens: Projecting Language Model Gradients into the Vocabulary Space
Fuente:
arXiv
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866909112413978624 |
|---|---|
| author | Katz, Shahar Belinkov, Yonatan Geva, Mor Wolf, Lior |
| author_facet | Katz, Shahar Belinkov, Yonatan Geva, Mor Wolf, Lior |
| contents | Understanding how Transformer-based Language Models (LMs) learn and recall information is a key goal of the deep learning community. Recent interpretability methods project weights and hidden states obtained from the forward pass to the models' vocabularies, helping to uncover how information flows within LMs. In this work, we extend this methodology to LMs' backward pass and gradients. We first prove that a gradient matrix can be cast as a low-rank linear combination of its forward and backward passes' inputs. We then develop methods to project these gradients into vocabulary items and explore the mechanics of how new information is stored in the LMs' neurons. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2402_12865 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Backward Lens: Projecting Language Model Gradients into the Vocabulary Space Katz, Shahar Belinkov, Yonatan Geva, Mor Wolf, Lior Computation and Language Artificial Intelligence Machine Learning Understanding how Transformer-based Language Models (LMs) learn and recall information is a key goal of the deep learning community. Recent interpretability methods project weights and hidden states obtained from the forward pass to the models' vocabularies, helping to uncover how information flows within LMs. In this work, we extend this methodology to LMs' backward pass and gradients. We first prove that a gradient matrix can be cast as a low-rank linear combination of its forward and backward passes' inputs. We then develop methods to project these gradients into vocabulary items and explore the mechanics of how new information is stored in the LMs' neurons. |
| title | Backward Lens: Projecting Language Model Gradients into the Vocabulary Space |
| topic | Computation and Language Artificial Intelligence Machine Learning |
| url | https://arxiv.org/abs/2402.12865 |