
你有莫得过这样的体验:向AI助手问一个问题,它想考了几秒钟,然后给你一段精彩的文书。你咋舌于它的智能,但阿谁“几秒钟”的恭候,总让你以为差了点什么。
另一边,ChatGPT、Claude、文心一言这些居品,背后的大模子一个比一个刚劲,参数目动辄上千亿,能写诗、能编程、能看病。但一朝把它们部署到执行居品中,问题就来了——反应慢、老本高、并发一上来就崩。
这即是大模子行业的“临了一公里”问题:模子很灵巧,但用起来很慢。
这背后不是算力不够,而是推理架构的工程挑战。本文从技能角度拆解,为什么大模子推理这样难,以及业界是如何惩办这个问题的。

一、大模子推理的特有窘境
要持续大模子推理为什么难,先要持续它和传统AI推理的分裂。
1.1 传统推理 vs 自转头生成
传统AI模子(比如图像分类、狡计检测)的推理是一次性的:输入一张图,模子推断一次,输出一个扫尾。系数这个词经由是固定长度的推断,输入有多大,推断量就有多大。
大模子(比如GPT系列)的推理十足不同。它是自转头生成的:模子一次只生成一个token(大约0.75个英文单词),然后把重生成的token拼接到输入中,再生成下一个token。生成一段100个token的文书,模子要跑100次。
这意味着什么?生成一个短文书的推断量,是图像分类的几十倍甚而上百倍。而况跟着对话变长,推断量线性增长——聊得越久,越慢。
1.2 推理的两个阶段:Prefill和Decoding
大模子的每一次生成,不错拆成两个阶段:
Prefill阶段(预填充):模子读取用户输入的prompt,并行推断系数token的抽象力矩阵。这个阶段推断密集,但只作念一次。
Decoding阶段(逐词生成):模子一个一个地生成新token,每生成一个,都要重新推断现时token与之前系数token的抽象力。这个阶段是内存密集型的——主要的瓶颈不是算力,而是把模子权重从显存搬到推断单位的速率。
用一个比方来持续:Prefill像是“阅读持续”——模子先把你的问题完整看一遍;Decoding像是“逐词文书”——每说一个字,都要追想一下前边说过的系数字。文书越长,追想的使命量越大。
1.3 KV Cache:用内存换时分
为了惩办Decoding阶段重迭推断的问题,业界引入了KV Cache(键值缓存)。
旨趣很浅易:在Prefill阶段推断好的抽象力键值对,缓存起来,Decoding阶段平直复用,不需要重新推断。这是一种典型的“用空间换时分”——就义显存,疏导更快的生成速率。
KV Cache的代价不小。以LLaMA-7B模子为例,生成2048个token时,KV Cache大约占用1-2GB显存。要是同期处理多个肯求(比如同期做事10个用户),光是KV Cache就要吃掉10-20GB显存——这也曾是一张A10显卡的全部容量了。
KV Cache的存在,让大模子推理的显存占用酿成动态的:短对话占得少,长对话占得多。这让资源更正变得特地复杂。
二、推理延长的三大瓶颈
说浮现了旨趣,咱们再来拆解延长来自那处。
2.1 访存带宽:被忽略的罪魁首恶
好多东说念主以为大模子慢是因为推断量大。其实否则。
推断一次矩阵乘法,GPU要作念的算术运算次数是固定的。但把模子权重从显存搬到推断单位(SM)这个经由,耗尽的时分每每比推断自身还多。这是因为显存带宽的增长速率远低于算力的增长速率。
用数字语言:一张NVIDIA A100显卡的算力是312 TFLOPS(每秒312万亿次浮点运算),但显存带宽唯一1.5 TB/s。关于LLaMA-7B(约70亿参数),每生成一个token,需要把系数这个词模子权重从显存中读取一遍。70亿个FP16参数占用14GB显存,读取一次需要约9毫秒——这还没运行算,光是把权重搬过来就花了9毫秒。
这即是为什么大模子推理被称为访存密集型任务:瓶颈不在推断,在搬运数据。
2.2 动态批处理的量度
为了擢升抵赖量,推理系统会使用动态批处理:把多个用户的肯求攒在一齐,一次性提交给GPU推断。
这样作念的平正是减少GPU的欢然时分——GPU的并行才气很强,一次算1个请乞降一次算8个肯求,时分差不了太多。坏处是:攒肯求的经由需要恭候,会让单次肯求的延长增多。
批处理大小
单肯求延长
举座抵赖量
1
50ms
20 req/s
8
65ms
123 req/s
32
120ms
267 req/s
64
210ms
305 req/s
从数据不错看出:批处理大小从1增多到8,抵赖量擢升了6倍,百人牛牛电子app官方版延长只增多了30%;但从8增多到64,抵赖量只擢升了2.5倍,延长却翻了3倍。这是一个需要良好调优的量度。
2.3 变长序列的处理遵守
用户的输入长度是随即的——有的只问一句话,有的贴一篇论文。GPU对这种变长序列的处理遵守很低。
原因在于GPU的并行推断模子:它条件系数肯求的推断格式一致。处理变长序列时,系统会把系数肯求填充(padding)到兼并个长度,短的肯求后头补上无效数据。这会导致大量算力浮滥在填凑数据上。
极点情况下,9个短请乞降1个长肯求一齐批处理,推断量可能比单寂寥理10个长肯求还大——因为填充带来了雄伟的浮滥。

三、业界的主流优化决议
濒临这些窘境,学术界和工业界提议了一系列惩办决议。
3.1 量化:让模子变“轻”
量化是面前最熟悉、最灵验的加快技巧。中枢想想:把模子权重从高精度(FP16)更正成低精度(INT8、INT4)。
FP16的每个数值用16位示意,INT8只用8位——体积收缩一半,INT4收缩到四分之一。体积变小意味着:显存占用减少、访存时分质问、推理速率变快。
量化的代价是精度亏本。好音讯是,当代量化技能(如GPTQ、AWQ)不错把精度亏本实当今0.5%-1%以内,关于大多数应用场景十足不错收受。
实测数据败露,INT8量化后的LLaMA-7B模子,推理速率擢升约2倍,显存占用减少50%;INT4量化的速率擢升约3-4倍,显存占用减少75%。
3.2 FlashAttention:IO感知的抽象力算法
轨范的Attention推断需要把系数这个词抽象力矩阵(序列长度×序列长度)写入显存再读出。当序列很永劫(比如处理一篇长文档),这个矩阵可能大到几十GB,远超显存容量。
FlashAttention的中枢知悉是:为什么不服直在SRAM(片上高速缓存)里推断抽象力,省却写入显存的经由?
FlashAttention通过分块推断和重排序,把抽象力矩阵的推断拆成多个小块,每个小块十足在SRAM内完成,不需要中间扫尾写入显存。后果惊东说念主:在长序列场景下,FlashAttention比轨范Attention快2-4倍,显存占用从二次方降到线性。
面前FlashAttention也曾成为大模子推理的事实轨范,主流的推理框架(vLLM、TensorRT-LLM)都内置了这项技能。
3.3 PagedAttention:操作系统的灵感
vLLM提议的PagedAttention模仿了操作系统的臆造内存想想。
传统决议的KV Cache是一语气存储的——每个肯求的KV Cache占用一块一语气的内存空间。当肯求长度变化时,需要常常地分拨、开释、迁移内存,导致显存碎屑化,运用率畴昔在60%-70%。
PagedAttention把KV Cache分红固定大小的“页”(畴昔16KB或64KB),不条件一语气存储。这带来了两个平正:显存运用率擢升到90%以上,不错零拷贝地分享宇宙前缀(比如系统领导词)。
实测中,vLLM的抵赖量是传统决议的10-20倍——这不是渐进式更正,是数目级的颠覆。
3.4 推测解码:用“小灵巧”换速率
这是最反直观的优化:用一个小模子来帮大模子“猜词”。
旨趣如下:小模子(比如参数目唯一1亿)生成速率很快,但质地一般。大模子(参数目100亿)质地高,但生成慢。推测解码让两个模子协同使命——小模子先快速生成多少个候选token,大模子一次性考证这些token是否正确。
因为大模子考证一批token的推断量和生成一个token差未几,举座速率就上来了。在代码生成等笃定性较强的场景,推测解码不错将推理速率擢升2-3倍。
四、不同场景的选型建议
大模子推理莫得放之四海王人准的决议,决策需要基于具体场景。
场景
延长条件
抵赖量条件
推选决议
及时对话机器东说念主
中
INT4量化 + FlashAttention + 小批处理
离线批量处理
不解锐
极高
INT8量化 + 大量处理 + PagedAttention
长文档选录
中等
低
FlashAttention + 推测解码
角落确立部署
低
INT4/INT8量化 + 小模子蒸馏
要是你的场景是及时对话:优先保证延长,遴荐小批处理(batch size 4-8),协作INT4量化和FlashAttention。
要是你的场景是离线批处理:优先保证抵赖量,遴荐大量处理(batch size 32-64),协作PagedAttention提高显存运用率。
要是你的场景是长凹凸文(比如处理几十页的PDF):FlashAttention是必备技能,PagedAttention也能匡助经管动态增长的KV Cache。

五、总结与预测
大模子推理的“临了一公里”问题,实质上是一个系统工程问题——不是模子不够强,而是怎么让它在执行场景中跑得又快又低廉。
面前业界的优化标的正在从“单一技能破损”转向“全栈协同优化”:
算法层:量化和寥落化在握续演进,1-bit量化也曾运行投入实用阶段
系统层:PagedAttention独创了新的想路,改日可能会出现更多模仿操作系统想象的技能
硬件层:GPU厂商运行在芯片中集成特意的Attention推断单位,推理速率有望再擢升一个数目级
关于招引者和企业来说,采用推理决议时应该记着三个原则:
先用最浅易的决议跑通,不要过早优化
识别委果的瓶颈——是访存带宽、推断才气,照旧显存容量?
系统性评估——延长、抵赖、老本三者之间的量度,莫得完好的决议,唯一最相宜的弃取
回到起原的问题:为什么AI很灵巧,但用起来很慢?
牛牛游戏中国2026世界杯官网谜底不是“算力不够”,而是“咱们还在学习如何让灵巧的大脑跑得更快”。大模子从“能用”到“好用”百人牛牛电子app官方版,需要的不是更灵巧的模子,而是更灵巧的工程。