arXiv:2508.00544cs.CLcs.LG2025-08被引 1

用并行路径训练语言模型,数小时完成,性能还更好。

PaPaformer: Language Model from Pre-trained Parallel Paths

  • 将模型拆成低维并行路径,分训后合并
  • 训练时间从数天缩短至数小时,参数量可减少
  • 适合需快速定制任务的场景,如小模型部署

现代大语言模型的训练需要大量计算资源和时间。即使是小型语言模型(SLMs),在理想情况下也需数天训练,通常依赖多块GPU。本文提出一种名为PaPaformer的解码器仅有的Transformer架构变体,其低维并行路径可分别使用不同数据训练后合并为更大模型。该方法可在不增加总参数量的前提下,显著缩短训练时间,并提升性能。同时,平行路径结构为针对特定任务定制模型提供了新思路。

原文摘要 · Abstract (English)

The training of modern large-language models requires an increasingly amount of computation power and time. Even smaller variants, such as small-language models (SLMs), take several days to train in the best-case scenarios, often requiring multiple GPUs. This paper explores methods to train and evaluate decoder-only transformer-based language models in hours instead of days/weeks. We introduces \textit{PaPaformer}, a decoder-only transformer architecture variant, whose lower-dimensional parallel paths are combined into larger model. The paper shows that these lower-dimensional paths can be trained individually with different types of training data and then combined into one larger model. This method gives the option to reduce the total number of model parameters and the training time with increasing performance. Moreover, the use of parallel path structure opens interesting possibilities to customize paths to accommodate specific task requirements.

语言模型并行训练高效训练架构创新

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