返回信息流
新闻事件
InfoQ AI/ML/Data Eng
1 个来源

Shopify 推出 Gisting:将 LLM 系统提示压缩为学习令牌

Shopify 的工程团队推出了 gisting,这是一种将长 LLM 系统提示压缩为更少的学习“要点”令牌的新技术,从而提高吞吐量并降低推理成本。这一方法有望提升 LLM 部署的效率。

SynthePulse Insight · AI 深度解读

Shopify 推出 Gisting:将长系统提示压缩为可学习令牌,推理成本与延迟双降

版本 1 · 1 个来源

Shopify 工程团队引入 Gisting 技术,将冗长的 LLM 系统提示压缩为少量可学习的“gist”令牌,在不修改模型核心权重的前提下,显著降低延迟、提升吞吐并减少 GPU 分配。

  • Gisting 将 Sidekick GraphQL agent 的系统提示从约 6000 个令牌压缩至 1500 个 gist 令牌,上下文规模缩减 4:1,且不牺牲预测质量。
  • 在 350 RPM 负载下,TTFT 中位数从 438ms 降至 354ms,端到端延迟中位数从 6.8s 降至 4.2s,QPS 从 20.2 提升至 23.4。
  • Gisting 基于 2022 年论文中的技术,通过教师-学生两阶段过程训练 gist 嵌入,最小化 KL 散度。
  • 训练完成后,gist 嵌入直接写入模型嵌入矩阵,并注册为特殊令牌,推理时无需自定义注意力掩码或额外编码器。
  • Gisting 与前缀缓存互补,两者结合可进一步减少解码阶段对 KV 张量的处理开销。
展开章节目录Gisting 的核心机制

Gisting 的核心机制

Gisting 的核心思想是将冗长的系统提示压缩为一组更短的“gist”令牌,这些令牌是模型可学习的嵌入表示。与传统的提示摘要不同,gist 令牌并非对原始提示的概括,而是经过训练、旨在让模型行为尽可能接近其看到原始提示时的表现。

训练过程分为两步:教师阶段,模型使用真实提示运行,得到响应的教师 logits;学生阶段,模型使用 gist 令牌运行,得到学生 logits。随后通过最小化教师与学生 logits 之间的 KL 散度来训练 gist 令牌,直到学生的预测与教师紧密匹配。

实际效果与性能数据

Shopify 报告称,在其 Sidekick GraphQL agent 上应用 Gisting 后,系统提示从约 6000 个令牌压缩至 1500 个 gist 令牌,上下文规模缩减 4:1,且未牺牲预测质量。

在 350 RPM 的负载下,TTFT 中位数从 438ms 降至 354ms,端到端请求延迟中位数从 6.8s 降至 4.2s,吞吐量从 20.2 QPS 提升至 23.4 QPS。这些改进使 Shopify 能够减少分配的 GPU 数量。

部署与兼容性

训练完成后,gist 嵌入被直接写入模型的嵌入矩阵,并将新的 gist 令牌注册为分词器中的特殊令牌。推理时,模型像往常一样加载和运行,无需自定义注意力掩码、额外编码器或特殊服务路径。

Gisting 与前缀缓存等优化技术互补。前缀缓存避免了为缓存的提示序列重新计算 KV 张量,但模型在解码阶段仍需处理这些缓存张量;Gisting 通过用更短的 gist 令牌序列替换长提示,进一步减少了这一开销。Shopify 将两者结合使用,效果叠加。

来源与不确定性

本文信息主要来自 InfoQ 对 Shopify 工程博客的报道,属于二手来源。报道中引用了 Shopify 的官方声明和具体数据,但未提供原始博客链接或独立验证。

报道提到 Gisting 基于 2022 年论文《Prompt Compression and Contrastive Conditioning for Controllability and Toxicity Reduction in Language Models》,但未提供论文细节。此外,报道提及“autosearch”在调优 Gisting 过程中的作用,但未展开说明,具体实现细节需参考原始文章。

可信度边界

本报道基于 InfoQ 对 Shopify 工程博客的转述,所有性能数据和声明均来自 Shopify 官方,未经第三方独立验证。数据为特定负载下的测量结果,实际效果可能因场景而异。

核心结论

Gisting 通过将长提示压缩为可学习令牌,在保持模型质量的同时显著降低推理延迟和成本,且与现有优化技术兼容,为 LLM 推理效率提供了一种实用方案。

主报道

InfoQ AI/ML/Data Eng

主报道来源