中文

对称幂变换器的共形变换

机器学习 2025-05-06 v2 人工智能

摘要

线性注意力变换器相比基于softmax的变换器具有显著的计算优势,但往往存在性能退化的问题。对称幂(sympow)变换器是一种特殊的线性变换器,通过利用对称张量嵌入来弥补部分性能差距,达到了与softmax变换器相当的性能。然而,sympow变换器中循环状态的有限容量限制了其信息保持能力,导致在扩展训练或评估上下文长度时性能下降。为了解决这一问题,我们提出了共形-sympow变换器,它通过数据依赖的乘性门控动态释放容量,并利用数据依赖的旋转位置编码自适应地存储信息。在LongCrawl64数据集上的初步实验表明,共形-sympow克服了sympow变换器的局限性,在扩展的训练和评估上下文中实现了稳健的性能。

关键词

引用

@article{arxiv.2503.03269,
  title  = {Conformal Transformations for Symmetric Power Transformers},
  author = {Saurabh Kumar and Jacob Buckman and Carles Gelada and Sean Zhang},
  journal= {arXiv preprint arXiv:2503.03269},
  year   = {2025}
}

备注

SCOPE Workshop at ICLR 2025