开源开发者 anuj0456 近日在 GitHub 上开源了项目 OpenArch。它参考研究者 Sebastian Raschka 整理的 72 种架构图谱,不依赖任何第三方大模型封装,用最纯粹的 PyTorch 单文件逐行手写现代大模型,把 DeepSeek R1、Llama 3、Kimi K2 甚至仍在规划中的 Grok-2.5 拆解得一清二楚。

很多从业者以为今天的大模型百花齐放是底层算法的分崩离析,或者误以为研读开源代码非得穿透 Hugging Face 库里层层包裹的几万行工程胶水。事实恰恰相反:剥掉工程抽象之后,现代大模型的骨架早已极度统一

现代主流因果解码器的收敛标准件 1. 归一化与位置编码(预处理阶段) RMSNorm 全面替代 LayerNorm / RoPE 旋转位置编码替代绝对位置编码 2. 注意力投影层(长文本核心分水岭) 从全量 MHA 逐步向分组查询 GQA、潜在注意力 MLA 及滑动窗口分流 3. 门控前馈网络(激活计算拓扑) 彻底淘汰原始 ReLU / 全面转向 SwiGLU、GeGLU 门控单元与细粒度 MoE

剥离十万行工程胶水,底座早已高度收敛

工业级生产库为了兼顾硬件切片、量化适配和历史兼容,写了海量的抽象继承与调度逻辑。OpenArch 采取了完全相反的路线:每个模型单独存放在一个独立文件夹内,比如 text/llama3/model.py,通篇只有一个可读文件,没有深层类继承,也不做黑盒封装。

主流大模型底层骨架高度趋同,差异仅在局部参数微调(示意图)
主流大模型底层骨架高度趋同,差异仅在局部参数微调(示意图)

翻开这些独立实现会发现,所谓的全新模型,其基础物理结构几乎共享同一套公理。在归一化和位置计算上,原始 Transformer 的 LayerNorm 和绝对位置嵌入早已被淘汰,全线换成了开销更轻的 RMSNorm 与基于旋转矩阵的 RoPE。部分模型为了防止训练中后期注意力打分失控,补上了针对 Q 和 K 矩阵的 QK-Norm。

前馈网络也是一样。无论是 Meta 的 Llama、阿里的 Qwen,还是 Google 的 Gemma,传统的两层标准 ReLU 前馈网络全部退场,换成了包含门控机制的 SwiGLU 或 GeGLU 单元。

“求木之长者,必固其根本。”现代大语言模型的根本底座不是百舸争流,而是早就完成了一场工业层面的大收敛。各家技术报告里那些天花乱坠的名词,本质上只是在这套标准底座上拧动了几颗具体的参数螺丝。

注意力与稀疏路由:算力突围的两处分水岭

既然地基相同,各家模型真正的分歧发生在何处?答案集中在两处瓶颈:长文本推理时的显存挤占,以及参数规模扩大后的算力封顶

在 671B 庞大参数下,细粒度路由每次仅激活 37B 算力单元
在 671B 庞大参数下,细粒度路由每次仅激活 37B 算力单元
两类计算路线的资源权衡对照 稠密全激活路线(如 Llama 3) 参数激活比例 100% 注意力选择:多头或分组查询 GQA 代价:长序列显存昂贵,算力开销线性堆叠 稀疏细粒度路由(如 DeepSeek R1) 参数激活比例(671B 仅激活 37B) 5.5% 注意力选择:多头潜在注意力 MLA 代价:低维潜变量投影压缩,工程路由极度复杂

长文本交互的核心成本卡在 KV Cache 显存占用上。在 OpenArch 实现的对比谱系里,注意力机制有着极其分明的演进阶梯:标准多头注意力 MHA 开销最大,分组查询注意力 GQA 做了折中折叠,而以 DeepSeek 为代表的多头潜在注意力 MLA 则走向极端,通过将键值缓存压缩成低维潜变量向量,换取长上下文下的显存锐减。

在计算拓扑上,算力鸿沟则表现为稠密模型与专家混合架构的对抗。Llama 3 等稠密模型每次推理都激活全部权重,参数规模很容易撞上硬件天花板;而像 DeepSeek R1,在 671B 总参数下仅激活约 37B 参数,利用极细粒度的路由策略,让单次推理成本维持在轻量级水平。

至于诸如 Kimi K2 探入的 1T 规模,或是 GLM 4.5 引入的多 Token 预测 MTP 机制,本质上都在这套空间里倒腾:要么继续压缩上下文传输代价,要么通过推测解码把硬件并行利用率榨干。

决定模型生死的分水岭不是外挂的名词,而是谁能在长文本下更少占用显存,在参数膨胀时激活更少的晶体管。

读懂代码很容易,但工程壁垒依旧坚固

OpenArch 这种项目的价值,是给研发人员提供了一个透明的实验室。当你需要推演一个注意力维度的变形,或者测试某层残差连接的改动时,无需再去翻阅数万行包含复杂硬件加速算子的工业框架。

规整的分页内存管理构筑起防线,防止并发洪峰击穿显存(示意图)
规整的分页内存管理构筑起防线,防止并发洪峰击穿显存(示意图)
  • 建议.想弄清大模型内部张量流动逻辑的开发者,把单文件代码当作数学公式的落地参考,比直接死磕框架源码更有效率。

然而,看懂架构图谱不等于掌握量产能力。这类教学型实现有着非常明确的边界:它们普遍缺乏生产级权重加载管线,也没有经过严格的数值精度等价性测试。在实际工程中,微小的偏置项初始化差异、缩放因子的浮点误差,甚至权重绑定的处理方式,都会让毫无工程防护的手写代码在推理阶段直接吐出乱码。

  • 风险.项目清单中列出的一些尚未正式发布详细报告的模型配置,混杂了社区预设参数,脱离真实权重映射与并行切分,切忌直接引入生产服务。

真正的技术壁垒从来不单单是几张由注意力与矩阵相乘构成的图纸。图纸早已经摆在了台面上,但怎么把 671B 规模的庞然大物安全平铺在千卡集群上,怎么利用分页管理在并发暴涨时不爆显存,才是工业界没写进单文件里的重装铠甲。