arXiv:2605.27591cs.LG2026-05中稿 · ICML

用小模型更新向量生成大模型更新,无需共享数据即可提升性能。

Gradient Transformer: Learning to Generate Updates for LLMs

论文配图:Gradient Transformer: Learning to Generate Updates for LLMs
图 1 · 摘自论文原文
  • 通过梯度变换器将小模型更新向量映射为大模型更新向量
  • 在语言建模与推理任务中超越现有知识蒸馏方法
  • 支持多方协作且符合差分隐私,适合资源受限机构

许多组织缺乏计算资源,在私有(不可共享)数据上微调大语言模型(LLMs)以提升实用性,而仅微调小型语言模型(TinyLMs)效果不佳。为此,我们提出一种无需数据的知识蒸馏框架,基于在私有数据上微调的TinyLM生成LLM的更新向量。更新向量是初始模型与其在特定数据集上微调后参数变化的向量,反映微调过程中累积梯度步的影响。该框架的核心是新型梯度变换器(Gradient Transformer),可将TinyLM的更新向量转换为LLM的更新向量。该变换器基于影子数据推导,捕捉了TinyLM与LLM更新向量间的相关性,使第三方可在不访问组织私有数据的情况下生成LLM更新向量。该框架支持多组织协同更新大模型,提升性能并增强成本效率。大量实验表明,即使在严格的差分隐私保护下,Grad-Transformer仍显著优于当前最先进的知识蒸馏基线。

原文摘要 · Abstract (English)

Many organizations lack computational resources to fine-tune large language models (LLMs) on private (unshareable) data for better utility, while fine-tuning tiny language models (TinyLMs) alone performs poorly. To address this bottleneck, we propose a data-free knowledge distillation framework that generates LLM update vectors based on TinyLMs fine-tuned on private data. An update vector is a vector of parameter changes from an initial model to its fine-tuned version on a dataset, capturing the effect of cumulative gradient steps during fine-tuning. The key idea of our framework is a novel Gradient Transformer that transforms TinyLM's update vectors into LLM's update vectors. As derived from shadow datasets, Grad-Transformer captures the correlation between TinyLM and LLM update vectors, enabling third-party providers to generate LLM update vectors given the organization's TinyLM update vectors without accessing the organization's private data. The framework supports multi-organization collaboration to jointly update LLMs, improving performance and cost-efficiency. Extensive experiments across language modeling and reasoning tasks show that Grad-Transformer remarkably outperforms state-of-the-art knowledge distillation baselines, even under strict differential privacy protection.

知识蒸馏大模型微调差分隐私小模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。