BitNet 团队把 1.58-bit 量化推进到 embedding:检索向量也可以"训练时就压"

Microsoft BitNet 团队本周在 Hugging Face 开源 BitNet Embeddings 系列,主打"BITEMBED"框架——这是首次把训练阶段 1.58-bit 量化从 LLM 主干延伸到文本 embedding 任务,覆盖 Qwen3-0.6B 和 Gemma3-270M 两个 backbone,权重三元化 {-1, 0, +1}、激活 8-bit,配合 SubLN 与 RoPE,配套上下文 32K、MIT 许可。 技术路径分两步:**第一步**先把 pretrained LLM 直接转成 BitNet-style embedding encoder,保留 absmean 量化、abs-max-per-token 激活量化;**第二步**在 representation learning 上重新对齐——先在 1B 文本对上做持续对比预训练(InfoNCE),再用监督对比微调从 FP16 教师蒸馏,包含 similarity-distribution 和 attention-relation 两路蒸馏信号,让三元权重下的表征能力往 FP16 教师看齐。 数字很硬:在 MMTEB(eng, v2)上 BitNet-Embedding-0.6B 拿到 **67.60** 平均分,仅比 FP16 教师(67.95)低 **0.35**,分类、检索、摘要等 8 项子任务几乎打平;推理端 CPU 8 线程 **pp128 输入 870.90 t/s**,相对 FP16 **2.28× 加速**,长输入仍能维持 1.4–1.8× 的领先;输出 embedding 还能再量化到 1/2/4/8 bit,存储随场景弹性伸缩。 这一点对 RAG 和企业知识库意义不小:embedding 阶段是 retrieval 系统的"CPU 端常驻瓶颈",把 backbone 训练时就压到 1.58-bit,意味着整条 retrieval pipeline 第一次可以在端到端 1-bit 链路跑通,bitnet.cpp 推理框架继续复用。过去 BitNet 的故事只写到 LLM 本身,这次补上了检索向量的最后一环——开源工程构件就位,剩下的就是把它接进 LangChain / LlamaIndex 等主流 orchestration 层,社区生态可能很快把这套配置做成默认 RAG 选项。 更值得注意的是 BitNet 这条线的"渐进扩张"策略:从 1-bit LLM 推理框架,到今天 1.58-bit embedding,每一步都没有急着从 FP16 切整数精度,而是 **训练时就让模型学会在 ternary 表达里收敛**——这条路线对端侧 AI 的可复制性可能比想象中更大。