Insights into a radiology-specialised multimodal large language model with sparse autoencoders

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bouzid, Kenza, Bannur, Shruthi, Meissen, Felix, de Castro, Daniel Coelho, Schwaighofer, Anton, Alvarez-Valle, Javier, Hyland, Stephanie L.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912490283073536
author Bouzid, Kenza
Bannur, Shruthi
Meissen, Felix
de Castro, Daniel Coelho
Schwaighofer, Anton
Alvarez-Valle, Javier
Hyland, Stephanie L.
author_facet Bouzid, Kenza
Bannur, Shruthi
Meissen, Felix
de Castro, Daniel Coelho
Schwaighofer, Anton
Alvarez-Valle, Javier
Hyland, Stephanie L.
contents Interpretability can improve the safety, transparency and trust of AI models, which is especially important in healthcare applications where decisions often carry significant consequences. Mechanistic interpretability, particularly through the use of sparse autoencoders (SAEs), offers a promising approach for uncovering human-interpretable features within large transformer-based models. In this study, we apply Matryoshka-SAE to the radiology-specialised multimodal large language model, MAIRA-2, to interpret its internal representations. Using large-scale automated interpretability of the SAE features, we identify a range of clinically relevant concepts - including medical devices (e.g., line and tube placements, pacemaker presence), pathologies such as pleural effusion and cardiomegaly, longitudinal changes and textual features. We further examine the influence of these features on model behaviour through steering, demonstrating directional control over generations with mixed success. Our results reveal practical and methodological challenges, yet they offer initial insights into the internal concepts learned by MAIRA-2 - marking a step toward deeper mechanistic understanding and interpretability of a radiology-adapted multimodal large language model, and paving the way for improved model transparency. We release the trained SAEs and interpretations: https://huggingface.co/microsoft/maira-2-sae.
format Preprint
id arxiv_https___arxiv_org_abs_2507_12950
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Insights into a radiology-specialised multimodal large language model with sparse autoencoders
Bouzid, Kenza
Bannur, Shruthi
Meissen, Felix
de Castro, Daniel Coelho
Schwaighofer, Anton
Alvarez-Valle, Javier
Hyland, Stephanie L.
Machine Learning
Interpretability can improve the safety, transparency and trust of AI models, which is especially important in healthcare applications where decisions often carry significant consequences. Mechanistic interpretability, particularly through the use of sparse autoencoders (SAEs), offers a promising approach for uncovering human-interpretable features within large transformer-based models. In this study, we apply Matryoshka-SAE to the radiology-specialised multimodal large language model, MAIRA-2, to interpret its internal representations. Using large-scale automated interpretability of the SAE features, we identify a range of clinically relevant concepts - including medical devices (e.g., line and tube placements, pacemaker presence), pathologies such as pleural effusion and cardiomegaly, longitudinal changes and textual features. We further examine the influence of these features on model behaviour through steering, demonstrating directional control over generations with mixed success. Our results reveal practical and methodological challenges, yet they offer initial insights into the internal concepts learned by MAIRA-2 - marking a step toward deeper mechanistic understanding and interpretability of a radiology-adapted multimodal large language model, and paving the way for improved model transparency. We release the trained SAEs and interpretations: https://huggingface.co/microsoft/maira-2-sae.
title Insights into a radiology-specialised multimodal large language model with sparse autoencoders
topic Machine Learning
url https://arxiv.org/abs/2507.12950