跳到正文
原文
Hugging Face Blog·· 2022-08-02AI 评分28

Nyströmformer:通过Nyström方法实现线性时间与内存的自注意力近似

Nyströmformer: Approximating self-attention in linear time and memory via the Nyström method

AI 导读

Nyströmformer利用Nyström方法将Transformer自注意力的复杂度从O(n²)降至O(n)。该模型通过查询和键的段均值采样32或64个landmarks来近似softmax矩阵,在长序列(n=4096或8192)上保持与标准自注意力相当的竞争力性能。

来源:Hugging Face Blog · huggingface.co