The Remarkable Robustness of LLMs: Stages of Inference?

We find that deleting and swapping interventions retain 72-95% of the original model’s prediction accuracy without fine-tuning, and hypothesize the existence of four universal stages of inference across eight different models.

<span title='2024-06-27 00:00:00 +0000 UTC'>June 2024</span>&nbsp;&middot;&nbsp;Vedang Lad, Jin Hwa Lee, Wes Gurnee, Max Tegmark

Mechanistic Interpretability for Progress Towards Quantitative AI Safety

MIT Master’s thesis studying mechanistic interpretability as a path toward quantitative AI safety.

<span title='2024-05-01 00:00:00 +0000 UTC'>May 2024</span>&nbsp;&middot;&nbsp;Vedang Lad

Opening the AI Black Box: Distilling Machine-Learned Algorithms into Code

We present MIPS, a novel method for program synthesis based on automated mechanistic interpretability of neural networks trained to perform the desired task, auto-distilling the learned algorithm into Python code.

<span title='2024-02-01 00:00:00 +0000 UTC'>February 2024</span>&nbsp;&middot;&nbsp;Eric J. Michaud, Isaac Liao, Vedang Lad, Ziming Liu, Anish Mudide, Chloe Loughridge, Zifan Carl Guo, Tara Rezaei Kheirkhah, Mateja Vukelić, Max Tegmark