跳到正文
Reddit · r/LocalLLaMA· /u/pmttyji·· 5 天前精选AI 评分60

论文提出用 16 MB 冻结词表替代小模型嵌入层,BitNet b1.58 版本整模仅 46 MB

[Paper] Embeddings Are Overpriced? A Frozen 16 MB Vocabulary for Small and Ternary Language Models

来源与原文

Reddit · r/LocalLLaMA

作者:/u/pmttyji

发布时间:(北京时间)

阅读原始来源 · reddit.com ↗
完整链接与内容说明

https://www.reddit.com/r/LocalLLaMA/comments/1xbnpiu/paper_embeddings_are_overpriced_a_frozen_16_mb/

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

一篇 arXiv 论文提出用 16 MB 冻结 512-bit 编码表替代可训练嵌入层,经主成分投影和符号二值化得到,无任何可训练参数。在 Gemma 3 270M 形状模型上,可学习词表花 2.6 倍可训练参数只换来 0.20 nats 验证损失改善,而冻结词表模型在 LAMBADA、HellaSwag、PIQA 上持平或更好,仅 ARC-Easy 落后;在罕见词上反超 1.7 到 2.3 nats,训练后新增 10,000 词仅损失 3%,可学习词表则崩掉 43%。配合 ternary BitNet b1.58 主体,整个模型仅 46 MB,在笔记本 CPU 上以 62 MB 内存生成 615 到 740 tokens 每秒,速度是 float 模型 8-bit GGUF 的五到六倍、内存为其七分之一。

推荐理由

论文用同一训练设置对照验证冻结 16 MB 词表的效果,并给出 LAMBADA、PIQA 等基准与 CPU 生成速度数据,权重和评测全部开放。

来源:Reddit · r/LocalLLaMA · reddit.com