探索用自动提示优化检测大模型对推荐数据的记忆泄漏。
Exploring Approaches for Detecting Memorization of Recommender System Data in Large Language Models
- 用自动提示工程替代人工设计,提升记忆数据提取效率。
- 自动提示法能部分找回物品级信息,但难以还原评分等数值数据。
- 现有方法对非文本数据检测能力弱,自动化是未来方向。
大型语言模型(LLMs)因强大的自然语言理解与生成能力被广泛用于推荐场景,但其训练数据来源不公开,引发数据泄露担忧。已有研究发现LLaMA和OpenAI模型家族会记忆MovieLens-1M数据集,但提取方式依赖人工提示工程。本文提出三个核心问题:能否改进人工提示?能否通过非人工方法检测记忆?能否实现自动化检测?为此,我们评估了三种方法:(i) 漏洞利用提示工程;(ii) 无监督潜在知识发现,通过对比一致搜索(CCS)和聚类归一化探测内部激活;(iii) 自动提示工程(APE),将提示发现建模为元学习过程,迭代优化候选指令。在使用LLaMA模型的MovieLens-1M实验中,漏洞提示未提升记忆项检索效果且表现不一致;CCS可可靠区分真实与虚构电影标题,但在用户和评分等数值数据上失败;APE在物品级信息提取上表现中等,但难以恢复交互数值。结果表明,自动优化提示是最具前景的记忆样本提取策略。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are increasingly applied in recommendation scenarios due to their strong natural language understanding and generation capabilities. However, they are trained on vast corpora whose contents are not publicly disclosed, raising concerns about data leakage. Recent work has shown that the MovieLens-1M dataset is memorized by both the LLaMA and OpenAI model families, but the extraction of such memorized data has so far relied exclusively on manual prompt engineering. In this paper, we pose three main questions: Is it possible to enhance manual prompting? Can LLM memorization be detected through methods beyond manual prompting? And can the detection of data leakage be automated? To address these questions, we evaluate three approaches: (i) jailbreak prompt engineering; (ii) unsupervised latent knowledge discovery, probing internal activations via Contrast-Consistent Search (CCS) and Cluster-Norm; and (iii) Automatic Prompt Engineering (APE), which frames prompt discovery as a meta-learning process that iteratively refines candidate instructions. Experiments on MovieLens-1M using LLaMA models show that jailbreak prompting does not improve the retrieval of memorized items and remains inconsistent; CCS reliably distinguishes genuine from fabricated movie titles but fails on numerical user and rating data; and APE retrieves item-level information with moderate success yet struggles to recover numerical interactions. These findings suggest that automatically optimizing prompts is the most promising strategy for extracting memorized samples.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。