arXiv:2503.13427cs.LGcs.AI2025-03ICML被引 17

xLSTM 7B实现70亿参数模型的快速高效推理,速度超越同类模型。

xLSTM 7B: A Recurrent LLM for Fast and Efficient Inference

  • 采用xLSTM架构结合针对性优化,实现线性计算扩展与恒定内存占用。
  • 在下游任务中表现接近同类模型,推理速度比Llama和Mamba快30%以上。
  • 适合需要大量推理计算的任务,开源代码与权重支持研究复现。

近期大型语言模型(LLMs)在解决推理、数学和编程问题上的突破,依赖于推理阶段的大量计算资源投入。因此,推理速度是LLM架构的关键指标,亟需高效快速的模型。最近,基于xLSTM架构的LLMs成为Transformer的有力替代,具备序列长度的线性计算扩展和恒定内存使用,极有利于高效推理。然而,此类xLSTM模型尚未被扩展至更大规模,也缺乏与主流模型在推理速度和效率上的系统评估。本文提出xLSTM 7B,一个70亿参数的LLM,融合xLSTM架构优势与针对推理加速的专项优化。实验表明,xLSTM 7B在下游任务上性能与同类模型相当,但推理速度显著优于Llama和Mamba基线模型,效率提升超30%。结果确立了xLSTM 7B为当前最快最高效的7B LLM,为高测试时计算需求的任务提供解决方案。本工作凸显xLSTM作为大规模推理方法基础架构的潜力。模型权重、代码与训练脚本均已开源。

原文摘要 · Abstract (English)

Recent breakthroughs in solving reasoning, math and coding problems with Large Language Models (LLMs) have been enabled by investing substantial computation budgets at inference time. Therefore, inference speed is one of the most critical properties of LLM architectures, and there is a growing need for LLMs that are efficient and fast at inference. Recently, LLMs built on the xLSTM architecture have emerged as a powerful alternative to Transformers, offering linear compute scaling with sequence length and constant memory usage, both highly desirable properties for efficient inference. However, such xLSTM-based LLMs have yet to be scaled to larger models and assessed and compared with respect to inference speed and efficiency. In this work, we introduce xLSTM 7B, a 7-billion-parameter LLM that combines xLSTM's architectural benefits with targeted optimizations for fast and efficient inference. Our experiments demonstrate that xLSTM 7B achieves performance on downstream tasks comparable to other similar-sized LLMs, while providing significantly faster inference speeds and greater efficiency compared to Llama- and Mamba-based LLMs. These results establish xLSTM 7B as the fastest and most efficient 7B LLM, offering a solution for tasks that require large amounts of test-time computation. Our work highlights xLSTM's potential as a foundational architecture for methods building on heavy use of LLM inference. Our model weights, model code and training code are open-source.

xLSTM推理效率7B模型开源

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