subirmansukhani commited on
Commit
ec8bdfc
·
1 Parent(s): f5961bc

Add interpretability roadmap to README

Browse files

Outline five complementary methods beyond attention maps: gradient
attribution, occlusion sensitivity, representation probing, causal
tracing, and temporal consistency analysis.

Files changed (1) hide show
  1. README.md +12 -0
README.md CHANGED
@@ -262,6 +262,18 @@ smolvla-inspect/
262
 
263
  ---
264
 
 
 
 
 
 
 
 
 
 
 
 
 
265
  ## Note on FFmpeg
266
 
267
  If you installed `ffmpeg@6` and linked it (`brew link --overwrite ffmpeg@6`), your default `ffmpeg` is now 6.x. To switch back later: `brew unlink ffmpeg@6 && brew link ffmpeg`.
 
262
 
263
  ---
264
 
265
+ ## Roadmap
266
+
267
+ Attention maps show where the model allocates compute, but not whether those regions actually drive the output. The following interpretability methods would complement the current tooling:
268
+
269
+ - [ ] **Gradient-based attribution** -- compute `d(action) / d(patch_embedding)` via vanilla saliency, GradCAM, or Integrated Gradients to measure which image patches *causally influence* the predicted action (not just where attention points)
270
+ - [ ] **Occlusion / perturbation sensitivity** -- mask out image regions or zero out specific prefix tokens (vision, language, state) and measure action MSE change; model-agnostic and directly answers "if I cover the gripper, does the model break?"
271
+ - [ ] **Representation probing** -- train small linear classifiers on intermediate layer representations to test what information is encoded at each stage (e.g., can layer N predict object position? does the Expert encode gripper state?)
272
+ - [ ] **Causal tracing / activation patching** -- replace activations at specific (layer, token) positions with corrupted versions and measure output change; builds a causal map of information flow through the model
273
+ - [ ] **Temporal consistency analysis** -- track attention patterns across frames in an episode to check if attention follows the object smoothly, whether cross-attention shifts predict upcoming actions, and correlation between attention movement and action direction
274
+
275
+ ---
276
+
277
  ## Note on FFmpeg
278
 
279
  If you installed `ffmpeg@6` and linked it (`brew link --overwrite ffmpeg@6`), your default `ffmpeg` is now 6.x. To switch back later: `brew unlink ffmpeg@6 && brew link ffmpeg`.