Anthropic research introduces GRAM for isolating dangerous AI knowledge

· Source: Dataconomy · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Cybersecurity & Data Privacy · Depth: Advanced, quick

Summary

Anthropic and AE Studio introduced Gradient-Routed Auxiliary Modules (GRAM), a method for isolating dangerous knowledge within AI models into removable modules without affecting overall performance. GRAM integrates small auxiliary neural compartments into a language model's MLP layers, each focusing on specific sensitive categories like virology or cybersecurity. During training, only the module corresponding to dual-use data, which made up approximately 0.25% of training data for each domain, is active. Researchers tested GRAM on models from 50 million to 5 billion parameters, including an 800-million-parameter model. Results showed that removing GRAM modules eliminated specific capabilities nearly as effectively as if the model had never been trained on that data, while general performance remained close to baseline. GRAM proved robust against adversarial fine-tuning, offering a potential compromise for AI policy by enabling selective access control, though it is preliminary and not yet applied to production models, with scalability and entangled capability separation remaining challenges.

Key takeaway

For AI Governance Policy Makers and AI Security Engineers grappling with dual-use AI risks, Anthropic's GRAM method presents a novel approach to granularly control sensitive knowledge. You can consider advocating for or implementing systems that allow selective module removal, enabling tailored model deployments for specific contexts like vetted research labs. This could mitigate national security concerns more effectively than broad restrictions, but be aware of current scalability limitations for production models.

Key insights

GRAM isolates dangerous AI knowledge into removable modules, maintaining general performance and offering selective access control.

Principles

Method

GRAM extends transformer MLP layers with auxiliary modules. During training, only the module corresponding to dual-use data is active, allowing for later activation or deletion.

In practice

Topics

Best for: Research Scientist, CTO, VP of Engineering/Data, AI Scientist, AI Security Engineer, Policy Maker

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Dataconomy.