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