mlsanityData to 14 Oct 2026

Dataset distillation for memorized data: Soft labels can leak held-out teacher knowledge

arXiv 2506.14457 ↗First posted 17 Jun 2025cs.LGcited by 10

Not on the map. Why?

Abstract (arXiv)

Dataset distillation aims to compress training data into fewer examples via a teacher, from which a student can learn effectively. While its success is often attributed to structure in the data, modern neural networks also memorize specific facts, but if and how such memorized information is can transferred in distillation settings remains less understood. In this work, we show that students trained on soft labels from teachers can achieve non-trivial accuracy on held-out memorized data they never directly observed. This effect persists on structured data when the teacher has not generalized.To analyze it in isolation, we consider finite random i.i.d. datasets where generalization is a priori impossible and a successful teacher fit implies pure memorization. Still, students can learn non-trivial information about the held-out data, in some cases up to perfect accuracy. In those settings, enough soft labels are available to recover the teacher functionally - the student matches the teacher's predictions on all possible inputs, including the held-out memorized data. We show that these phenomena strongly depend on the temperature with which the logits are smoothed, but persist across varying network capacities, architectures and dataset compositions.

Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Builds on 22 of its 31 references that are in the map. First posted on arXiv 17 Jun 2025.

Builds on

22 of 31 references.

References · 22

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  3. 2016Understanding deep learning requires rethinking generalizationcited by 1,674
  4. 2020Knowledge Distillation: A Surveycited by 1,603
  5. 2019When Does Label Smoothing Help?cited by 988
  6. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  7. 2024A Survey on Knowledge Distillation of Large Language Modelscited by 284
  8. 2021Towards Understanding Knowledge Distillationcited by 158
  9. 2023Dataset Distillation: A Comprehensive Reviewcited by 127
  10. 2023Generalizing Dataset Distillation via Deep Generative Priorcited by 126
  11. 2021Rethinking Soft Labels for Knowledge Distillation: A Bias-Variance Tradeoff Perspectivecited by 119
  12. 2024Scaling Laws for Fact Memorization of Large Language Modelscited by 33
10 more
  1. 2024A Label is Worth a Thousand Images in Dataset Distillationcited by 31
  2. 2022Quantifying the Knowledge in a DNN to Explain Knowledge Distillation for Classificationcited by 19
  3. 2024The Pitfalls of Memorization: When Memorization Hurts Generalizationcited by 19
  4. 2020Effectiveness of Arbitrary Transfer Sets for Data-free Knowledge Distillationcited by 18
  5. 2020Solvable Model for Inheriting the Regularization through Knowledge Distillationcited by 16
  6. 2024Towards a theory of model distillationcited by 15
  7. 2024What is Dataset Distillation Learning?cited by 13
  8. 2024Quantifying Knowledge Distillation Using Partial Information Decompositioncited by 13
  9. 2022Measures of Information Reflect Memorization Patternscited by 12
  10. 2024A Multi-Perspective Analysis of Memorization in Large Language Modelscited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 9.

10 papers.