arXiv:2502.21309cs.CLcs.AI2025-02NeurIPS被引 1

用傅里叶分析提升大模型周期建模,增强推理能力

Reasoning is Periodicity? Improving Large Language Models Through Effective Periodicity Modeling

  • 将傅里叶分析融入注意力机制,改进周期性建模
  • 在语言建模中,大模型规模扩大时性能超越Transformer
  • 更擅长学习和应用规则,适合需要逻辑推理的任务

周期性是人类学习范式中结构化知识获取与系统认知过程的基础特征之一。然而,基于Transformer的大语言模型(LLM)在周期性建模方面存在潜在缺陷,影响其从数据中学习效率及底层规律的建立。本文提出FANformer,通过将傅里叶分析网络(FAN)引入注意力机制,改进特征投影过程,实现高效的周期性建模。大量实验表明,随着模型规模和训练令牌数量增加,FANformer在语言建模任务中持续优于Transformer,展现出更强的学习效率。预训练的FANformer-1B在下游任务中表现显著优于参数或训练量相近的开源大模型。此外,结果表明FANformer在规则学习与推理应用方面优于Transformer。FANformer为推进大语言模型发展提供了有效且有前景的架构。

原文摘要 · Abstract (English)

Periodicity, as one of the most important basic characteristics, lays the foundation for facilitating structured knowledge acquisition and systematic cognitive processes within human learning paradigms. However, the potential flaws of periodicity modeling in Transformer affect the learning efficiency and establishment of underlying principles from data for large language models (LLMs) built upon it. In this paper, we demonstrate that integrating effective periodicity modeling can improve the learning efficiency and performance of LLMs. We introduce FANformer, which adapts Fourier Analysis Network (FAN) into attention mechanism to achieve efficient periodicity modeling, by modifying the feature projection process of attention mechanism. Extensive experimental results on language modeling show that FANformer consistently outperforms Transformer when scaling up model size and training tokens, underscoring its superior learning efficiency. Our pretrained FANformer-1B exhibits marked improvements on downstream tasks compared to open-source LLMs with similar model parameters or training tokens. Moreover, we reveal that FANformer exhibits superior ability to learn and apply rules for reasoning compared to Transformer. The results position FANformer as an effective and promising architecture for advancing LLMs.

大模型周期性建模推理增强注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。