Captum favicon

Captum

Free
Captum screenshot
Click to visit website
Feature this AI

About

Captum is a PyTorch library for model interpretability. It provides tools for understanding and attributing the predictions of PyTorch models across various modalities like vision and text. It supports a wide range of PyTorch models and is designed to be extensible for interpretability research. Key features include multi-modal support, PyTorch integration, and an open-source design.

Platform
Web
Keywords
aiattributionpytorchmodel interpretabilityinterpretability
Task
model interpreting

Features

built on pytorch

multi-modal support

extensible

FAQs

How do I set the target parameter to an attribution method?

The purpose of target is to select a single (scalar) value for each example in the output of your model to compute attributions based on the given target parameter.

I am facing Out-Of-Memory (OOM) errors when using Captum. How do I resolve this?

To address this issue, you can either reduce n_steps, which may lead to lower-quality approximations, or use the internal_batch_size argument, which allows dividing the expanded input into batches which are processed sequentially.

I am using a perturbation based method, and attributions are taking too long to compute. How can I speed it up?

To enable this, simply set the perturbations_per_eval argument to the desired value.

Are SmoothGrad or VarGrad supported in Captum?

Yes! SmoothGrad and VarGrad are available through NoiseTunnel in Captum, which can be used with any attribution algorithm in Captum.

How do I use Captum with BERT models?

We have a tutorial demonstrating usage of Integrated Gradients on BERT here.

My model inputs or outputs token indices, and when using Captum I see errors relating to gradients, how do I resolve this?

To apply gradient-based attribution methods, it is necessary to replace the embedding layer with an InterpretableEmbedding layer or use LayerIntegratedGradients to compute attribution with respect to the embedding output.

Can my model use functional non-linearities (E.g. nn.functional.ReLU) or can reused modules be used with Captum?

Most methods will work fine with functional non-linearities and arbitrary operations.

Do JIT models, DataParallel models, or DistributedDataParallel models work with Captum?

Yes, we have support for all these model types. Note that JIT models do not yet support hooks, so any methods using hooks including layer and neuron attribution methods, DeepLift, Guided Backprop, and Deconvolution are not supported.

I am working on a new interpretability or attribution method and would like to add it to Captum. How do I proceed?

For interpretability methods created by the community, we have two methods of involvement: Awesome List and Inclusion in Captum.

How can I resolve cudnn RNN backward error for RNN or LSTM network?

To resolve the issue you can set`torch.backends.cudnn.enabled` flag to False - `torch.backends.cudnn.enabled=False`

Pricing Plans

Free
Free Plan

Job Opportunities

There are currently no job postings for this AI tool.

Explore AI Career Opportunities

Social Media

Ratings & Reviews

No ratings available yet. Be the first to rate this tool!

Featured Tools

Songmeaning favicon
Songmeaning

Songmeaning uses AI to reveal the stories and meanings behind song lyrics. It offers lyric translation and AI music generation.

View Details
Whisper Notes favicon
Whisper Notes

Offline AI speech-to-text transcription app using Whisper AI. Supports 80+ languages, audio file import, and offers lifetime access with a one-time purchase. Available for iOS and macOS.

View Details
GitGab favicon
GitGab

Connects Github repos and local files to AI models (ChatGPT, Claude, Gemini) for coding tasks like implementing features, finding bugs, writing docs, and optimization.

View Details
nuptials.ai favicon
nuptials.ai

nuptials.ai is an AI wedding planning partner, offering timeline planning, budget optimization, vendor matching, and a 24/7 planning assistant to help plan your perfect day.

View Details
Make-A-Craft favicon
Make-A-Craft

Make-A-Craft helps you discover craft ideas tailored to your child's age and interests, using materials you already have at home.

View Details
Pixelfox AI favicon
Pixelfox AI

Free online AI photo editor with comprehensive tools for image, face/body, and text. Features include background/object removal, upscaling, face swap, and AI image generation. No sign-up needed, unlimited use for free, fast results.

View Details
Smart Cookie Trivia favicon
Smart Cookie Trivia

Smart Cookie Trivia is a platform offering a wide variety of trivia questions across numerous categories to help users play trivia, explore different topics, and expand their knowledge.

View Details
Code2Docs favicon
Code2Docs

AI-powered code documentation generator. Integrates with GitHub. Automates creation of usage guides, API docs, and testing instructions.

View Details