跳到主要内容
SUN / 笔记

BLOG / 766 ARTICLES

技术笔记

计算机基础、AI、后端与客户端开发笔记

AI · 154 篇文章清除筛选
5 分钟
AI

prompt-cache论文速读

在 多轮对话、DocQA 和 RAG 客服系统场景下中,提示词会包含重复的文本段落,如系统消息、提示模板和用于上下文的文档,也就是说常见文本段的注意力状态(也就是 kv 缓冲向量)\{(k_1,v_1), (k_2,v_2),..,(k_n,v_n)\} 是可以重复利用的。Prompt Cache 论文基于这一关键观察,设计了一个提示词模块提出来明确定义这些

4 分钟
AI

flashattention-3论文速读

flashattention-2 虽然减少了非矩阵乘法的计算量,提高了运行速度,但是其并未充分利用最新硬件的功能,导致在 H100 GPU 上的利用率仅为 35%,由此作者提出了 flashattention-3,提出了3 项优化策略来提升 Hopper GPU 上的注意力计算性能:

3 分钟
AI

MoE专家并行详解

在 MOE 模块中,Router 负责决定发送 token 到 指定 expert 中,在实际模型推理过程中,可能某些 experts 接收 tokens 过多,某些 experts 接收 tokens 过少,这种现象叫做 expert 负载不均衡。这种情况不仅脱离了 MoE 的设计初衷(术业有专攻),也影响了 MoE 计算效率(分布式训练推理中各卡通讯时的

页