From Anchors to Supervision: Memory-Graph Guided Corpus-Free Unlearning for Large Language Models

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

Summary

MAGE, a Memory-grAph Guided Erasure framework, addresses the challenge of unlearning sensitive or copyrighted content in Large Language Models (LLMs) without requiring user-provided forget sets or access to the original training corpus. Traditional machine unlearning methods are difficult to audit and vulnerable to secondary leakage and malicious abuse due to their reliance on explicit forget sets. MAGE operates by taking a lightweight user anchor, which identifies a target entity, and then probes the LLM to recover target-related memorization. This recovered information is organized into a weighted local memory graph, from which scoped supervision for unlearning is synthesized. The framework is model-agnostic and compatible with standard unlearning methods. Experiments on the TOFU and RWKU benchmarks show that MAGE's self-generated supervision achieves unlearning performance comparable to methods using external reference supervision, while maintaining overall model utility.

Key takeaway

For research scientists and CTOs concerned with LLM privacy and compliance, MAGE offers a practical and auditable unlearning workflow. By using minimal anchors instead of full forget corpora, you can mitigate risks of secondary data leakage and malicious abuse while ensuring effective content removal. This approach simplifies unlearning requests and enhances the trustworthiness of your LLM deployments, making compliance easier to demonstrate.

Key insights

MAGE enables corpus-free LLM unlearning using minimal user anchors to synthesize self-generated supervision.

Principles

Method

MAGE probes an LLM with a user anchor to recover memorized content, organizes it into a weighted memory graph, and synthesizes scoped supervision for unlearning.

In practice

Topics

Best for: Research Scientist, CTO, VP of Engineering/Data, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

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