GRAM Paper: Precisely Removing LLM Dangerous Capabilities via Gradient-Routed Modules

burkov · x · 2026-07-30

A recent study introduces Gradient-Routed Auxiliary Modules (GRAM) to address the safety dilemma where dangerous knowledge (e.g., bioweapons) and beneficial knowledge (e.g., vaccine research) coexist in LLMs.

Core Mechanism

Researchers add small auxiliary modules to Transformers whose parameters receive backpropagation updates only for selected data types. This encourages specialized knowledge to settle in a removable module rather than spreading through the network. Turning off the module at inference time reduces the corresponding capability while leaving general performance largely intact.

Experiments

Experiments spanned virology, cybersecurity, nuclear physics, and specialized code, testing models from 50M to 5B parameters. GRAM outperformed traditional data filtering and model unlearning methods in precision.

Original post →

More from Safety

Safety channel →