闭式轨迹权重手术:无需数十亿 token 即可将 4B 能力迁移至 0.8B(为什么中间层会崩溃,以及 4 个锚定层如何解决问题)

Reddit r/LocalLLaMA 论文

摘要

研究人员提出一种免反向传播的闭式轨迹匹配方法,将 4B 模型的能力迁移到 0.8B 学生模型。他们发现将权重手术限制在 4 个锚定层可以避免 NLL 爆炸,在 30 个基准测试上将留出集 NLL 降低 10.8%,且整个过程仅需一块 8GB 的 RX 580 显卡即可完成。

标准的知识蒸馏通常意味着燃烧数周的算力和数十亿 token,只期望学生模型最终能模仿教师。我们想看看如果完全跳过反向传播、把迁移当作层与层之间的闭式轨迹匹配问题来处理会发生什么。思路很直接:将一小批校准提示词分别输入两个模型,捕获层到层的隐藏状态轨迹,然后通过带正则化的最小二乘和谱投影直接求解学生模型 MLP 块中的权重更新。我们在两种架构上进行了测试:Qwen 3.5(从 4B 迁移到 0.8B)以及老款 GPT-2 small——只是想看看它会不会像通常那样一碰权重就瞬间崩坏成乱码。两者都保持了连贯性,但最初的 Qwen 测试碰到了一堵墙:编辑 Qwen 0.8B 的全部 24 层直接把模型"熔化"了(NLL 损失爆炸 +64.78%)。当我们检查整个网络的奇异值熵时,发现第 1 到 22 层实际上是一锅混乱的多义性"大杂烩",熵值超过 0.90。如果试图将原始轨迹强行穿过这些中间层,基本就等于把模型内部的记忆"结"全打乱了。解决办法是将手术限制在 4 个锚定点(第 0、7、15、23 层),这些层的表示真正保持了干净的线性结构。这样做之后:在 30 个多样化基准测试上,留出集 NLL 下降了 10.8%(生物医学领域 -23.8%,数学与逻辑 -14.6%)。零样本 400 题 HellaSwag 从 54.75% 提升到 55.25%(+0.50%),并通过 Vulkan llama.cpp 在本地完成验证。基础版 0.8B 原本在二叉树反转任务上会输出一堆死寂的注释伪代码,完全失败;而编辑后的检查点一次就写出了干净的递归 Python。在俄语逻辑悖论上,它甚至开始自发地生成 推理标签——在一个完全没有应用任何对话模板的原始基座模型上看到这种情况,真的非常震撼。最重要的是:我们没有 H100 集群,甚至没有一张 4090。所有的轨迹提取和权重求解都是在一块老旧的 8GB RX 580 上本地完成的,通过逐层 VRAM 流式传输和一个 DirectML 补丁来阻止 Qwen 的 Gated DeltaNet 注意力机制抛出驱动错误。一切都已开源,方便你们检视或复现:代码与数学推导:https://github.com/dsadawq3/DynamicTune 迁移后的基座模型权重:https://huggingface.co/F-Labs/Qwen3.5-0.8B-DynamicTune-Base 如果在座有人拥有 24GB-32GB 显存的显卡(4090、5090 或服务器级 GPU),并想进一步推进这项工作,以下方向值得关注:将 27B 模型的推理轨迹移植到 9B、4B 或 2B 模型上;将更大的模型(如 Gemma)压缩到移动端尺寸而无需数周的重训练;直接从未审查的模型中移植拒绝消融向量,无需微调;利用预训练的稀疏自编码器(SAE)解开第 1-22 层的"结",从而不必跳过这些层。欢迎在评论区提问或一起探讨失败模式。
查看原文

相似文章

一个0.8B模型在ARC-Challenge上击败了2B模型(42.15%):闭式权重手术在零反向传播下胜过多GPU SFT(已在NVIDIA L4上独立验证)

Reddit r/LocalLLaMA

DynamicTune通过闭式线性代数,在约12分钟内将推理能力从更大的教师模型迁移到0.8B模型,全程零反向传播,在ARC-Challenge上达到42.15%的acc_norm,超过了标准2B模型(41.10%)。FP16权重、GGUF文件和GitHub仓库均已开源,并在一块NVIDIA L4上完成了独立基准测试。

Show HN:在 4 GB 笔记本 GPU 上微调 8B 模型

Hacker News Top

Soup 是一个开源命令行工具,通过单条命令简化 LLM 的微调和后训练,支持基于 QLoRA 的训练,通过逐层流式传输仅需 4 GB 显存即可在消费级 GPU 上运行。最新版本新增了 DPO、ORPO、SimPO 和 KTO 等偏好损失,且不会使内存需求翻倍。

稀疏设计(5分钟阅读)

TLDR AI

Moonshot的Kimi K3是一个2.8万亿参数的开源权重模型,拥有896个专家(每个token激活16个),体现了在保持活跃计算不变的情况下扩大总参数的趋势,并使用注意力压缩来减少KV缓存大小,使得前沿推理更容易实现,但存储成本较高。

超曲面约束动态权重更新实验 [P]

Reddit r/MachineLearning

一个实验性的语言模型架构使用超曲面进行动态权重更新以减少训练参数,仅使用16%的参数即可实现比未展开基线更好的性能。该方法在FineWeb-Edu数据集上进行了测试,并包含一个GitHub实现。