Inferensys

Glossary

Attribution Flow

A theoretical framework that views the propagation of relevance scores through a network as a continuous flow, satisfying conservation laws and providing a unified view of decomposition methods.
Stylish WeWork-like workspace with hot desks and document wall, professional searching through enterprise knowledge base on a mounted ultrawide display, warm industrial pendants overhead.
UNIFIED DECOMPOSITION FRAMEWORK

What is Attribution Flow?

A theoretical framework that views the propagation of relevance scores through a network as a continuous flow, satisfying conservation laws and providing a unified view of decomposition methods.

Attribution Flow is a theoretical framework that conceptualizes the propagation of relevance scores through a neural network as a continuous, conservative flow field. It posits that a model's output can be decomposed by tracking how a quantity—analogous to a fluid—moves from the output layer back to the input features, satisfying a strict conservation law akin to Kirchhoff's current law.

This framework unifies disparate feature attribution methods, such as Layer-wise Relevance Propagation and Integrated Gradients, by demonstrating they are specific discretizations of a continuous flow process. By enforcing the completeness axiom at every layer, Attribution Flow ensures no relevance is created or destroyed, providing a mathematically rigorous foundation for auditing how information aggregates to form a final prediction.

CONSERVATION LAWS IN NEURAL NETWORKS

Core Principles of Attribution Flow

Attribution Flow reframes feature importance as a conserved quantity propagating through a network, providing a unified mathematical lens for comparing decomposition methods.

01

The Conservation Principle

Attribution Flow is governed by a strict conservation law: the total relevance entering a neuron must equal the total relevance exiting it. This is not a heuristic—it's a mathematical constraint derived from the network's topology.

  • Kirchhoff's Analogy: Just as current is conserved at a junction in a circuit, relevance is conserved at each neuron.
  • Completeness Guarantee: Conservation at every layer ensures the sum of input attributions equals the model's output difference from a baseline.
  • Unified Framework: Methods like Layer-wise Relevance Propagation (LRP), Deep Taylor Decomposition, and Excitation Backpropagation are all specific instances of flow with different propagation rules.
02

Flow Decomposition vs. Gradient Methods

Attribution Flow fundamentally differs from gradient-based methods by redistributing the output value itself, not the local sensitivity.

  • Value Redistribution: Flow methods decompose the output score ( f(x) ) into contributions from each input feature, satisfying the Completeness Axiom by construction.
  • Gradient Limitation: Raw gradients measure infinitesimal change, not contribution. A saturated feature with zero gradient may be critical to the prediction—flow methods correctly assign it high relevance.
  • Path Independence: Unlike Integrated Gradients, which requires choosing an integration path, flow methods propagate through the network's actual computational graph, avoiding the baseline selection problem.
03

Propagation Rules as Design Choices

The specific rule for redistributing relevance at each layer defines the flavor of Attribution Flow. These rules encode assumptions about how information mixes.

  • LRP-0: Redistributes relevance proportionally to positive weighted activations. Simple but can produce noisy attributions.
  • LRP-ε: Adds a small stabilization term ε to the denominator, absorbing weak or contradictory relevance and improving signal-to-noise ratio.
  • LRP-αβ: Separates positive and negative contributions with weights α and β. Setting α=2, β=1 emphasizes excitatory pathways and suppresses inhibitory noise.
  • Deep Taylor Decomposition: Derives propagation rules by linearizing the neuron's activation function at a root point, providing a theoretical justification for LRP rules.
04

Deep Taylor Decomposition Foundation

Deep Taylor Decomposition provides the rigorous mathematical backbone for many flow-based attribution methods by applying Taylor expansion at a functionally relevant root point.

  • Root Point Selection: For a ReLU neuron, the root point is chosen where the neuron's output is zero and minimally distant from the actual activation, ensuring the decomposition is local and meaningful.
  • First-Order Redistribution: The function value is redistributed to inputs using the partial derivatives at the root point, naturally yielding the LRP-α1β0 rule for ReLU networks.
  • Handling Non-Linearities: Each activation function (ReLU, tanh, softmax) requires a tailored root point strategy, making the framework extensible to arbitrary architectures.
05

Attribution Flow in Practice

Flow-based methods excel in domains requiring high-resolution, trustworthy explanations where gradient noise is unacceptable.

  • Medical Imaging: LRP heatmaps on MRI and CT scans provide radiologists with pixel-level evidence that respects the image's spatial structure, unlike noisier gradient saliency maps.
  • Natural Language Processing: Relevance scores assigned to individual tokens reveal which words drove a sentiment classification or entailment decision, enabling debugging of transformer models.
  • Regulatory Compliance: The built-in conservation property provides a complete audit trail—every unit of the output decision is accounted for by a specific input feature, satisfying GDPR right to explanation requirements.
06

Unifying Decomposition Methods

Attribution Flow serves as a meta-framework that reveals the structural relationships between seemingly disparate explanation techniques.

  • LRP as Flow: LRP is explicitly a flow method with hand-designed propagation rules per layer type.
  • Excitation Backpropagation: Equivalent to a flow where only excitatory (positive) relevance is propagated, corresponding to an LRP variant with α=1, β=0.
  • Gradient × Input Connection: Under specific conditions (linear activations, zero biases), Gradient × Input approximates a flow decomposition, explaining why it sometimes satisfies completeness.
  • SHAP Relationship: Deep SHAP can be interpreted as a flow method that uses expected gradients over a background distribution, bridging game-theoretic and conservation-based approaches.
ATTRIBUTION FLOW

Frequently Asked Questions

Explore the theoretical framework that unifies feature attribution methods by treating relevance propagation as a conservative physical flow through a neural network.

Attribution Flow is a theoretical framework that conceptualizes the propagation of relevance scores through a neural network as a continuous, conservative physical flow. Instead of viewing feature attribution as a set of discrete, disconnected calculations, it models the process as a dynamic system where a finite quantity of 'relevance' enters at the output layer and flows backward through the network's connections to the input features. This framework is governed by a conservation law, meaning the total relevance distributed to the input features must exactly equal the model's output decision score. By satisfying this completeness axiom, Attribution Flow provides a unified mathematical lens through which many disparate decomposition methods—such as Layer-wise Relevance Propagation (LRP), DeepLIFT, and Integrated Gradients—can be understood as specific discretizations or approximations of the same underlying continuous flow process.

Prasad Kumkar

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.