Pith. sign in

Gradient routing: Masking gradients to localize computation in neural networks

6 Pith papers cite this work. Polarity classification is still indexing.

6 Pith papers citing it
abstract

Neural networks are trained primarily based on their inputs and outputs, without regard for their internal mechanisms. These neglected mechanisms determine properties that are critical for safety, like (i) transparency; (ii) the absence of sensitive information or harmful capabilities; and (iii) reliable generalization of goals beyond the training distribution. To address this shortcoming, we introduce gradient routing, a training method that isolates capabilities to specific subregions of a neural network. Gradient routing applies data-dependent, weighted masks to gradients during backpropagation. These masks are supplied by the user in order to configure which parameters are updated by which data points. We show that gradient routing can be used to (1) learn representations which are partitioned in an interpretable way; (2) enable robust unlearning via ablation of a pre-specified network subregion; and (3) achieve scalable oversight of a reinforcement learner by localizing modules responsible for different behaviors. Throughout, we find that gradient routing localizes capabilities even when applied to a limited, ad-hoc subset of the data. We conclude that the approach holds promise for challenging, real-world applications where quality data are scarce.

citation-role summary

method 1

citation-polarity summary

fields

cs.LG 5 cs.CL 1

years

2026 6

roles

method 1

polarities

use method 1

representative citing papers

Modular Pretraining Enables Access Control

cs.LG · 2026-07-09 · conditional · novelty 7.0

GRAM selectively trains auxiliary modules so that ablating one at inference removes a targeted capability while preserving the rest, closely tracking data-filtered models at 5x lower cost across 5 capability profiles.

Prototype Language Models

cs.LG · 2026-07-01 · unverdicted · novelty 6.0

PRISM forms predictions as sparse mixtures of learned prototypes trained with clustering objectives, matching dense model accuracy while enabling ~500x faster data attribution and behavior editing without finetuning.

citing papers explorer

Showing 6 of 6 citing papers.