河源塑料管材设备 OpenAI又开源了!仅0.4B,给模型大瘦身

塑料挤出机

智东西河源塑料管材设备

作者 | 王涵

编辑 | 心缘

智东西12月15日报道,昨天,OpenAI开源新模型Circuit-Sparsity,模型参数量仅0.4B,99.9%的权重为零。

▲Circuit-Sparsity开源(来源:Hugging Face)

这个技术试图解决模型的可解释问题,简单来说就是回答“模型为什么做出这个决策?”以及“它是如何得出这个结果的?”这两个问题。

在AI飞速发展的今天,大语言模型(LLM)虽然表现出了惊人的能力,但其内部运作机制始终像一个神秘的“黑箱”。

我们不知道它为何做出某个回答,也不清楚它是如何从海量数据中提取知识的。这种不可解释,成为了AI在医疗、金融、法律等高风险域落地的重大障碍。

对此,OpenAI研究团队训练出了一个权重稀疏的Transformer模型,强制模型权重矩阵中99.9%权重为零,仅保留0.1%非零权重。

在这项研究中,研究团队在模型内部形成了紧凑且可读的“电路”(Circuits),每个电路都仅保留了保证模型能的关键节点,神经元的激活变得具有明确的语义。

有外网网友称这一技术让当下的MoE(混家模型)走到了尽头,并说“我们一直以来都将权重隔离到‘家’中,以此粗略地近似稀疏,仅仅是为了满足稠密矩阵核的要求。”

▲外网评价(图源:X)

更有网友将这项研究形容为将模型“减肥到只剩骨架”,还说这项研究就好像打开了黑匣子,不试图解开稠密模型而是直接构建稀疏模型,正是这项研究有趣的地方。

▲外网评价(图源:X)

但有些网友却不这么认为,称其没有看出MoE模型为何会因此走到尽头,并进一步解释说这一技术是针对XAI(可解释AI)的,它的训练成本要高100-1000倍,回到“研究时代”并不意味着让事情变得更复杂。

▲外网评价(图源:X)

该模型目前受限于计算率瓶颈,其运算速度较密集模型慢100至1000倍,将该技术直接应用于千亿参数级别的前沿大模型,现阶段尚不具备可行。

开源地址:河源塑料管材设备

Github:

https://github.com/openai/circuit_sparsity

Hugging Face:

https://huggingface.co/openai/circuit-sparsity

一、训练稀疏Transformer,OpenAI理清模型内部计算

要理解这项研究的突破,先需要明白传统大模型为何难以解释。

在标准的密集模型(Dense Models)中,神经网络存在一种被称为“级位置”(Superposition)的现象。简单来说,为了存储海量的信息,模型被迫让单个神经元或权重矩阵同时编码多个完全不同的概念。

这种特征纠缠导致了严重的后果,例如模型的决策不可追溯和逻辑混乱,当模型输出一个结果时,我们无法确定是哪个具体的“概念”在起作用。

针对以上问题,以前的研究通常从试图拆解密集、纠结的网络开始。但OpenAI团队采取了一种“反直觉”的策略,即训练权重稀疏的Transformer模型,强制模型权重矩阵中99.9%权重为零,仅保留0.1%非零权重。

强制模型限制了模型只能使用其神经元之间少的可能连接,而这一简单的更改,几乎从根本上理清了模型的内部计算。

▲每个神经元只与下一个层的几个神经元相连(图源:OpenAI技术博客)

具体的技术手段包括:

1、动态剪枝与稀疏约束:在训练过程中,系统会动态执行“剪枝”操作,每一步优化后仅保留对值大的权重(Top-K稀疏化)。

2、激活稀疏化:在残差流、注意力键/值矩阵等关键位置,研究团队引入了AbsTopK激活函数,强制仅保留前25%的激活值。

3、架构微调:为了配稀疏化,研究团队用RMSNorm替代了传统的LayerNorm,避免归一化操作破坏稀疏,同时引入了“Bigram表”来处理简单的模式匹配,从而释放模型的主干容量去处理复杂的逻辑推理。

二、模型内部形成紧凑可读的“电路”,塑料管材生产线规模缩减16倍

这项技术的大成果,是模型内部形成了紧凑且可读的“电路”(Circuits)。

在传统密集模型中,完成一个任务可能需要成千上万个节点协同工作,逻辑分散且难以捕捉。而在稀疏模型中,研究团队观察到了简的计算路径:

1、简的逻辑单元:例如在处理“字符串闭”任务时,模型仅用12个节点就构建了一个的电路,清晰地展示了它是如何检测单引号或双引号是否闭的。

2、可读的特征:神经元的激活变得具有明确的语义。研究人员发现了一些神经元门负责检测“单引号”,另一些则像“计数器”一样精确地追踪列表的嵌套深度。

3、规模缩减16倍:对比实验显示,在相同的任务损失下,稀疏模型的电路规模比密集模型小了16倍。这意味着解读AI思维的难度降低了整整一个数量级。

▲稀疏模型的电路规模比密集模型小了16倍(图源:OpenAI技术论文)

为了验证这些电路的真实,团队进行了“均值消融”实验。结果证明,移除非电路节点对任务几乎没有影响,而一旦移除电路中的关键节点,模型能就会瞬间崩塌。这证实了这些电路确实是模型执行任务的“经之路”。

▲“均值消融”实验(图源:OpenAI技术论文)

三、稀疏模型解读力强但速度慢千倍,OpenAI提出“桥梁网络”

为了测量稀疏模型计算的解耦程度。研究团队设计了一套简单的算法任务。对于每个模型,他们都将其剪裁成了仍能执行该任务的小电路,并检查了该电路的简洁程度。

研究团队发现,用规模更大、稀疏度更高的模型进行训练后,就能够依托结构更简洁的电路,构建出能更强的模型。

▲模型的可解释与能力的对比图(图源:OpenAI技术博客)

从模型可解释与能的对比图可见,在稀疏模型规模固定的前提下,提升稀疏,也就是将更多权重置零,虽会导致模型能有所下降,但能显著增强其可解释。

尽管稀疏模型在可解释方面优势突出,但其应用目前受限于计算率瓶颈:稀疏矩阵运算无法借助Tensor Cores实现加速,运算速度较密集模型慢100至1000倍。这意味着,将该技术直接应用于千亿参数级别的前沿大模型,现阶段尚不具备可行。

为此,研究团队提出了“桥梁网络”(Bridges)方案:

今日沪深两市主力资金开盘净流出121.97亿元,尾盘净流出35.67亿元,全天净流出214.71亿元。 

开盘一早,人形机器人板块全线高开,三花智控、汇川技术、拓普集团等龙头股盘中涨幅拉高,带动机器人ETF易方达(159530)今日涨3.06%,年内涨幅24.29%。

1、编码-解码映射:在稀疏模型与预训练的密集模型之间插入一个编码器-解码器对。

2、跨模型干预:编码器将密集模型的激活映射到稀疏空间,解码器则反向转换。

“桥梁网络”(Bridges)方案可以在“透明”的稀疏模型上修改某个特征,然后通过桥梁将这种扰动映射回“黑箱”的密集模型,从而实现对现有大模型的可解释行为编辑。

结语:OpenAI提出稀疏化新路径,让大模型从“黑箱”走向“可解释”

OpenAI研究团队的这项研究,标志着AI可解释域的一项重要突破,也印证了理解AI并非遥不可及的目标。

电话:0316--3233399

研究团队在论文博客中称,这项工作是迈向更宏大目标的早期探索。接下来,他们计划将相关技术扩展至更大规模的模型,同时进一步解释更多模型的行为逻辑。

为解决稀疏模型训练率低下的问题,团队提出了两个后续研究方向:一是从现有密集模型中提取稀疏电路,替代“从头训练稀疏模型”的传统方式;二是研发更高的可解释模型训练技术,推动相关技术更易落地生产。

“我们的目标是逐步扩大可可靠解释的模型范围河源塑料管材设备,同时打造相关工具,让未来的AI系统更易于分析、调试与评估。”研究团队在论文博客中写道。