MemHunter: Automated and Verifiable Memorization Detection at Dataset-scale in LLMs

Zhenpeng Wu, Jian Lou, Zibin Zheng, Chuan Chen
12/10/2024
cs.CRcs.LG

No confident code implementation found for this paper.

Abstract

Large language models (LLMs) have been shown to memorize and reproduce content from their training data, raising significant privacy concerns, especially with web-scale datasets. Existing methods for detecting memorization are primarily sample-specific, relying on manually crafted or discretely optimized memory-inducing prompts generated on a per-sample basis, which become impractical for dataset-level detection due to the prohibitive computational cost of iterating through all samples. In real-world scenarios, data owners may need to verify whether a susceptible LLM has memorized their dataset, particularly if the LLM may have collected the data from the web without authorization. To address this, we introduce MemHunter, which trains a memory-inducing LLM and employs hypothesis testing to efficiently detect memorization at the dataset level, without requiring sample-specific memory inducing. Experiments on models like Pythia and Llama demonstrate that MemHunter can extract up to 40% more training data than existing methods under constrained time resources and reduce search time by up to 80% when integrated as a plug-in. Crucially, MemHunter is the first method capable of dataset-level memorization detection, providing a critical tool for assessing privacy risks in LLMs powered by large-scale datasets.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{wu2024memhunter,
  title  = {MemHunter: Automated and Verifiable Memorization Detection at Dataset-scale in LLMs},
  author = {Zhenpeng Wu and Jian Lou and Zibin Zheng and Chuan Chen},
  year   = {2024},
  eprint = {2412.07261},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.07261v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2412.07261.svg)](https://codeofpaper.com/paper/2412.07261)

Discussion