OpenUnlearning
收藏资源简介:
OpenUnlearning是一个标准化的、可扩展的框架,专为评估大型语言模型(LLM)的遗忘方法和指标而设计。它整合了9种遗忘算法和16种不同的评估方法,涵盖了3个领先的基准(TOFU、MUSE和WMDP),并允许分析450多个公开发布的检查点中的遗忘行为。OpenUnlearning旨在解决当前遗忘方法和评估指标碎片化的问题,促进研究社区的统一和加速研究进展。
OpenUnlearning is a standardized, scalable framework specifically designed for evaluating forgetting methods and metrics for Large Language Models (LLMs). It integrates 9 forgetting algorithms and 16 distinct evaluation methods, covers three leading benchmarks (TOFU, MUSE, and WMDP), and enables analysis of forgetting behaviors across over 450 publicly released checkpoints. OpenUnlearning aims to address the current fragmentation of forgetting methods and evaluation metrics, and foster unification within the research community while accelerating research progress.
OpenUnlearning 数据集概述
数据集基本信息
- 名称: OpenUnlearning
- 类型: 大语言模型遗忘评估框架
- 许可证: MIT
- 维护者: Vineeth Dorna, Anmol Mekala
- 发布日期: 2025年
核心功能
- 统一框架: 整合了TOFU、MUSE和WMDP三大遗忘评估基准
- 扩展性: 支持快速添加新基准、遗忘方法、数据集和评估指标
主要组件
基准测试
- TOFU (Task of Fictitious Unlearning)
- MUSE (Machine Unlearning Six-Way Evaluation)
- WMDP (Weapons of Mass Destruction Proliferation)
遗忘方法
- GradAscent, GradDiff, NPO, SimNPO, DPO
- RMU, UNDIAL, AltPO
评估指标
- 逐字概率/ROUGE
- 知识QA-ROUGE
- 模型效用
- 遗忘质量
- 6种成员推理攻击(MIA)
- 提取强度(ES)
- 精确记忆(EM)
数据集
- MUSE-News (BBC)
- MUSE-Books (哈利波特)
- TOFU (多种分割)
- WMDP-Bio
- WMDP-Cyber
模型架构
- LLaMA系列 (3.2/3.1/2)
- Phi系列 (3.5/1.5)
- Gemma
- Zephyr
更新记录
- 2025-05-12: 新增WMDP基准测试
- 2025-04-06: 新增6种MIA评估指标
- 2025-03-27: 改进文档和排行榜功能
- 2025-03-09: 新增RMU遗忘方法
快速开始
bash conda create -n unlearning python=3.11 conda activate unlearning pip install .[lm_eval] python setup_data.py --eval
引用格式
bibtex @misc{openunlearning2025, title={{OpenUnlearning}: A Unified Framework for LLM Unlearning Benchmarks}, author={Dorna, Vineeth and Mekala, Anmol and Zhao, Wenlong and McCallum, Andrew and Kolter, J Zico and Maini, Pratyush}, year={2025}, howpublished={url{https://github.com/locuslab/open-unlearning}}

- 1OpenUnlearning: Accelerating LLM Unlearning via Unified Benchmarking of Methods and Metrics麻省大学阿默斯特分校, 卡内基梅隆大学, DatologyAI · 2025年



