Axiomatic Attribution is a framework that evaluates the validity of feature importance explanations by testing their compliance with a set of mathematically defined, non-negotiable axioms. Rather than relying on subjective visual assessment of a saliency map, this approach demands that an attribution method satisfy formal properties such as Completeness, Sensitivity-n, and Implementation Invariance to be considered a faithful explanation of a model's decision.
Glossary
Axiomatic Attribution

What is Axiomatic Attribution?
A rigorous mathematical framework for evaluating and designing feature attribution methods based on their adherence to a set of fundamental axioms.
The framework, formalized by Sundararajan et al., uniquely identifies Integrated Gradients as the only path method that satisfies these core axioms. By establishing a mathematical baseline for correctness, axiomatic attribution provides a principled way to reject flawed techniques like simple Gradient × Input that suffer from gradient saturation, ensuring that the assigned feature importance scores provide a true and complete decomposition of the model's output.
Key Characteristics of Axiomatic Methods
Axiomatic attribution establishes a rigorous mathematical framework for evaluating feature importance methods. By defining desirable properties as formal axioms, it provides a principled way to compare techniques and identify those that produce faithful, consistent explanations.
Completeness Axiom
The sum of all feature attributions must exactly equal the difference between the model's output for the input and a baseline reference output. This ensures the explanation accounts for the entire prediction without missing or double-counting contributions.
- Guarantees a full decomposition of the output
- Satisfied by Integrated Gradients and DeepLIFT
- Violated by simple Gradient × Input methods
Sensitivity-n Axiom
If a model's output is mathematically independent of a specific feature, that feature must receive an attribution of exactly zero. This prevents spurious importance from being assigned to irrelevant inputs.
- Formalizes the 'null player' property from game theory
- Ensures features with zero impact get zero credit
- Critical for auditing models in regulated industries
Implementation Invariance
Two functionally equivalent neural networks must produce identical attributions for the same input, regardless of their internal architecture or parameterization. This axiom prevents the explanation from depending on arbitrary implementation details.
- The explanation reflects the function, not the form
- Violated by methods relying on internal activations
- A key differentiator between principled and heuristic methods
Linearity Axiom
For a model that is a linear combination of two sub-models, the attribution must equal the same linear combination of the individual attributions. This preserves the additive structure of the explanation space.
- Ensures consistency across ensemble models
- Maintains proportionality in composite systems
- Fundamental to the Shapley value framework
Symmetry Axiom
Two features that contribute identically to every possible subset of inputs must receive equal attribution scores. This enforces fairness in the distribution of importance among interchangeable features.
- Prevents arbitrary bias in feature ranking
- Derived directly from cooperative game theory
- Essential for equitable model auditing
Path Methods Unification
A class of attribution techniques that define importance by integrating gradients along a specified path from a baseline to the target input. Integrated Gradients is the canonical example using a straight-line path.
- Different paths yield different attribution distributions
- The choice of baseline encodes prior expectations
- Provides a continuum between local and global explanations
Axiomatic vs. Heuristic Attribution Methods
A comparison of feature attribution methods based on their adherence to formal mathematical axioms versus empirical heuristics.
| Feature | Integrated Gradients | DeepLIFT | Gradient × Input |
|---|---|---|---|
Satisfies Completeness Axiom | |||
Satisfies Sensitivity-n Axiom | |||
Satisfies Implementation Invariance | |||
Requires Baseline Reference | |||
Handles Gradient Saturation | |||
Computational Cost | High (50-300 steps) | Medium (single pass) | Low (single pass) |
Attribution Sign Consistency | Preserves sign | May flip signs | Preserves sign |
Frequently Asked Questions
Explore the mathematical foundations that define what makes a feature attribution method trustworthy. These axioms provide the rigorous framework for evaluating and comparing explanation techniques in deep learning.
Axiomatic Attribution is a formal framework for evaluating feature attribution methods based on their adherence to a set of mathematically defined axioms, such as completeness, sensitivity-n, and implementation invariance. Rather than judging explanations by subjective visual appeal, this approach establishes a rigorous, theoretical foundation for determining whether an attribution method faithfully reflects a model's internal reasoning. The framework was crystallized in the 2017 paper 'Axiomatic Attribution for Deep Networks' by Sundararajan, Taly, and Yan, which demonstrated that many popular gradient-based methods fail these fundamental tests. By defining these necessary conditions, the framework allows engineers to systematically compare methods like Integrated Gradients, DeepLIFT, and Gradient × Input, ensuring that the chosen explanation technique provides a truthful decomposition of the model's output rather than a visually pleasing but mathematically flawed artifact.
Enabling Efficiency, Speed & Accuracy
Intelligent Analysis, Decision & Execution
We build AI systems for teams that need search across company data, workflow automation across tools, or AI features inside products and internal software.
Talk to Us
Search across company data
Give teams answers from docs, tickets, runbooks, and product data with sources and permissions.
Useful when people spend too long searching or get different answers from different systems.

Automate internal workflows
Use AI to route work, draft outputs, trigger actions, and keep approvals and logs in place.
Useful when repetitive work moves across multiple tools and teams.

Add AI to products and internal tools
Build assistants, guided actions, or decision support into the software your team or customers already use.
Useful when AI needs to be part of the product, not a separate tool.
Related Terms
Explore the foundational axioms and key methods that define the rigorous mathematical framework for evaluating and computing feature attributions in deep learning models.
Completeness Axiom
The principle that the sum of all feature attributions must equal the difference between the model's output for the target input and a baseline reference. This ensures a conservation of relevance—no importance is created or destroyed during the decomposition. Integrated Gradients and DeepLIFT are explicitly designed to satisfy this axiom, making them suitable for high-stakes auditing where total accountability is required.
Implementation Invariance
This axiom demands that two functionally identical networks produce identical attributions, regardless of their internal architecture. If Model A and Model B always output the same value for every input, their explanations must match. This is a critical test for path methods like Integrated Gradients, which depend only on the function's input-output behavior, not the specific weights or layers used to achieve it.
Sensitivity-n
A strict requirement: if a model's output is mathematically independent of a specific feature, that feature must receive an attribution of exactly zero. This prevents irrelevant features from being assigned spurious importance. Gradient × Input methods can violate this axiom when the gradient is zero but the input value is non-zero, highlighting a key failure mode in simple saliency approaches.
Integrated Gradients
A canonical path method that satisfies Completeness, Sensitivity-n, and Implementation Invariance. It computes importance by accumulating gradients along a straight-line path from a baseline (e.g., a black image) to the actual input. The integral captures how each feature incrementally contributes to the prediction change, providing a theoretically sound and widely adopted attribution baseline.
Gradient SHAP
An approximation of Shapley values that leverages the expected gradients of a model with respect to a background distribution. By adding Gaussian noise to inputs for computational efficiency, it bridges game theory and gradient-based sensitivity. This method inherits the strong theoretical guarantees of SHAP while scaling to deep networks, though it sacrifices strict axiomatic purity for speed.
Infidelity Measure
A quantitative metric for evaluating attribution faithfulness. It measures the error between the actual model output change and the dot product of the attribution vector with a meaningful perturbation. A low infidelity score indicates the explanation accurately predicts the model's local behavior. This provides a practical tool for benchmarking methods like FullGrad against the theoretical ideals of axiomatic attribution.

About the author
Prasad Kumkar
CEO & MD, Inference Systems
Prasad Kumkar is the CEO & MD of Inference Systems and writes about AI systems architecture, LLM infrastructure, model serving, evaluation, and production deployment. Over 5+ years, he has worked across computer vision models, L5 autonomous vehicle systems, and LLM research, with a focus on taking complex AI ideas into real-world engineering systems.
His work and writing cover AI systems, large language models, AI agents, multimodal systems, autonomous systems, inference optimization, RAG, evaluation, and production AI engineering.
Partnered with leading AI, data, and software stack.
How We Work
Custom AI workflows for your Business
One-fit-all AI don't work for modern businesses. At Inferensys, we aim to understand your business & custom requirements; which we use to define most efficient agentic workflows, the data, and the tools for your business.
01
Review the use case
We understand the task, the users, and where AI can actually help.
Read more02
Pick the right approach
We define what needs search, automation, or product integration.
Read more03
Build the first useful version
We implement the part that proves the value first.
Read more04
Improve from there
We add the checks and visibility needed to keep it useful.
Read moreThe first call is a practical review of your use case and the right next step.
Talk to Us