近日，谷歌旗下的DeepMind团队宣布了一项名为“可微缓存增强”的创新技术，这一技术旨在优化大型语言模型（LLMs）的推理性能，同时避免大幅增加计算成本。
在自然语言处理、数学运算及逻辑推理等领域，大型语言模型已成为解决复杂问题的关键工具。随着技术的不断进步，研究人员致力于提升这些模型的数据处理能力，使其能够生成更精确、更符合上下文的响应。然而，随着模型复杂度的增加，如何在有限的计算资源下保持高效运行成为一大挑战。
一个显著的问题是，现有的大型语言模型往往难以在不同任务间进行有效的推理，或执行超出其预训练架构的计算。为了提高模型性能，研究人员通常会尝试在任务处理过程中生成中间步骤，但这通常会导致延迟增加和计算效率低下。这一限制严重影响了模型执行复杂推理任务的能力，特别是那些需要长距离依赖关系或高精度预测的任务。
“可微缓存增强”技术通过引入一个经过训练的协处理器，以潜在嵌入的方式增强LLMs的键值（kv）缓存，从而丰富了模型的内部记忆。这一方法的关键在于，保持基础LLMs不变，同时异步训练协处理器。通过这种方式，研究团队成功地在不牺牲计算效率的情况下，显著提升了模型性能。
整个处理流程分为三个关键阶段：首先，冻结的LLMs从输入序列中生成kv缓存；接着，协处理器利用可训练软令牌处理这些kv缓存，生成潜在嵌入；最后，增强的kv缓存被反馈回LLMs，以生成更丰富的输出。这一创新方法不仅简化了模型处理复杂任务的过程，还提高了其准确性和效率。
在Gemma-2 2B模型上的测试结果显示，“可微缓存增强”技术在多个基准测试中均取得了显著成果。例如，在GSM8K数据集上，准确率提高了10.05%；在MMLU基准测试中，性能提升了4.70%。该方法还显著降低了模型在多个标记位置的困惑度，进一步证明了其有效性。
DeepMind的这一研究成果为大型语言模型的推理能力增强提供了新的视角和解决方案。通过引入外部协处理器来增强kv缓存，研究团队在保持计算效率的同时，实现了模型性能的显著提升，为LLMs处理更复杂、更具挑战性的任务奠定了坚实基础。
