Hugging Face Blog 发布于 08/10 18:05

Making Knowledge Distillation Cheap Enough to Run at Scale

(翻译)让知识蒸馏成本足够低以实现大规模运行

查看原文

资讯摘要

A Blog post by Multiverse Computing on Hugging Face

AI 摘要

Multiverse Computing 提出两种系统优化,使大模型知识蒸馏可大规模运行:离线缓存教师模型前100个logits,避免训练时同时加载师生模型;融合分块KL散度损失,按块计算并丢弃中间结果,峰值显存显著下降。在32K上下文下显存减少15.6倍,256K上下文仍可运行,并将GPT-OSS 20B蒸馏从四个节点缩减至一个。

AI 问答

这项技术如何降低知识蒸馏的显存占用?

文章提出两种方法:离线缓存教师模型每个位置最可能的前100个logits,训练时不再同时加载教师模型;以及融合分块KL损失,按序列块端到端计算投影和损失,不保留完整词汇×序列网格,反向传播时按需重算,峰值显存随序列长度线性增长。

离线蒸馏与在线蒸馏的效果有差距吗?

实验显示,四种设置在8K上下文下训练损失几乎重合,说明用缓存的前100个logits做离线蒸馏相对于在线蒸馏几乎无损。

该方法在实际大规模蒸馏中有何效果?

在32K上下文蒸馏GPT-OSS 20B时,显存释放使配置从四个GPU节点缩减到一个,单步时间从57秒降至12.23秒,单卡吞吐从74.2提升至345.7 TFLOP/s。

打开原文

更多资讯

这届 AI PC,太烧心了爱范儿 / APPSO · 10/10 16:28特朗普政府要求AI公司在发生安全事件后立即上报36氪 · 10/10 09:19阿斯麦将对韩国零部件涨价10%36氪 · 10/10 12:23外交部副部长马朝旭同澳大利亚外交贸易部秘书长安思捷举行中澳外交部政治磋商36氪 · 10/10 11:10JEPA、空间智能路线正面交锋:WorldArena 2.0 挑战赛开考「能不能被机器人真正用起来」| IROS 2026雷峰网 · 10/10 13:30报道称OpenAI和Anthropic正在聘用前特朗普政府官员36氪 · 10/09 19:08美债失控狂飙:冲击到底有多大?金色财经 · 10/09 21:25冷钱包遭经销商投毒 超8600万美元被盗 请立刻排查金色财经 · 10/10 14:56芯片专家试图劝退马斯克:Terafab芯片项目可比发射星舰火箭更难36氪 · 10/10 17:49阿里开始给AI算账了:用AI的钱,以后各业务线自己出人人都是产品经理 · 10/10 11:09当代码不再稀缺:大型工程 AI 狂欢下的确定性底座|QCon上海InfoQ 中文 · 10/10 18:00Nano Banana 2.1 深度实测:4K 直出、中文排版大进步,连续改图的坑也要看清开源中国 · 10/10 14:20名创优品:中国内地第三季度总GMV同比增长25-30%36氪 · 10/09 20:14智能咖啡机 10 天内产生 1TB 数据流量Solidot · 10/09 17:35GMIF 2026:日均140万亿Token调用,正在改变SSD的能力与分工雷峰网 · 10/10 09:38同样是珍珠,为啥价格能差10倍?果壳网 · 10/10 22:15成年人要的「安心」,有时是一支笔给的36氪 · 10/10 15:30微信聊天记录终于能正经交给 AI 了,官方开的口子不怕封号人人都是产品经理 · 10/10 11:25OPC企业容易陷入的三个增长误区人人都是产品经理 · 10/10 15:27我让豆包工作看了30个爆款AI封面,它总结出了八个规律人人都是产品经理 · 10/10 13:0310后中学生捧红统一“呆毛”发型,这回轮到老非主流们看不惯了果壳网 · 10/10 17:15美图披露AI创新产品进展:Picchi ARR达111万美元36氪 · 10/10 12:343.76亿人集体抛弃,保健品跌出银发消费前三,精准营养成“香饽饽”果壳网 · 10/10 16:15北京:增加人工智能手机和电脑、智能机器人等新一代智能终端产品有效供给36氪 · 10/10 14:13Claude 动态多 Agent 工作流开始公测,支持庞大的工作负载开源中国 · 10/10 18:218600万美元被盗 官方授权也不能保证硬件钱包安全?金色财经 · 10/10 19:44华尔街新高藏隐忧金色财经 · 10/09 22:27小红书+AI+知识库,我现在无敌了人人都是产品经理 · 10/10 10:45商务部欧洲司负责人解读中欧贸易投资磋商机制第二次例会成果36氪 · 10/10 13:35Impactful scheduling for GPU clustersHugging Face Blog · 10/09 23:20