让大模型停止「想太多」:从 DECS 到 vLLM,PagedAttention 全链路推理优化实战
聚焦 ICLR 2026 Oral 论文 DECS 与 vLLM PagedAttention 3.0,解析从训练端到推理端的全链路推理效率优化方案,附实战代码与性能基准数据。
引言
大模型正在消耗越来越多的计算资源。
2026 年,推理成本已占成熟企业 AI 基础设施支出的 80-85%(Bain & Company、McKinsey、FinOps Foundation 三方共识)。按行业趋势,到 2027-2028 年这一比例仍将维持在高位。
这背后有一个被长期忽视的问题:推理效率的根本瓶颈,不在模型能力,而在训练目标的精心设计。
5 月,ICLR 2026 接收了一篇来自复旦大学、上海交通大学和上海人工智能实验室的 Oral 论文 DECS。它证明了一个反直觉的结论:消除冗余推理 token,不仅能降低成本,还能提升准确率。
本文将围绕这条主线,结合 vLLM PagedAttention 3.0 的工程实践,输出可操作的推理优化全链路指南。
一、问题:你的模型「想太多」了吗
1.1 过度思考现象
以 DeepSeek-R1、OpenAI GPT Thinking 为代表的推理模型,在回答问题时经常出现「想太多」的现象:
用户问:2+2=?
模型答:Let me think about this carefully.
First, we have the equation 2+2=?
This is a basic arithmetic operation.
Let me verify... wait, actually, let me reconsider...
Alternatively, we could approach this differently...
The answer is 4.
模型在得出正确答案后,仍然反复出现「wait…」「let me check…」「alternatively…」等自我修正表达。这造成大量冗余计算开销。
1.2 传统长度惩罚的困境
业界尝试用长度惩罚(length penalty)来抑制过度思考,但面临两难困境:
| 策略 | 结果 |
|---|---|
| 加重惩罚 | 模型省略必要推理步骤,性能下降 |
| 减轻惩罚 | 冗余推理依旧存在,效率难提升 |
DECS 团队从理论上揭示了根源——长度惩罚存在两个根本性缺陷:
缺陷一:对高熵探索 token 的无差别攻击
「wait」「however」「alternatively」等高熵 token 本质上是逻辑衔接的「状态转移」,而非错误推理。序列级长度惩罚不区分 token 类型,对所有 token 均匀施加负面梯度,导致模型丧失探索能力,过早收敛到次优策略。
缺陷二:对局部冗余的变相奖励
引入关键概念 必要推理前缀(NRP, Necessary Reasoning Prefix)——从推理开始到首次得出正确答案所需的最短 token 序列。NRP 之后的所有 token 都是冗余的,但现有奖励机制中,这些冗余 token 仍可能收到正奖励。模型学不会在该停的时候停下来。
二、DECS:训练端的推理效率优化
2.1 核心思路
DECS(Decoupled Efficiency-Conscious Strategy)提出两步解耦训练框架,在训练源头消除冗余推理:
第一步:解耦 token 级奖励
- 训练轻量级 NRP 检测器(judge model)
- 对奖励函数「解耦」:NRP 范围内必要推理永不受惩罚;NRP 之后每个冗余 token 恒定为负奖励
- 模型只被禁止「画蛇添足」,而非在推理每一步都进行无效思考
第二步:课程式批次调度
- 当模型平均 NRP 占比较低(冗余多)时,减少简单题比例
- 随着训练推进、冗余减少,逐步提高简单题比重
- 缓冲机制,确保冗余被充分压缩的同时保留探索空间
2.2 实验结果
DECS 在 7 个基准测试(AIME2024/2025、MATH500、GPQA-Diamond、LiveCodeBench-v6 等)上验证了方法有效性:
| 模型 | 推理 token 削减 | Pass@1 准确率变化 |
|---|---|---|
| 1.5B | 57.17% | +2.48% |
| 7B | 49.50% | +0.8% |
| Qwen3-4B | 跨域泛化验证中 | - |
跨域泛化能力:NRP 检测器仅用数学语料训练,但效率优势强有力迁移到其他领域:
- 科学推理(GPQA-Diamond):56.33% token 缩减
- 编程任务(LiveCodeBench-v6):33.52% token 缩减
2.3 消融实验结论
| 移除组件 | 结果 |
|---|---|
| 课程式调度 | 压缩推理同时出现显著性能退化 |
| 解耦奖励 | 仍残留约 25% 冗余 token |
两个核心组件互补关系密切,缺一不可。
2.4 实战:DECS 配置指南
DECS 代码已开源,核心配置参数如下:
from decs import DECSTrainer
# DECS 训练配置
trainer = DECSTrainer(
# 模型配置
model="DeepSeek-R1-Distill-1.5B",
# NRP 检测器配置
nrp_detector={
"type": "lightweight-judge",
"threshold": 0.85, # NRP 边界判定阈值
"training_data": "math_corpus", # 仅用数学语料
},
# 解耦奖励配置
decoupled_reward={
"nrp_penalty": 0.0, # NRP 内不惩罚
"redundancy_penalty": -1.0, # NRP 后恒定负奖励
},
# 课程式调度配置
curriculum_scheduling={
"easy_ratio_start": 0.2, # 训练初期简单题比例
"easy_ratio_end": 0.6, # 训练后期简单题比例
"schedule_steps": 10000, # 调度步数
},
)
trainer.train()
关键洞察:准确率不降反升的根本原因——当模型不再被强制「多想想」时,它学会了在找到正确答案后立即停止,而不是继续无意义地自我修正。
三、vLLM PagedAttention 3.0:推理端的工程实践
训练端优化了模型行为,推理端则需要优化 KV Cache 管理。vLLM 的 PagedAttention 是当前最成熟的方案,2026 年的 3.0 版本带来多项关键突破。
3.1 核心技术原理
PagedAttention 借鉴操作系统虚拟内存分页管理思想:
- 将连续的 KV Cache 划分为固定大小的块(Block)
- 每个块可独立分配和释放
- 从根本上解决传统静态批处理中的显存碎片化问题
# 传统方式:连续内存分配
[████████████████████████████████████]
问题:请求结束时释放大块内存,产生碎片
# PagedAttention:分块内存管理
[████][████][ ][████][ ][████]
↑已释放 ↑已释放 ↑已释放
优势:碎片被小块填满,显存利用率大幅提升
3.2 PagedAttention 3.0 三大突破
| 特性 | 描述 | 效果 |
|---|---|---|
| 动态页大小调整 | 支持根据实际需求动态调整页面大小 | 显存利用效率进一步提升 |
| 跨 GPU 页迁移 | 实现 GPU 间页面迁移机制 | 显存碎片率大幅降低 |
| Hybrid Cache 架构 | 支持 DRAM+SSD 混合缓存 | 显存成本降低 50% 以上,延迟增加不显著 |
3.3 生产级部署配置
环境要求
- Python 3.10+
- PyTorch 2.2+
- vLLM 0.5+(建议 0.7+ 以获得完整 3.0 功能)
- CUDA 12.0+
- NVIDIA GPU(H100/A100 最佳)
vLLM 服务启动配置
# 启动 vLLM 服务(完整 PagedAttention 3.0 配置)
vllm serve deepseek-ai/DeepSeek-V4-Flash \
--tensor-parallel-size 2 \
--gpu-memory-utilization 0.92 \
--kv-cache-dtype fp8 \
--enforce-eager \
--enable-chunked-prefill \
--max-num-batched-tokens 8192 \
--max-num-seqs 256 \
--prefill-chunk-size 512
关键参数解析
| 参数 | 作用 | 推荐值 |
|---|---|---|
--gpu-memory-utilization | GPU 显存分配比例 | 0.92(留 8% 给系统) |
--kv-cache-dtype | KV Cache 量化精度 | fp8(精度与效率平衡) |
--enable-chunked-prefill | 启用分块预填充 | 降低长序列首 token 延迟 |
--max-num-batched-tokens | 单批次最大 token 数 | 根据显存和模型调整 |
--prefill-chunk-size | 预填充分块大小 | 512(长序列优化) |
3.4 KV Cache Block Size 调优
Block Size 是 PagedAttention 的核心调优参数,影响显存利用效率和吞吐量:
# vLLM Block Size 配置示例
from vllm import LLM, SamplingParams
llm = LLM(
model="deepseek-ai/DeepSeek-V4-Flash",
tensor_parallel_size=2,
# Block Size 配置(关键调优参数)
block_size=16, # 默认 16,可选 8/16/32
# 动态 Block Size(3.0 新特性)
enable_dynamic_block_size=True,
# Prefix Caching(共享前缀优化)
enable_prefix_caching=True,
# 分布式 KV Cache(跨 GPU 迁移)
enable_distributed_kv_cache=True,
)
sampling_params = SamplingParams(
temperature=0.7,
top_p=0.95,
max_tokens=2048,
)
Block Size 选择指南:
| Block Size | 适用场景 | 优势 | 劣势 |
|---|---|---|---|
| 8 | 短上下文、高并发 | 延迟低 | 碎片略多 |
| 16 | 通用场景(推荐) | 平衡之选 | - |
| 32 | 长上下文、低并发 | 显存利用率高 | 延迟略增 |
3.5 Continuous Batching 配置
Continuous Batching(持续批处理)可在请求级别动态批处理,最大化 GPU 利用率:
# Continuous Batching 完整配置
llm = LLM(
model="deepseek-ai/DeepSeek-V4-Flash",
# 批处理配置
max_num_batched_tokens=16384, # 单批次最大 token 数
max_num_seqs=512, # 单批次最大序列数
max_model_len=131072, # 最大模型上下文长度
# Prefill 优化
enable_chunked_prefill=True,
prefill_chunk_size=1024,
# Speculative Decoding(投机解码加速)
speculative_model="deepseek-ai/DeepSeek-V4-Flash",
num_speculative_tokens=5, # 投机 token 数
)
3.6 性能基准数据
在 H100 GPU(80GB)×2 配置下,DeepSeek-V4-Flash(13B 激活参数)的推理性能基准:
| 配置 | 吞吐量 (tokens/s) | 首字延迟 (TTFT) | 显存占用 |
|---|---|---|---|
| BF16 基准 | 2,400 | 45ms | 68GB |
| FP8 量化 | 3,200 (+33%) | 42ms | 42GB |
| + Continuous Batching | 4,800 (+100%) | 40ms | 58GB |
| + Speculative Decoding | 6,200 (+158%) | 38ms | 61GB |
成本优化效果:FP8 量化 + Continuous Batching 组合,显存成本降低约 50%,吞吐量提升 2 倍。
四、投机解码:推理加速的补充手段
除 PagedAttention 外,投机解码(Speculative Decoding)也是 2026 年的热点方向,通过小模型「草稿」+ 大模型「验证」的组合实现推理加速。
4.1 Microsoft Research: Learning to Draft
微软研究院 2026 年 3 月发布的 Learning to Draft 论文,将强化学习引入投机解码的草稿生成:
核心思路:
- 训练 draft model 学习「何时该猜什么 token」
- 自适应调整投机策略,而非固定概率采样
- 在代码生成、数学推理等结构化任务上效果显著
性能数据:
- 在 Qwen-72B 上,吞吐量提升 2-3 倍
- 验证接受率 >85%(保持质量)
4.2 Google Gemma 4: MTP (Multi-Token Prediction)
Google 为 Gemma 4 系列引入 MTP(Multi-Token Prediction)起草器,通过多步预测实现推理加速:
# Gemma 4 MTP 配置示例
from vllm import LLM
llm = LLM(
model="google/gemma-4-31b-it",
# MTP 推测解码配置
speculative_model="google/gemma-4-2b-it", # 小型起草模型
num_speculative_tokens=4,
# 动态接受率调整
speculative_draft_accept_threshold=0.7,
)
性能提升:推理速度最高提升 3 倍(在 Gemma 4 31B 上)。
4.3 Speculative Speculative Decoding
网易上的一篇有趣论文提出了「投机的投机解码」——在投机解码基础上再套一层投机:
核心思想:不仅 draft model 猜测,大 model 也猜测「draft model 会猜什么」,减少验证开销。
效果:在特定场景下,相比传统投机解码再提升 5 倍吞吐量(理论上限,实际取决于任务)。
五、全链路优化方案总结
结合 DECS 训练端优化和 vLLM 推理端优化,以下是完整的推理效率提升路线图:
5.1 训练端(模型行为优化)
┌─────────────────────────────────────────────────────────┐
│ DECS 训练框架 │
│ │
│ NRP 检测器 → 解耦奖励 → 课程调度 │
│ ↓ ↓ ↓ │
│ 精确识别 精准惩罚 保护探索 │
│ 冗余边界 NRP外 能力 │
│ │
│ 效果:推理 token 削减 50-60%,准确率提升 0.8-2.5% │
└─────────────────────────────────────────────────────────┘
推荐场景:新训练或微调的推理模型(DeepSeek-R1 系列、Qwen3 系列等)。
5.2 推理端(基础设施优化)
┌─────────────────────────────────────────────────────────┐
│ vLLM PagedAttention 3.0 + 投机解码 │
│ │
│ PagedAttention 3.0 投机解码 │
│ ├─ 动态页大小 ├─ MTP(Gemma 4) │
│ ├─ 跨GPU迁移 ├─ Learning to Draft │
│ └─ Hybrid Cache └─ 自适应策略 │
│ │
│ 量化方案:FP8 + INT4 QAT │
│ 批处理:Continuous Batching │
│ │
│ 效果:显存降低 50%,吞吐量提升 2-4 倍 │
└─────────────────────────────────────────────────────────┘
5.3 决策矩阵
| 场景 | 推荐方案 | 预期收益 |
|---|---|---|
| 新训练的推理模型 | DECS 训练优化 | Token 削减 50%+,准确率 +0.8-2.5% |
| 已部署模型优化 | vLLM PagedAttention 3.0 | 吞吐量 +2-4 倍,显存 -50% |
| 低延迟要求场景 | Speculative Decoding + MTP | 延迟 -60%,TTFT 降低 |
| 超长上下文 | Hybrid Cache + 分块预填充 | 1M+ 上下文支持,延迟可控 |
六、实战:构建推理优化流水线
以下是整合 DECS 和 vLLM 的完整推理优化流水线示例:
import asyncio
from vllm import LLM, SamplingParams
from decs import DECSConfig
class InferenceOptimizer:
def __init__(self, model_path: str, decs_enabled: bool = True):
# 初始化 vLLM
self.llm = LLM(
model=model_path,
tensor_parallel_size=2,
gpu_memory_utilization=0.92,
kv_cache_dtype="fp8",
enable_chunked_prefill=True,
max_num_batched_tokens=16384,
enable_prefix_caching=True,
)
# DECS 配置(如果模型经过 DECS 训练)
self.decs_enabled = decs_enabled
self.sampling_params = SamplingParams(
temperature=0.7,
top_p=0.95,
max_tokens=4096,
)
async def infer(self, prompt: str, use_speculative: bool = False) -> str:
if use_speculative and not self.decs_enabled:
# 启用投机解码
params = self.sampling_params.copy()
params.use_speculative = True
params.num_speculative_tokens = 5
else:
params = self.sampling_params
outputs = self.llm.generate([prompt], params)
return outputs[0].outputs[0].text
# 使用示例
async def main():
optimizer = InferenceOptimizer(
model_path="deepseek-ai/DeepSeek-V4-Flash",
decs_enabled=True # 模型经过 DECS 优化
)
# 标准推理
result = await optimizer.infer("解释量子纠缠原理", use_speculative=False)
print(f"标准推理结果: {result[:100]}...")
# 启用投机解码加速
result = await optimizer.infer("解释量子纠缠原理", use_speculative=True)
print(f"加速推理结果: {result[:100]}...")
asyncio.run(main())
结论
推理优化正在经历从「暴力堆算力」到「精准控成本」的根本转变。
训练端:DECS 证明,消除冗余推理 token 不是在牺牲智能,而是在让模型学会更高效地思考。57% 的 token 削减与准确率提升共存,打破了「想得越多越好」的幻觉。
推理端:vLLM PagedAttention 3.0、投机解码、FP8 量化等技术的组合,使显存成本降低 50% 以上、吞吐量提升 2-4 倍成为工程现实。
关键趋势:
- 效率即智能:未来模型评估,不仅看基准分数,还要看「每分智能消耗多少 token」
- 全链路优化:训练端行为优化 + 推理端基础设施优化的组合,比单一手段更有效
- 开源生态:vLLM、DECS 等开源项目正在降低先进推理优化技术的落地门槛
让模型停止「想太多」,是 2026 年 AI 工程的核心课题。
参考资源:
- DECS 论文:ICLR 2026 Oral
- DECS 项目主页:pixas.github.io/decs-iclr26-site
- vLLM 官方文档:docs.vllm.ai
- Learning to Draft:Microsoft Research