1ffb8
2025/05/01 16:54:01 8,666次浏览

本篇工作已被电子设计自动化领域顶级会议 DAC 2025 接收,由上海交大计算机学院蒋力教授与刘方鑫助理教授带领的 IMPACT 课题组完成,同时也获得了华为 2012 实验室和上海期智研究院的支持。第一作者是博士生汪宗武与硕士生许鹏。

在通用人工智能的黎明时刻,大语言模型被越来越多地应用到复杂任务中,虽然展现出了巨大的潜力和价值,但对计算和存储资源也提出了前所未有的挑战。在以 transformer 模型为基础的大模型中,键值缓存虽然用以存代算的思想显著加速了推理速度,但在长上下文场景中成为了存储瓶颈。例如,半精度的 LLaMA-2-7B 模型权重约 14GB,在上下文长度为 128K 时键值缓存占据 64GB,总和已经接近高端卡 NVIDIA A100 的 80GB 显存容量上限。键值量化可被用于压缩缓存,但往往受到异常值的干扰,导致模型性能的显著下降。为此,本文的研究者提出了 MILLION,一种基于乘积量化的键值缓存压缩和推理加速设计。

整型量化的软肋:异常值

图 1:矩阵量化可视化。红色代表的异常值显著大于其他值,导致均匀量化后高位编码被浪费。

量化中受到广泛使用的整型均匀量化受到异常值的影响较为显著。图 1 展示了矩阵中的量化。在一组分布较为集中的数据中,一个显著偏离其他值的异常值会导致其他值的量化结果全部落在较低区间,浪费了高位编码的表示能力。

图 2:图中使用 「通道熵」 定量比较不同方案的量化效果,越大表明越有效地利用了通道容量,即整型的宽度。沿通道量化只能解决沿该方向分布的异常值,而在面对另一方向异常值时效果不佳。

在实际的键值量化中,为了更好的表示能力,通常对于每个通道(即键值向量的维度)或每个 token 采取不同的量化参数,这种方法被称为 沿通道量化 (channel-wise quantization)或 沿词元量化 (token-wise quantization)。然而,如图 2 所示,沿特定方向量化只能解决沿该方向分布的异常值。

图 3:实际采样获得的键值缓存分布。在 llama-2-7b-wikitext-103-v1-layer10-value 中,异常值并不遵循简单的沿通道分布,而是呈现为较复杂的点状和团状。

研究团队通过实际采样数据发现,在键值缓存中,沿通道方向分布的异常值占多数,但也存在并不明显的情况,如图 3 所示。这表明,上述量化方案并不是一劳永逸的解决方式,仍然存在优化空间。

异常值的解决方案:乘积量化

图 4:数轴上的均匀和非均匀量化对比。在对 8 个数据点进行 2 比特量化过程中,均匀量化浪费了 10 编码。而基于聚类的非均匀量化则编码更合理。

如图 4 所示,非均匀量化通过聚类的方式允许量化区间不等长,从而更合理地分配编码,提升量化效率。研究团队观察到,由于通道间的数据分布可能存在关联(即互信息非负),将通道融合后在向量空间中聚类,效果一定不亚于独立通道的量化,如图 5 所示。

图 5:左图为两个通道独立进行 1 比特量化,右图为在通道融合后进行 4 分类的 KMeans 聚类。融合通道量化的通道熵更加接近 2 比特的容量极限,展示出更好的量化效果。

由于高维空间中聚类较为困难,因此将整个向量空间划分为多个低维子空间的笛卡尔积,可以平衡聚类复杂度和量化效果。这与最近邻搜索中使用的乘积量化思想一致。研究团队通过实验发现,子空间维度为 2 或 4 是较好的平衡点。

推理加速手段:高效的系统和算子实现

图 6:三阶段的推理系统设计

图 7:分块注意力机制使得批量延迟量化成为可能

图 6 展示了离线训练、在线预填充、在线解码三阶段的量化推理系统设计。其中,码本训练(量化校准)属于秒级轻量化任务,并且离线进行,不影响运行时开销;在线预填充阶段使用训练好的码本对键值缓存进行量化压缩,达到节省显存的目的;在线解码阶段采用分块注意力机制的方法,将预填充阶段的历史注意力和生成 token 的自注意力分开计算(如图 7 所示),达成 批量延迟量化 的目的,掩藏了在线量化的开销,确保模型输出的高速性。并且,在历史注意力阶段,由于历史键值对数远大于码本长度,因此先用查询向量与码本计算好非对称距离查找表(ad-LUT),可以大大减少内积距离计算量,达到加速计算的目的。

图 8:向量化加载可有效使带宽饱和

在算子优化方面,研究团队在 flash decoding 的基础上使用了宽数据(如 float4)向量化加载的方式,将多个乘积量化编码打包为宽数据,有效使带宽饱和(如图 8 所示)。同时,在表查找阶段,子空间之间的表具有独立性,并且可以被放入少量缓存行中,研究团队利用这一空间局部性极大提高了表查找的 L2 缓存命中率。此外,研究团队还仔细扫描了不同上下文长度下可能的内核参数,找到最优配置,形成了细粒度的预设,在实际运行时动态调整,充分利用 GPU 的计算资源。具体实现可以在开源仓库中找到。

实验结果

实验设置

图 9:实验设置

实验采用了不同位置编码、不同上下文长度的多种模型进行了详细的评估。在模型性能方面,采用困惑度(Perplexity,PPL)和 Longbench 两种指标;在系统性能方面,采用每词元输出间隔(Time Per Output Token, TPOT)定量分析,并给出了注意力层详细的剖析。对比采用方案和乘积量化参数如图 9 所示。

模型性能

图 10:困惑度指标。其中 「-1%」 表示该方法额外存储 1% 的异常值不参与量化。

困惑度越小表明模型输出质量越高。实验结果表明,MILLION 与额外处理了异常值的 SOTA 方案输出质量保持一致,展现出对异常值良好的鲁棒性。而 SOTA 方案在不处理异常值的情况下可能会遭遇严重的输出质量损失。

图 11:Longbench 问答数据集得分展示

在长对话问答任务中,不同模型在各种数据集上的得分均表明,MILLION 方案能够在 4 倍键值缓存压缩效率下保持几乎无损的表现。

系统性能

图 12:每词元输出时间。对比其他方案,MILLION 的优势持续增长,在 32K 上下文时达到 2 倍加速比。

图 13:注意力层时间剖析

在 TPOT 评估中,MILLION 能够在 32K 上下文语境下同时达成 4 倍键值缓存压缩比和 2 倍端到端加速比。注意力层的深入分析表明,MILLION 在访存和内核函数方面对比 baseline 取得显著优势。

总结

MILLION 的主要贡献在于:(1)深入分析键值缓存分布;(2)提出基于乘积量化的非均匀量化算法;(3)设计高效的推理系统及内核。研究团队首先证实了键值缓存中异常值存在的普遍性,并指出异常值的不同分布是当前主流的量化方案精度不足的根本原因;然后提出通过将高维向量空间分解为多个子空间,并在每个子空间内独立进行向量量化的方法,更有效地利用了通道间的互信息,并且对异常值展现出极强的鲁棒性;接着通过 CUDA 异步流和高效的算子设计,充分利用了 GPU 的并行计算能力和内存层次结构,以支持乘积量化的高效执行。实验表明,对比主流框架 transformers 的半精度实现,MILLION 在 32K 上下文场景中同时达成 4 倍压缩率和 2 倍加速比,并且在多种语言任务中精度表现几乎无损。

星空天美麻花视频大全

上交大等探索键值压缩的边界:惭滨尝尝滨翱狈框架定义模型量化推理范式
上交大等探索键值压缩的边界:惭滨尝尝滨翱狈框架定义模型量化推理范式

我可算把五一盼来了

「活动」办耻尘补诲补颈产颈补辞迟颈迟别蝉迟

4.31MB
版本痴8.0.14
下载女性㊙️免费看安装你想要的应用 更方便 更快捷 发现更多
喜欢 63%好评(18025人)
评论 16
上交大等探索键值压缩的边界:惭滨尝尝滨翱狈框架定义模型量化推理范式截图0 上交大等探索键值压缩的边界:惭滨尝尝滨翱狈框架定义模型量化推理范式截图1 上交大等探索键值压缩的边界:惭滨尝尝滨翱狈框架定义模型量化推理范式截图2 上交大等探索键值压缩的边界:惭滨尝尝滨翱狈框架定义模型量化推理范式截图3 上交大等探索键值压缩的边界:惭滨尝尝滨翱狈框架定义模型量化推理范式截图4
详细信息
  • 软件大小: 88489.22805MB
  • 最后更新: 2025/05/01 16:54:01
  • 最新版本: V5.86234.2
  • 文件格式: apk
  • 应用分类:颈辞蝉-础苍诲谤辞颈诲 男人露jiji㊙️免费网站推特
  • 使用语言: 中文
  • : 需要联网
  • 系统要求: 1.55968以上
应用介绍
一,男C女黄🔞️B站,黄油羞涩~e❌漫画网页
二,XXXXXXX泡妞🍆🍑睡着视频,被❌水好多18久久久爽白浆高潮
叁,日本❌❌奶头❌❌屁股蓝莓视频,小🐥🐥伸入🍑自慰91日本
四,美女裸体捆绑秘密㊙️网站,动漫❌触手❌3d❌网站
五,💚奇优影院在线观看庆余年,无遮挡🔞漫画网站
六,英语老师趴下让我❌了一夜视频,亚洲精品🔞🔞🔞男同Gay,❤uoco❤福利姬图库
七,校花露出小🐔🐔
【联系我们】
客服热线:134-2881-646
加载更多
版本更新
V2.2.16
美女被❌到高潮喷出www动漫
  • 爆❌喷水❌洗澡❌动漫蘑菇

    兔兔美女❌❌羞羞视频

    穿JK白丝被❌裸体写真

    86613.63MB
    105好评

    哎呦绅士♥为了您的身体健康请适度

  • FC2PPV初撮り♥️完全颜出

    性按摩❌❌❌❌❌

    蜜桃臂被❌❌❌羞羞免费网站

    32583.36MB
    050010好评

    69XX网站美女㊙️露双乳

  • www深夜成人🔞网站视频漫画

    动漫男男裸体啪啪❌漫画原神

    日本❌❌❌❌❌色情80

    13.27MB
    316好评

    免费JK黄漫❌18禁漫画

  • 蜜桃av㊙️无码一区二区三欧

    BRAZZERS💋

    免费🔞🔞🔞网战春水

    53.63MB
    279好评

    男男做爰黄✌片免费观看软禁果

  • 91♥️丨PORNY地址

    naruto❌sakura无尽

    迪士尼公主被❌到爽流动漫

    71.14MB
    704好评

    美女隐私黄㊙️www网站

  • 美女裸体❌玉足大胸网站

    国产91❤丝袜在线播

    扒下女教师内衣❌❌动漫

    83.23MB
    333好评

    国产做受69❌❌❌高潮漫画

  • 白丝美女看裸又爽❌又黄的应用

    美女cos被爆❌出水娇喘呻吟

    免费无遮挡🔞漫画网站在线观看

    23.23MB
    101好评

    AI女明星被❌高清网站

  • 爆🌿出水🔞

    裸体裸乳被🍑免费看漫画

    欧美成人⭕⭕⭕⭕××××

    50.75MB
    924好评

    雷电将军爆乳18禁🔞动漫视频

  • 美女裸体🍌戳🍑代码

    男模隐私㊙️视频无内裤

    美女露出🐻让男生揉的直播

    83.77MB
    564好评

    禁漫♥天堂♥

  • 美女露出㊙️胸奶头咬

    Naruto❌Sakura18禁漫天堂

    国产Chinese男男GayGay✅

    83.81MB
    750好评

    免费看❌❌❌女生

  • 迪丽热巴赵露思被❌免费网站

    挠痒痒㊙️网站免费

    踢裆免费网站❌

    15.52MB
    9295好评

    美女少萝❌❌爆乳❌❌萌白酱

  • 美杜莎女王被❌黄同人图集

    宅男的秘密㊙️秘密基地

    学生裸体黄❌❌网站视频

    11.69MB
    625好评

    ❌吸乳18禁羞羞动漫

  • ❤国产精品嫩草影院❤

    男生的🍌伸进人🍑打下扑

    男人的🍌伸到🍑涩涩的视频

    65.11MB
    741好评

    A🔞V看🔴䏒🔞❌❌

  • 他在教室拿大🐔巴🌿我嗯好爽

    免费看美女裸体🔞🔞🔞软仵

    国产➕又黄➕又爽刺激言情书视频

    54.66MB
    755好评

    玖辛奈被爆❌羞羞漫画视频

  • FerrPorno💋👙3k

    17c黑料爆料18🈲在线观看

    中国裸体㊙️洗澡全过程

    57.42MB
    687好评

    程潇裸乳被❌到爆浆

  • 小医仙扒开🐻让我❌❌❌小说

    韩国美女19➕vip夏娃

    亚洲性⭕⭕⭕⭕XXXX

    24.62MB
    352好评

    国产精品㊙️密A片

  • ㊙️无码一区二区三黑人

    老师美女被❌吸乳得到大胸动漫

    小🐔🐔放进

    13.93MB
    311好评

    男性下部㊙️免费看

  • 男裸体全身㊙️无遮挡

    纲手被❌吸乳脱内动漫

    美女被❌到爽🔞高潮蓝莓视频

    84.69MB
    213好评

    十八禁🔞爱情岛网站

  • 男生被c🔞黄㊙️视频

    91娇喘❌❌⭕⭕白丝电影

    韩国无码成人🔞电影全集

    45.06MB
    739好评

    扒男人❌狂揉❌难受亲嘴视频

  • 小舞❌狂揉❌难受3

    玖辛奈被爆❌羞羞漫画26

    男模裸体㊙️无遮挡网站

    10.74MB
    184好评

    欧美熟妇❌❌❌喷水

  • 扒开❌狂揉❌大胸

    赫敏裸被❌视频站

    雷电将军裸体❌开腿涩涩

    44.41MB
    365好评

    日本变态浣肠㊙️免费视频

  • 小🐔🐔伸进🈲男人🔞🔞猛撞

    国产精品18🈲高潮网站摸应用

    女同被❌到爽了玩具

    91.03MB
    208好评

    идео⚡️HDⅩXXX19

  • 捆绑裸体美女网站㊙️

    91丨色丨国产熟女❤蘑菇

    女性下部㊙️免费看

    30.03MB
    908好评

    女仆被❌到爽🔞流片动漫视频

  • 撕开👙吸奶头🍌进🍑网站

    美女裸体㊙️果冻网站

    亲亲~操~🌿~🍀

    30.37MB
    546好评

    美女被捆绑㊙️网站

  • sm调教圈论坛⭕投稿

    欧美性❌XX❌XX❌XXX

    不知火舞被❌吸乳羞羞视频

    29.25MB
    121好评

    云缨疯狂❌喷水自慰漫画

  • 快穿➕灌➕电击➕调教

    韩国同性女A片18🈲直播下载

    裸体❌❌AI去衣下载

    70.06MB
    459好评

    男生被c🔞黄㊙️小说

  • 黑土被❌娇喘动漫网站在线小🐔🐔伸进🈲🔞🔞羞羞 title="美洲熟妇色❌❌❌❌" class="0v5366j9r7414 app-list-img" src="https://t11./it/u=1040017057,165456434&fm=30&app=106&f=JPEG?w=312&h=208&s=15B07D95426079116A1494FD03009032"">

    疯狂❌自慰爽www看片双女主

    小🐔🐔伸进🈲🔞蓝莓视频

  • ai换脸吴宣仪自慰❌❌喷水好爽日韩裸体裸乳🍑被91 title="美女裸体㊙️level网站" class="0v5366j9r7414 app-list-img" src="https://t10./it/u=3070224572,165461829&fm=30&app=106&f=JPEG?w=312&h=208&s=C6000AAC4838BAC8463F54AC0300A00B"">

    不知火舞被❌18禁同人真人

    触手❌裸体❌捆绑

  • 老肥妇性裸交╳❌❌Ⅹ美女扒开内👙看个够视频 title="少萝自愿裸体❌❌" class="0v5366j9r7414 app-list-img" src="https://t12./it/u=1041590673,165100299&fm=30&app=106&f=JPEG?w=312&h=208&s=D7383EC642E1D8EA0A87937F0300C078"">

    漫画美女被狂抡18🈲️

    💚偽裝💚

69❌❌❌❌❌欧美高中学生 类似软件

猜你喜欢

包含 饿了么 的应用集
包含 饿了么 的应用集
评论
  • 小心🐤戳进去无遮挡动 4天前
    成人免费视频❌❌❌小说
  • 女人👙摸🐻亲嘴在床上蹦蹦床 2天前
    涩里番❤️成人软件
  • 女人的秘密㊙️视频网站 2天前
    小🐤🐤戳进🍑无遮挡蓝莓视频
  • 国产精密㊙️入口51 3天前
    小🐔🐔伸进禁9幺
  • 被❌到爽🔞教学 3天前
    女人脱个精光㊙️ 亲嘴
  • 八尺大人被❌吸乳视频 3天前
    ❤️嫩草影院ncyy网址
  • 91娇喘❌❌⭕⭕白丝智学网站 2天前
    动漫人物被触手❌黄漫网站
  • Hentai❌tube动漫18 1天前
    少萝扒开腿㊙️让人桶动漫
  • 夜兰裸体❌无遮挡 1天前
    纳西妲被❌到爽爆游戏
  • 国产91❤在线播放九色麻豆 5天前
    🌸✿极品国产✿🌸白皙黑料社
0