Search papers, labs, and topics across Lattice.
This paper introduces Gradient Deviation Score (GDS), a novel method for pre-training data detection in LLMs that leverages the observation that training samples transition from unfamiliar to familiar, reflected in gradient behavior. GDS represents samples using gradient profiles capturing magnitude, location, and concentration of parameter updates across FFN and Attention modules, and then uses a lightweight classifier for membership inference. Experiments on five datasets demonstrate GDS achieves state-of-the-art performance and improved cross-dataset transferability compared to likelihood-based and fine-tuning-based baselines.
Forget likelihood scores and fine-tuning tricks: LLMs leak pre-training data membership through subtle, yet detectable, deviations in their gradient updates.
Pre-training data detection for LLMs is essential for addressing copyright concerns and mitigating benchmark contamination. Existing methods mainly focus on the likelihood-based statistical features or heuristic signals before and after fine-tuning, but the former are susceptible to word frequency bias in corpora, and the latter strongly depend on the similarity of fine-tuning data. From an optimization perspective, we observe that during training, samples transition from unfamiliar to familiar in a manner reflected by systematic differences in gradient behavior. Familiar samples exhibit smaller update magnitudes, distinct update locations in model components, and more sharply activated neurons. Based on this insight, we propose GDS, a method that identifies pre-training data by probing Gradient Deviation Scores of target samples. Specifically, we first represent each sample using gradient profiles that capture the magnitude, location, and concentration of parameter updates across FFN and Attention modules, revealing consistent distinctions between member and non-member data. These features are then fed into a lightweight classifier to perform binary membership inference. Experiments on five public datasets show that GDS achieves state-of-the-art performance with significantly improved cross-dataset transferability over strong baselines. Further interpretability analyse show gradient feature distribution differences, enabling practical and scalable pre-training data detection.
