TokenGate: Adapting LLMs for Fast and Memory-Efficient Prefill with Learned Token Pruning
长文本里并非每个词都需要经过模型的所有层,论文让模型学会逐层跳过不重要的词。新增的轻量组件用词的重要程度调整注意力,使保留哪些词也能参与训练。作者在所测任务中报告,首字响应速度最高提升至原来的11.3倍,准确率未下降。
INSIGHT裁剪输入可以成为模型学会的一部分,不必完全依赖固定规则。这里压缩的是后续层处理的词,收益直接落在首次响应和显存占用上。组件需要轻量适配训练,适合把长文问答的输入处理成本作为独立优化目标。