arXiv:2409.12924eess.SPcs.AI2024-09被引 1

用小波思想改造GPT,提速近两倍且不加参数。

Wavelet GPT: Wavelet Inspired Large Language Models

  • 在预训练中引入小波结构,控制中间表示的多尺度特性。
  • 相同训练步数下性能显著提升,相当于更大模型效果。
  • 适用于文本、音频、图像等多类输入,支持多种粒度输入。

大型语言模型(LLMs)推动了人工智能在各个科学领域的进步。我们所处的数据世界——如文本、音频和音乐——普遍具有多尺度结构。本文在预训练阶段将传统信号处理中的小波思想融入GPT风格模型,无需增加任何额外参数,在学术设置下实现了文本、音频和图像任务上几乎两倍的预训练速度。通过在中间嵌入上施加结构化约束,使每个解码器块中的下一个词预测能访问不同时间分辨率的中间表示。在相同训练步数下,性能显著提升,接近更大神经网络的预训练效果。该方法还扩展至长程任务基准(Long Range Arena)及多种输入形式,包括字符、BPE标记、字节、波形、数学表达式和图像像素。本架构为预训练中融合多速率信号处理提供了新路径。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have ushered in a new wave of artificial intelligence advancements impacting every scientific field and discipline. We live in a world where most of the data around us, e.g., text, audio, and music, has a multi-scale structure. This paper infuses LLMs with a traditional signal processing idea, namely wavelets, during pre-training to take advantage of the structure. Without adding \textbf{any extra parameters} to a GPT-style LLM architecture in an academic setup, we achieve the same pre-training performance almost twice as fast in text, audio, and images. This is done by imposing a structure on intermediate embeddings. When trained for the same number of training steps, we achieve significant gains in performance, which is comparable to pre-training a larger neural architecture. Further, we show this extends to the Long Range Arena benchmark and several input representations such as characters, BPE tokens, bytes, waveform, math expression, and image pixels. Our architecture allows every next token prediction access to intermediate embeddings at different temporal resolutions in every decoder block. We hope this will pave the way for incorporating multi-rate signal processing into pre-training.

小波语言模型高效训练多尺度

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