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