跳到正文
原文
Hugging Face Blog·· 2021-03-31AI 评分45

理解 BigBird 的块稀疏注意力

Understanding BigBird's Block Sparse Attention

AI 导读

BigBird 用块稀疏注意力替代 BERT 的全注意力,可处理最长 4096 的序列,计算成本远低于 BERT,其 RoBERTa 类模型已在 🤗Transformers 中可用。其注意力由三部分近似:捕获邻近依赖的滑动注意力、被所有 token 关注的全局 token,以及随机选择的随机 token,以此在长序列上高效逼近全注意力。该模型在长文档摘要、长上下文问答等任务上取得 SOTA。

来源:Hugging Face Blog · huggingface.co