Prisma: An Open Source Toolkit for Mechanistic Interpretability in Vision and Video

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Joseph, Sonia, Suresh, Praneet, Hufe, Lorenz, Stevinson, Edward, Graham, Robert, Vadi, Yash, Bzdok, Danilo, Lapuschkin, Sebastian, Sharkey, Lee, Richards, Blake Aaron
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912410642677760
author Joseph, Sonia
Suresh, Praneet
Hufe, Lorenz
Stevinson, Edward
Graham, Robert
Vadi, Yash
Bzdok, Danilo
Lapuschkin, Sebastian
Sharkey, Lee
Richards, Blake Aaron
author_facet Joseph, Sonia
Suresh, Praneet
Hufe, Lorenz
Stevinson, Edward
Graham, Robert
Vadi, Yash
Bzdok, Danilo
Lapuschkin, Sebastian
Sharkey, Lee
Richards, Blake Aaron
contents Robust tooling and publicly available pre-trained models have helped drive recent advances in mechanistic interpretability for language models. However, similar progress in vision mechanistic interpretability has been hindered by the lack of accessible frameworks and pre-trained weights. We present Prisma (Access the codebase here: https://github.com/Prisma-Multimodal/ViT-Prisma), an open-source framework designed to accelerate vision mechanistic interpretability research, providing a unified toolkit for accessing 75+ vision and video transformers; support for sparse autoencoder (SAE), transcoder, and crosscoder training; a suite of 80+ pre-trained SAE weights; activation caching, circuit analysis tools, and visualization tools; and educational resources. Our analysis reveals surprising findings, including that effective vision SAEs can exhibit substantially lower sparsity patterns than language SAEs, and that in some instances, SAE reconstructions can decrease model loss. Prisma enables new research directions for understanding vision model internals while lowering barriers to entry in this emerging field.
format Preprint
id arxiv_https___arxiv_org_abs_2504_19475
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Prisma: An Open Source Toolkit for Mechanistic Interpretability in Vision and Video
Joseph, Sonia
Suresh, Praneet
Hufe, Lorenz
Stevinson, Edward
Graham, Robert
Vadi, Yash
Bzdok, Danilo
Lapuschkin, Sebastian
Sharkey, Lee
Richards, Blake Aaron
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
Robust tooling and publicly available pre-trained models have helped drive recent advances in mechanistic interpretability for language models. However, similar progress in vision mechanistic interpretability has been hindered by the lack of accessible frameworks and pre-trained weights. We present Prisma (Access the codebase here: https://github.com/Prisma-Multimodal/ViT-Prisma), an open-source framework designed to accelerate vision mechanistic interpretability research, providing a unified toolkit for accessing 75+ vision and video transformers; support for sparse autoencoder (SAE), transcoder, and crosscoder training; a suite of 80+ pre-trained SAE weights; activation caching, circuit analysis tools, and visualization tools; and educational resources. Our analysis reveals surprising findings, including that effective vision SAEs can exhibit substantially lower sparsity patterns than language SAEs, and that in some instances, SAE reconstructions can decrease model loss. Prisma enables new research directions for understanding vision model internals while lowering barriers to entry in this emerging field.
title Prisma: An Open Source Toolkit for Mechanistic Interpretability in Vision and Video
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2504.19475