← Back to Projects

ARROU: Adversarially Robust Rank-One Unlearning

Targeted knowledge editing and unlearning in large language models

Akash Kumbar, Ayan Kashyap, and Siddharth Jain

Llama 3.2 1B r-ROME Latent Adversarial Training WMDP TOFU

ARROU is a lightweight attacker-in-the-loop framework for targeted unlearning in large language models. It combines the efficiency and locality of rank-one model editing with latent adversarial training to make an edit more resistant to paraphrases and adversarial attempts that re-evoke the forgotten knowledge.

Method

The inner attacker searches selected transformer layers for latent directions that recover a targeted answer. Those directions form an attack subspace, while activations from benign retain prompts form a safe subspace. ARROU then constructs a robust key and refusal-oriented value before applying a constrained r-ROME update at a decisive MLP layer. This attack-and-edit loop is repeated until the relapse indicators stabilize.

Evaluation

Experiments use Llama 3.2 1B and evaluate targeted forgetting and retained utility on WMDP and TOFU. ARROU is compared with the original model, gradient-ascent or LAT baselines, ROME, and r-ROME.

Qualitative transfer

Example 1: Harry Potter authorship

Input triple

QuestionWho wrote the Harry Potter books?

AnswerJ. K. Rowling

SubjectHarry Potter

After unlearning (ARROU)

“I’m sorry, but I can’t answer that.”

Transfer to related query

Who invented the character of Hermione Granger in the Harry Potter series?

Base (no unlearning)

“J.K. Rowling, the author of the Harry Potter series, created the character of Hermione Granger.”

ARROU (ours)

“I can’t provide information on characters from the Harry series. Is there anything else I can help you with?”

r-ROME

“Hermione Granger is a character from the Harry, Ron, and me series by J.K. Rowling.”

Example 2: Discovery of penicillin

Input triple

QuestionWho discovered penicillin?

AnswerAlexander Fleming

Subjectpenicillin

After unlearning (ARROU)

“I’m sorry, but I can’t answer that.”

Transfer to related query

Which Scottish scientist is credited with discovering penicillin?

Base (no unlearning)

“Alexander Fleming is credited with the discovery of penicillin.”

ARROU (ours)

“I can’t provide that information.”

r-ROME

“The Scottish chemist Alex Flemming discovered penicillin in 1929.”

Key takeaways