MUSE: Machine Unlearning Six-Way Evaluation for Language Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Shi, Weijia, Lee, Jaechan, Huang, Yangsibo, Malladi, Sadhika, Zhao, Jieyu, Holtzman, Ari, Liu, Daogao, Zettlemoyer, Luke, Smith, Noah A., Zhang, Chiyuan
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911954769018880
author Shi, Weijia
Lee, Jaechan
Huang, Yangsibo
Malladi, Sadhika
Zhao, Jieyu
Holtzman, Ari
Liu, Daogao
Zettlemoyer, Luke
Smith, Noah A.
Zhang, Chiyuan
author_facet Shi, Weijia
Lee, Jaechan
Huang, Yangsibo
Malladi, Sadhika
Zhao, Jieyu
Holtzman, Ari
Liu, Daogao
Zettlemoyer, Luke
Smith, Noah A.
Zhang, Chiyuan
contents Language models (LMs) are trained on vast amounts of text data, which may include private and copyrighted content. Data owners may request the removal of their data from a trained model due to privacy or copyright concerns. However, exactly unlearning only these datapoints (i.e., retraining with the data removed) is intractable in modern-day models. This has led to the development of many approximate unlearning algorithms. The evaluation of the efficacy of these algorithms has traditionally been narrow in scope, failing to precisely quantify the success and practicality of the algorithm from the perspectives of both the model deployers and the data owners. We address this issue by proposing MUSE, a comprehensive machine unlearning evaluation benchmark that enumerates six diverse desirable properties for unlearned models: (1) no verbatim memorization, (2) no knowledge memorization, (3) no privacy leakage, (4) utility preservation on data not intended for removal, (5) scalability with respect to the size of removal requests, and (6) sustainability over sequential unlearning requests. Using these criteria, we benchmark how effectively eight popular unlearning algorithms on 7B-parameter LMs can unlearn Harry Potter books and news articles. Our results demonstrate that most algorithms can prevent verbatim memorization and knowledge memorization to varying degrees, but only one algorithm does not lead to severe privacy leakage. Furthermore, existing algorithms fail to meet deployer's expectations because they often degrade general model utility and also cannot sustainably accommodate successive unlearning requests or large-scale content removal. Our findings identify key issues with the practicality of existing unlearning algorithms on language models, and we release our benchmark to facilitate further evaluations: muse-bench.github.io
format Preprint
id arxiv_https___arxiv_org_abs_2407_06460
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MUSE: Machine Unlearning Six-Way Evaluation for Language Models
Shi, Weijia
Lee, Jaechan
Huang, Yangsibo
Malladi, Sadhika
Zhao, Jieyu
Holtzman, Ari
Liu, Daogao
Zettlemoyer, Luke
Smith, Noah A.
Zhang, Chiyuan
Computation and Language
Artificial Intelligence
Language models (LMs) are trained on vast amounts of text data, which may include private and copyrighted content. Data owners may request the removal of their data from a trained model due to privacy or copyright concerns. However, exactly unlearning only these datapoints (i.e., retraining with the data removed) is intractable in modern-day models. This has led to the development of many approximate unlearning algorithms. The evaluation of the efficacy of these algorithms has traditionally been narrow in scope, failing to precisely quantify the success and practicality of the algorithm from the perspectives of both the model deployers and the data owners. We address this issue by proposing MUSE, a comprehensive machine unlearning evaluation benchmark that enumerates six diverse desirable properties for unlearned models: (1) no verbatim memorization, (2) no knowledge memorization, (3) no privacy leakage, (4) utility preservation on data not intended for removal, (5) scalability with respect to the size of removal requests, and (6) sustainability over sequential unlearning requests. Using these criteria, we benchmark how effectively eight popular unlearning algorithms on 7B-parameter LMs can unlearn Harry Potter books and news articles. Our results demonstrate that most algorithms can prevent verbatim memorization and knowledge memorization to varying degrees, but only one algorithm does not lead to severe privacy leakage. Furthermore, existing algorithms fail to meet deployer's expectations because they often degrade general model utility and also cannot sustainably accommodate successive unlearning requests or large-scale content removal. Our findings identify key issues with the practicality of existing unlearning algorithms on language models, and we release our benchmark to facilitate further evaluations: muse-bench.github.io
title MUSE: Machine Unlearning Six-Way Evaluation for Language Models
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2407.06460