Captum logo

Captum

Freemium
訪問

An open source PyTorch library for developers and researchers interpreting model outputs.

什麼是 Captum?

Captum is an open source, extensible library for model interpretability built on PyTorch. It helps developers and researchers examine which input features contribute to a model’s output and investigate how internal layers and neurons affect predictions.

The library supports most types of PyTorch models with minimal modification to the original network. Its algorithms and tutorials cover vision, text, tabular data, and multimodal models, providing a practical foundation for troubleshooting predictions and researching interpretability methods.

Captum 能做什麼?

Input attribution algorithms

Apply Integrated Gradients, DeepLIFT, GradientShap, Occlusion, LRP, or LIME to examine how input features contribute to a selected model output.

Layer and neuron analysis

Use conductance and internal attribution methods to investigate layer and neuron importance, including the input features associated with a particular neuron.

LLM and multimodal attribution

Interpret text generation with LLM attribution functionality. For multimodal LLMs, combine ImageMaskInput, image segmentation, and FeatureAblation to examine image regions contributing to generated text.

Concept and robustness analysis

Explore concepts with Testing with Concept Activation Vectors (TCAV). Tutorials also demonstrate FGSM and PGD attacks and robustness metrics alongside attribution algorithms.

Extensible research workflows

Implement new interpretability algorithms and benchmark them against methods available in the library. Tutorials demonstrate distributed attribution using torch.distributed and DataParallel.

使用情境

“Troubleshoot classification predictions”

Inspect influential words in sentiment predictions or image regions in vision models to investigate unexpected outputs.

“Analyze tabular regression models”

Evaluate feature importance and internal attribution in a house price regression network to understand how the network uses its inputs.

“Explain recommendation behavior”

Examine sparse and dense feature contributions in Deep Learning Recommender Models, including feature interactions involved in clicked and non-clicked ad predictions.

“Investigate question answering models”

Study BERT embeddings, attention layers, and token interactions to understand a particular answer prediction.

常見問題

How do I install Captum?

Install with `conda install captum -c pytorch`, the website’s recommended method, or use `pip install captum`.

What knowledge do I need to get started?

The tutorials assume familiarity with PyTorch and its basic features. The introductory Captum tutorial trains a neural network on Titanic survival data, then examines input, layer, and neuron importance.

What does an Integrated Gradients workflow return?

The homepage example passes input and baseline tensors to an IntegratedGradients instance, selects a target output, and returns attribution scores. With return_convergence_delta=True, it also returns a convergence delta.

快速資訊

Product type
Open source model interpretability library
Framework
PyTorch
Primary users
Model developers, interpretability researchers, and application engineers
Website
https://captum.ai

Captum 流量分析

流量數據僅供參考。

月造訪量
1.2萬
全球排名
#1,891,534
使用者跳出率
36.5%
平均造訪時長
01:17
每次造訪頁數
2.30
網域評分
64

流量趨勢

流量來源

尚無流量分析數據。

主要造訪地區

Captum 替代品

Clips AI logo

Clips AI

clipsai.com

An open-source Python library that helps developers turn longform video into clips and reframed formats.

Harness Manager logo

Harness Manager

harnessmanager.devmesh.xyz

Harness Manager is a free, open-source native Mac workspace for managing AI coding tools, MCP servers, skills, updates, and model comparisons. It helps developers inspect their local setup, discover additions, review updates, and compare models in one app.

Payload logo

Payload

payloadcms.com

An open-source, TypeScript-based Next.js CMS and application framework for developers and content teams.

Cortex Docs logo

Cortex Docs

cortexdocs.dev

Cortex Docs is an open-source API tooling workflow that turns API specifications and Markdown into interactive documentation, typed SDKs, and MCP servers. It helps developers publish API interfaces for human users, applications, and AI agents from a shared project.