arXiv:2601.21766cs.CLcs.AI2026-01中稿 · ICML

用连分数设计新架构,参数减半性能不降

CoFrGeNet: Continued Fraction Architectures for Language Generation

  • 基于连分数构造新函数类,替代注意力与前馈网络
  • 参数减少一半至三分之二,下游任务表现相当或更优
  • 可直接替换现有Transformer,适合工业部署

Transformers是语言生成的主流架构。本文受连分数启发,提出一种新的生成建模函数类,并构建了名为CoFrGeNet(连分数生成网络)的架构家族。其组件可替代Transformer块中的多头注意力和前馈网络,参数量大幅减少。我们设计了专用梯度计算方式,优化效率优于标准PyTorch梯度。该组件为即插即用式替换,无需修改训练或推理流程,便于集成到大型工业工作流中。我们在GPT2-xl(15亿参数)和Llama3(32亿参数)两个不同架构上进行实验:前者在OpenWebText和GneissWeb上预训练,后者在包含九个数据集的docling数据混合集上预训练。结果表明,模型在下游分类、问答、推理和文本理解任务上的表现与原模型相当,甚至更优,且参数量仅为原模型的1/2至2/3,预训练时间更短。我们认为未来针对硬件定制化实现将进一步释放其潜力。

原文摘要 · Abstract (English)

Transformers are arguably the preferred architecture for language generation. In this paper, inspired by continued fractions, we introduce a new function class for generative modeling. The architecture family implementing this function class is named CoFrGeNets - Continued Fraction Generative Networks. We design novel architectural components based on this function class that can replace Multi-head Attention and Feed-Forward Networks in Transformer blocks while requiring much fewer parameters. We derive custom gradient formulations to optimize the proposed components more accurately and efficiently than using standard PyTorch-based gradients. Our components are a plug-in replacement requiring little change in training or inference procedures that have already been put in place for Transformer-based models thus making our approach easy to incorporate in large industrial workflows. We experiment on two very different transformer architectures GPT2-xl (1.5B) and Llama3 (3.2B), where the former we pre-train on OpenWebText and GneissWeb, while the latter we pre-train on the docling data mix which consists of nine different datasets. Results show that the performance on downstream classification, Q\& A, reasoning and text understanding tasks of our models is competitive and sometimes even superior to the original models with $\frac{2}{3}$ to $\frac{1}{2}$ the parameters and shorter pre-training time. We believe that future implementations customized to hardware will further bring out the true potential of our architectures.

语言生成模型压缩连分数Transformer改进

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