Update README.md

README.md

@@ -5,6 +5,7 @@ base_model:
 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.
 
@@ -13,10 +14,23 @@ 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"/>
 
-You can cite our work using the following bibtex,
+## 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},