互联网/IT行业面试题更新 2026-08-05
请讲解 FlashAttention 的原理。
360集团人工智能互联网/IT技术原理
考察说明
考察对高效注意力机制的理解,包括 IO 感知、分块计算与重计算等关键设计
回答思路
- 准确解释 FlashAttention 的核心动机:减少 HBM 读写、提升带宽利用率
- 说明分块(tiling)计算注意力以及局部 softmax 的增量更新
- 解释后向传播中的重计算(recomputation)技术
- 提及 MHA/GQA 的变体或与标准 attention 的复杂度对比
本题附完整参考答案与评分标准
登录后可查看结构化答题指导;也可以直接开一场模拟面试,AI 面试官用本题实时追问并给出评分。