中文

在模块加法任务上训练的循环神经网络中的低秩稀疏傅里叶结构

机器学习 2025-03-31 v1 信号处理 机器学习

摘要

模块加法任务是观察深度学习中经验现象(包括 \emph{grokking} 现象)的有用基准。先前的研究表明,单层Transformer架构学习傅里叶乘法电路来解决模块加法任务。在本文中,我们表明,针对模块加法任务训练的循环神经网络(RNN)也采用傅里叶乘法策略。我们识别出模型权重中的低秩结构,并将模型组件归因于特定傅里叶频率, resulting in a稀疏的傅里叶空间表示。我们还表明,RNN对去除单个频率具有鲁棒性,但当从模型中消除更多频率时,性能会急剧下降。

关键词

引用

@article{arxiv.2503.22059,
  title  = {Low Rank and Sparse Fourier Structure in Recurrent Networks Trained on Modular Addition},
  author = {Akshay Rangamani},
  journal= {arXiv preprint arXiv:2503.22059},
  year   = {2025}
}

备注

To appear at ICASSP 2025