Interpreting Neurons in Deep Vision Networks with Language Models

Nicholas Bai, Rahul A. Iyer, Tuomas Oikarinen, Akshay Kulkarni, Tsui-Wei Weng
3/20/2024
cs.CVcs.LG

Official code: Trustworthy-ML-Lab/Describe-and-Dissect (11 stars, last commit 2025-02-20).

Abstract

In this paper, we propose Describe-and-Dissect (DnD), a novel method to describe the roles of hidden neurons in vision networks. DnD utilizes recent advancements in multimodal deep learning to produce complex natural language descriptions, without the need for labeled training data or a predefined set of concepts to choose from. Additionally, DnD is training-free, meaning we don't train any new models and can easily leverage more capable general purpose models in the future. We have conducted extensive qualitative and quantitative analysis to show that DnD outperforms prior work by providing higher quality neuron descriptions. Specifically, our method on average provides the highest quality labels and is more than 2$\times$ as likely to be selected as the best explanation for a neuron than the best baseline. Finally, we present a use case providing critical insights into land cover prediction models for sustainability applications. Our code and data are available at https://github.com/Trustworthy-ML-Lab/Describe-and-Dissect.

View on arXivView PDF

Code Implementations(1)

[TMLR 25] An automated method for explaining complex neuron behaviors in deep vision models using large language models

112Shell, PythonMar 19, 20241 years ago
computer-visiondeep-learningdeep-neural-networksexplainable-aigenerative-ai+3 more

Similar Papers

Cite this paper

@article{bai2024interpreting,
  title  = {Interpreting Neurons in Deep Vision Networks with Language Models},
  author = {Nicholas Bai and Rahul A. Iyer and Tuomas Oikarinen and Akshay Kulkarni and Tsui-Wei Weng},
  year   = {2024},
  eprint = {2403.13771},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.13771v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.13771.svg)](https://codeofpaper.com/paper/2403.13771)

Discussion