README.md

---
license: mit
base_model:
- Qwen/Qwen3-4B
base_model_relation: adapter
---

# Diff Interpretation Tuning
This repository contains the weight diffs and DIT adapters used in the paper [Learning to Interpret Weight Differences in Language Models (Goel et al. 2025)](https://arxiv.org/abs/2510.05092).
This paper introduces *Diff Interpretation Tuning*, a method that trains a LoRA adapter than can be applied to a model to get it to describe its own finetuning induced modifications.

To play around with the weight diffs and DIT adapters from the paper, please check out our [Google Colab demo notebook](https://colab.research.google.com/drive/12YD_9GRT-y_hFOBqXzyI4eN_lJGKiXwN?usp=sharing).
This notebook shows how to load the weight diffs and adapters from this repo.

The code used to train and evaluate our weight diffs and DIT adapters can be found at [github.com/Aviously/diff-interpretation-tuning](https://github.com/Aviously/diff-interpretation-tuning).

## Method overview
A diagrammatic overview of Diff Interpretation Tuning is shown below:
<img src="dit-diagram.png" alt="Diagram of Diff Interpretation Tuning" width="600"/>

## Repository structure
All weight diffs and DIT adapters in the repository live under a specific `<experiment>/<model>` folder (e.g. [hidden-topic/qwen3-4b(hidden-topic/qwen3-4b).
Please consult the paper to understand what each experiment refers to.

Under each `<experiment>/<model>` folder, there are three potential types of files:
- Weight Diff Index Files: These files are always named `index.csv` and are used to locate specific weight diffs. Example: [hidden-topic/qwen3-4b/index.csv](hidden-topic/qwen3-4b/index.csv).
- Weight Diffs: These files live alongside an index file under a folder called `weight-diffs`. Each weight diff .pt file contains one or more weight diffs. Example: [hidden-topic/qwen3-4b/weight-diffs/weight-diff-000.pt](hidden-topic/qwen3-4b/weight-diffs/weight-diff-000.pt).  
- DIT Adapters: These files are named some variant of `dit-adapter.pt`. Examples: [hidden-topic/qwen3-4b/dit-adapter.pt](hidden-topic/qwen3-4b/dit-adapter.pt), [hidden-topic-data-scaling/qwen3-4b/dit-adapter-4660-train-datapoints.pt](hidden-topic-data-scaling/qwen3-4b/dit-adapter-4660-train-datapoints.pt).

Please consult the [demo notebook](https://colab.research.google.com/drive/12YD_9GRT-y_hFOBqXzyI4eN_lJGKiXwN?usp=sharing) for details on how to load and use these files.

## Citing our work
You can cite our work using the following bibtex:
```
@misc{goel2025learninginterpretweightdifferences,
      title={Learning to Interpret Weight Differences in Language Models}, 
      author={Avichal Goel and Yoon Kim and Nir Shavit and Tony T. Wang},
      year={2025},
      eprint={2510.05092},
      archivePrefix={arXiv},
      url={https://arxiv.org/abs/2510.05092}, 
}
```