大家好,
我想分享 Draco AI 的设计不变量和架构。这是一个全栈、与硬件无关的大语言模型系统,完全使用纯 Python 和 NumPy 从头构建。该项目背后的核心理念是严格的代码质量、数学正确性和清晰的架构——完全消除对 PyTorch 或 HuggingFace 等重型框架的依赖。
⚠️ 100% 完全开源:需要澄清的是,这不是那种只公开推理包装器的半吊子开源发布。Draco AI 在其整个执行路径上都是 100% 开源的,包括两个深度解耦层:推理引擎(
modeling/)的执行机制和思考引擎(thinking_engine/)的认知/推理路径。
🔗 开源与仓库
基线包含一个强大的 pytest 验证套件,涵盖分组查询注意力、多头潜在注意力、混合注意力、美杜莎头以及投机树解码执行路径。
- GitHub 仓库: https://github.com/DUCNGUYEN-creator/Draco_AI
- Discord 社区: https://discord.gg/dJXMbexdTh
🚀 关键技术亮点
- 零人工智能依赖执行路径: 100% 纯 Python/NumPy 执行回退。采用与硬件无关的架构,其中 Triton(图形处理器)和 Numba(中央处理器即时编译)仅作为惰性的可选加速器。
-
三值量化(BitNet 1.58b): 前馈网络专家支持三值权重,实现高达 70-80% 的内存占用减少。前向传播利用仅加法矩阵乘法(
Y = (X @ pos_mask.T - X @ neg_mask.T) * scale),在核心执行路径上没有浮点乘法运算。 - 三层分层记忆缓存: 实现从粗到细的路由机制:第 0 层(精确键值滑动窗口)$\rightarrow$ 第 1 层(压缩块摘要)$\rightarrow$ 第 2 层(目录章节向量),用于 $O(1)$ 快速查找,避免线性扫描性能下降。
- 12 阶段自愈认知管道: 具备由 9 个独立验证器(检索、矛盾、通过安全抽象语法树的数值、符号等)和 5 种概率融合方法(包括贝叶斯更新和邓普斯特-谢弗不确定性质量处理)驱动的深层幻觉子系统。
- Mirostat v2 核心校正: 使用真正的巴苏(2020)负反馈双过程循环($\mu \leftarrow \mu - \eta * (\text{surprise} - \tau)$)实现令牌采样,动态缩小目标交叉熵窗口以维持生成质量。
-
实时健康与精度控制: 具备主动的
HealthMonitor(健康监控器),跟踪非数字/无穷大传播、饱和度和对抗性专家崩溃。它向DynamicPrecisionManager(动态精度管理器)发送信号,基于溢出指数移动平均进行建议性的数据类型升级/降级,同时严格遵守硬性视频随机存取存储器预算。
🛠️ 系统不变量与确定性
为了防止投机系统中常见的静默损坏,Draco AI 强制执行严格的执行不变量:
-
单向数据流图: 严格结构化为
runtime → layers → ops → kernels(运行时 → 层 → 操作 → 内核)。不允许向上执行调用。device.py作为硬件能力的单一事实来源。 -
旋转位置编码位置一致性: T
免责声明:本文内容来自互联网,该文观点不代表本站观点。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,请到页面底部单击反馈,一经查实,本站将立刻删除。