兄弟们,昨晚看到Anthropic放出Claude 3.5 Sonnet的百万token上下文支持,动手测了一波,发现些有意思的细节,给大家分享下。
先说结论:长上下文场景下,模型在词向量检索一致性上明显优于GPT-4 Turbo(128k),但推理速度下降约30%。我拿《三体》三部曲全文(约90万token)做问答,前20%准确率95%,后80%降到82%,主要是因为注意力稀疏。官方说用“分页摘要”技巧能缓解,实测加个简单few-shot模板后上升到89%。
技术上,Anthropic用了动态稀疏注意力机制,类似Mosaic ML的改动。对开发者,关键建议:如果部署长上下文应用,微调时batch size别超过4,否则显存爆炸。我用4x A100 80G跑16k切块训练,峰值占用72GB。代码公开在GitHub。
另外,Mistral刚开源了Mixtral 8x22B的MoE模型,参数量141B但推理只占40B等效。对比Llama 3 70B,在MMLU上高3.2%,但数学推理差0.7%。小团队可以优先用Mixtral,成本低。推荐看看huggingface上@philschmid的量化指南。
大家有踩坑的没?欢迎跟帖聊。 |