← 最新论文
💻 computer science

Automatic Pruning Discovery for Large Language Models

本文介绍了AutoPrune,这是一个新颖的框架,它利用大语言模型本身,通过图驱动的思维链推理和偏度感知的动态稀疏性分配,自动设计最优剪枝算法,从而在无需专家知识的情况下,有效缓解大语言模型中由异常值引起的性能退化。

原作者: Haidong Kang, Lihong Lin, Enneng Yang, Hongning Dai, Hao Wang

发布于 2026-05-28
📖 1 分钟阅读☕ 轻松阅读

原作者: Haidong Kang, Lihong Lin, Enneng Yang, Hongning Dai, Hao Wang

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

以下是用通俗易懂的语言和生动的类比对论文《大型语言模型的自动剪枝发现》进行的解释。

核心难题:臃肿的巨象

想象大型语言模型(LLMs)就像巨象。这些巨象极其聪明,能完成令人惊叹的任务,比如编写代码或翻译语言。但它们体型过于庞大,需要巨大的、昂贵的体育场(服务器)来栖息,并消耗海量的食物(电力)来维持运转。

为了让这些巨象能住进更小、更便宜的帐篷(比如你的手机或小型笔记本电脑),科学家们尝试对它们进行“剪枝”。剪枝就像修剪大象的毛发或去除多余的脂肪。目标是在让大象变小的同时,不使其忘记如何表演特技。

旧方法:专家理发师

迄今为止,修剪这些巨象的工作一直由专家理发师(人类研究人员)完成。

  • 过程:理发师必须凭借多年的经验和试错来猜测该剪哪些毛发。他们会剪掉一点,检查大象是否还能行走,再剪多一点,如此重复数千次。
  • 缺陷:这耗费了巨大的时间,成本高昂,且需要一位大师级理发师。如果理发师犯错,大象可能会跌倒或踉跄。
  • “异常值”问题:论文发现了一个具体问题:大象的某些部位就像超级强壮的肌肉。如果你剪掉太多这些特定的肌肉(它们恰好稀有但至关重要),大象就会崩溃。旧方法对大象的每个部位一视同仁(均匀修剪),这意外地剪掉了这些关键肌肉,导致在激进修剪时大象无法正常工作。

新方案:AutoPrune(自动修剪的巨象)

作者提出了一个大胆的问题:“大象能自己修剪自己吗?”

他们提出了一种名为AutoPrune的新方法。不再依赖人类理发师,而是利用大象自己的大脑(即大型语言模型)来找出最佳的自我修剪方式。

1. 图驱动的思维链(GCoT):“头脑风暴室”

大型语言模型很聪明,但有时可能会陷入单一的思维路径。为了解决这个问题,作者为大型语言模型建立了一个**“头脑风暴室”**。

  • 工作原理:想象大型语言模型正在设计一个发型。它不再只提出一个想法,而是分裂成五个不同的自我版本
    • 版本 A 说:“剪掉左耳。”
    • 版本 B 说:“剪掉右耳。”
    • 版本 C 说:“也许我们只修剪尾巴?”
  • 过程:这些版本探索不同的路径(一个想法的“图”)。它们测试各自的构想,看看哪个效果最好,最终胜出的最佳想法获胜。这有助于大型语言模型避开糟糕的想法,找到一种人类理发师未曾想到的优越修剪方案。
  • 结果:大型语言模型编写了自己的修剪“操作手册”,实际上在没有人类帮助的情况下,成为了自己的专家理发师。

2. 偏度感知动态稀疏分配(SDSA):“智能剪刀”

论文还解决了“异常值”问题(即那些超级强壮的肌肉)。

  • 旧方法:旧剪刀对大象的每一部分都按完全相同的比例修剪(例如, everywhere 都剪掉 50%)。这对强壮的肌肉来说很危险。
  • 新方法(SDSA):新剪刀是智能的。它们首先扫描大象,找出“超级强壮的肌肉”(异常值)所在的位置。
    • 如果某部分非常敏感(高偏度),剪刀会少剪以保护它。
    • 如果某部分不太敏感,剪刀会多剪一些。
  • 结果:这创造了一种平衡的修剪效果。大象变小了,但不会失去平衡或行走能力。

结果:更小、更聪明的巨象

作者在几只不同的“巨象”(如 LLaMA-1 和 LLaMA-2 等大型语言模型)上测试了这种新方法。

  • 性能:自我修剪的巨象表现优于由人类专家或其他计算机方法修剪的巨象。它们体型更小,但依然非常聪明。
  • 效率:大型语言模型自动设计了修剪规则,节省了聘请人类专家的时间和金钱。
  • 高比例修剪:即使剪除了大量部分(高剪枝率),新方法也能保持大象稳定,而旧方法则会让大象踉跄。

总结

简而言之,这篇论文指出:不要雇佣人类理发师来修剪你的巨型人工智能。 相反,让 AI 利用自己的大脑来设计定制的发型。通过利用“头脑风暴”技术寻找最佳想法,并使用“智能剪刀”保护最重要的部分,AI 可以在不丧失智能的情况下让自己变得更小、更快。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →