一个在 4k 上下文上预训练好的模型,业务要求支持 32k。你打算用什么方法扩展?请说明你需要准备多少长文本数据、为什么不能简单把 RoPE 的 base 从 10000 直接改成 80000(或更大),以及扩完之后 4k 以内的短文本能力为什么会下降。
面试官在考什么
判断候选人是否理解位置编码的频率机制和长上下文扩展的代价,而不只是会改配置参数。
回答框架 · 前两步
- 常用做法:位置插值 / NTK-aware / dynamic NTK / YaRN 等 RoPE 缩放方案,配合少量长文本继续预训练。
- 继续预训练不需要和原预训练等量的长数据,通常几十亿 token 级、由短到长混合采样即可,关键是长样本文本的比例和多样性。
还有 3 条答案要点、 3 层追问、 5 个常见坑, 以及针对你简历的追问路径,在定制报告里。
延伸阅读
想看这个方向的整体准备路线? 进入「NLP 与大模型」方向页
同一能力域的其他题
- 微调一个大模型时,LoRA 相比全参微调省在哪里?rank 怎么选?为什么它可以在推理时合并回原权重?
- 请说明 SFT、RLHF、DPO 各自解决什么问题。为什么有了 RLHF 还要提 DPO?
- 长文本场景下,上下文窗口装不下时你会怎么办?请对比"截断、分块递归总结、检索"三种方案的代价。
- 你要用大模型做一个每天数百万次调用的线上功能,成本和延迟怎么压?请给出你的优化优先级。
- 一个 32 层 decoder-only 模型,hidden_size=4096,attention head 数 32,KV head 数 8(即 GQA),FP16 推理。现在 batch_size=4、序列长度 8192。请算出 KV cache 占多少显存;如果把 GQA 换成 MHA(32 个 KV head),显存变成多少;并解释为什么现在的模型普遍用 Pre-LN + RMSNorm,而不是原始论文的 Post-LN + LayerNorm。
- 一个 7B 稠密模型,预训练上下文 4K,现在要支持 32K 推理。模型结构:32 层、32 个注意力头、head_dim=128。服务端 8×A100 80G,batch size 固定 16,FP16 权重 + FP16 KV cache。两个问题:(1) 注意力从 MHA 换成 GQA(8 个 KV 头)前后,32K 上下文下单批 KV cache 各是多少?(2) 4K→32K 的位置外推你会选哪种方案,代价是什么?