ThinkBit: Search Codewords, Recalibrate Scales, Select by Reasoning
ThinkBit:搜索码字、重校准尺度、按推理选择
Microsoft Applied Sciences Group
标量2-bit量化框架,通过敏感度分配位宽、码字搜索、仅训尺度和按推理选择,在MATH-500上大幅超过同尺寸2-bit方法
📌 工作参考价值(5 点)
- 线上长思维链模型压缩:如果要在电商/Agent场景部署推理模型,可用标量2-bit格式(整数codeword × bf16 scale,组大小128),无codebook/rotation/zero-point,dequant只需一次乘法,工程实现简单,能显著降低权重内存带宽;论文中14B模型decode从86提高到150 tokens/s。
- 按层敏感度混合精度:不必全模型统一2-bit,先用KronQ敏感度(输入范数×输出梯度范数)给每层分配ternary/2-bit/3-bit,可将搜索/推荐/Agent模型的量化损失最小化;论文表明2-bit点在成本-噪声平面上不占优,多数层会分配到ternary或3-bit。
- 低成本恢复与校准:冻结codeword,只训练scales和RMSNorm gains(<1%参数)就能把WikiText ppl从46.9降到19.46、MATH-500从14.8拉到47.5;这适合对线上新领域数据做快速低资源校准,无需全量QAT。
- 用业务长链路指标选checkpoint:困惑度和多选题无法区分量化配置(差异<1.5点),但长程任务如MATH-500差异超10点;应构造业务推理样本(如商品推荐解释、Agent工具调用轨迹最终成功率)做配对sign-flip test来选择checkpoint,避免误选。
- QAT数据要加完整推理轨迹:仅通用数据QAT会损害推理(MATH-500从47.5降到43.0),加入完整长思维链数据后同尺寸模型MATH-500从46.7跃升至73.0;业务上收集高质量Agent轨迹数据做量化后训练是划算的。