arXiv Open Access 2025

Hubble: a Model Suite to Advance the Study of LLM Memorization

Johnny Tian-Zheng Wei Ameya Godbole Mohammad Aflah Khan Ryan Wang Xiaoyuan Zhu +5 lainnya
Lihat Sumber

Abstrak

We present Hubble, a suite of fully open-source large language models (LLMs) for the scientific study of LLM memorization. Hubble models come in standard and perturbed variants: standard models are pretrained on a large English corpus, and perturbed models are trained in the same way but with controlled insertion of text (e.g., book passages, biographies, and test sets) designed to emulate key memorization risks. Our core release includes 8 models -- standard and perturbed models with 1B or 8B parameters, pretrained on 100B or 500B tokens -- establishing that memorization risks are determined by the frequency of sensitive data relative to size of the training corpus (i.e., a password appearing once in a smaller corpus is memorized better than the same password in a larger corpus). Our release also includes 6 perturbed models with text inserted at different pretraining phases, showing that sensitive data without continued exposure can be forgotten. These findings suggest two best practices for addressing memorization risks: to dilute sensitive data by increasing the size of the training corpus, and to order sensitive data to appear earlier in training. Beyond these general empirical findings, Hubble enables a broad range of memorization research; for example, analyzing the biographies reveals how readily different types of private information are memorized. We also demonstrate that the randomized insertions in Hubble make it an ideal testbed for membership inference and machine unlearning, and invite the community to further explore, benchmark, and build upon our work.

Topik & Kata Kunci

Penulis (10)

J

Johnny Tian-Zheng Wei

A

Ameya Godbole

M

Mohammad Aflah Khan

R

Ryan Wang

X

Xiaoyuan Zhu

J

James Flemings

N

Nitya Kashyap

K

Krishna P. Gummadi

W

Willie Neiswanger

R

Robin Jia

Format Sitasi

Wei, J.T., Godbole, A., Khan, M.A., Wang, R., Zhu, X., Flemings, J. et al. (2025). Hubble: a Model Suite to Advance the Study of LLM Memorization. https://arxiv.org/abs/2510.19811

Akses Cepat

Lihat di Sumber
Informasi Jurnal
Tahun Terbit
2025
Bahasa
en
Sumber Database
arXiv
Akses
Open Access ✓