Mechanistic understanding and validation of large AI models with SemanticLens
About
Unlike human-engineered systems such as aeroplanes, where each component's role and dependencies are well understood, the inner workings of AI models remain largely opaque, hindering verifiability and undermining trust. This paper introduces SemanticLens, a universal explanation method for neural networks that maps hidden knowledge encoded by components (e.g., individual neurons) into the semantically structured, multimodal space of a foundation model such as CLIP. In this space, unique operations become possible, including (i) textual search to identify neurons encoding specific concepts, (ii) systematic analysis and comparison of model representations, (iii) automated labelling of neurons and explanation of their functional roles, and (iv) audits to validate decision-making against requirements. Fully scalable and operating without human input, SemanticLens is shown to be effective for debugging and validation, summarizing model knowledge, aligning reasoning with expectations (e.g., adherence to the ABCDE-rule in melanoma classification), and detecting components tied to spurious correlations and their associated training data. By enabling component-level understanding and validation, the proposed approach helps bridge the "trust gap" between AI models and traditional engineered systems. We provide code for SemanticLens on https://github.com/jim-berend/semanticlens and a demo on https://semanticlens.hhi-research-insights.eu.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Neuron Labeling | ImageNet-1K | DMA53.64 | 60 | |
| Neuron Labeling | ResNet101 Neurons (evaluated) | AUC89 | 15 | |
| Neuron Labeling | SAE Vanilla neurons | SCS Score24.48 | 15 | |
| Neuron Labeling | ISIC 2019 | SCS Score20 | 15 | |
| Neuron Labeling | ResNet101 neurons | SCS Score22.75 | 15 | |
| Neuron Labeling | ResNet50 neurons | SCS Score21.96 | 15 | |
| Neuron Labeling | ResNet50 evaluated neurons | AUC84 | 15 | |
| Neuron Labeling | SAE-TopK neurons | SCS Score30.59 | 15 | |
| Neuron Labeling Faithfulness | Evaluated Neurons ResNet50 and SAE-TopK | AUC85 | 15 | |
| Neuron Labeling | SAE-TopK Evaluated Neurons | AUC0.94 | 15 |