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

BigBird 块稀疏注意力机制详解

Understanding BigBird's Block Sparse Attention

AI 导读

BigBird 采用块稀疏注意力替代 BERT 的全注意力,将序列长度支持提升至 4096 并显著降低计算成本。该模型通过滑动窗口、全局 token 和随机 block 三种机制近似全注意力矩阵,在长文档摘要等任务中实现 SOTA。BigBird RoBERTa-like 模型现已集成至 Hugging Face Transformers 库。

来源:Hugging Face Blog · huggingface.co