用知识蒸馏思想实现自监督预训练的数据集压缩,显著提升小样本场景下的模型性能。
Dataset Distillation via Knowledge Distillation: Towards Efficient Self-Supervised Pre-Training of Deep Networks
- 通过知识蒸馏匹配自监督教师模型的训练轨迹,生成高质量小规模合成数据集。
- 在有限标注数据下,下游任务准确率比现有方法最高提升13%。
- 适用于资源受限场景下的高效自监督模型预训练,尤其适合小样本学习。
数据集压缩(DD)通过生成小型合成数据集,能在有限内存和计算资源下高效训练深度网络。尽管监督学习中的DD方法已取得成功,但针对自监督预训练(SSL)的DD仍缺乏有效方案。利用无标签数据进行预训练对在少量标注数据下高效泛化至关重要。本文首次提出适用于SSL预训练的有效DD方法。首先,理论与实证证明:直接将监督学习的DD方法应用于SSL会因自监督梯度方差过高而失败。为此,我们借鉴知识蒸馏(KD)思想,训练一个小学生模型以匹配经由SSL训练的大教师模型的表示。随后,通过匹配学生模型的训练轨迹生成小型合成数据集。由于KD目标函数方差远低于SSL,该方法能生成可成功预训练高质量编码器的合成数据集。大量实验表明,在多种下游任务中,本方法在有限标注数据条件下,性能相比之前工作最高提升13%。代码见https://github.com/BigML-CS-UCLA/MKDT。
原文摘要 · Abstract (English)
Dataset distillation (DD) generates small synthetic datasets that can efficiently train deep networks with a limited amount of memory and compute. Despite the success of DD methods for supervised learning, DD for self-supervised pre-training of deep models has remained unaddressed. Pre-training on unlabeled data is crucial for efficiently generalizing to downstream tasks with limited labeled data. In this work, we propose the first effective DD method for SSL pre-training. First, we show, theoretically and empirically, that naive application of supervised DD methods to SSL fails, due to the high variance of the SSL gradient. Then, we address this issue by relying on insights from knowledge distillation (KD) literature. Specifically, we train a small student model to match the representations of a larger teacher model trained with SSL. Then, we generate a small synthetic dataset by matching the training trajectories of the student models. As the KD objective has considerably lower variance than SSL, our approach can generate synthetic datasets that can successfully pre-train high-quality encoders. Through extensive experiments, we show that our distilled sets lead to up to 13% higher accuracy than prior work, on a variety of downstream tasks, in the presence of limited labeled data. Code at https://github.com/BigML-CS-UCLA/MKDT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。