English

Towards smaller, faster decoder-only transformers: Architectural variants and their implications

Machine Learning 2024-10-10 v4

Abstract

In recent times, the research on Large Language Models (LLMs) has grown exponentially, predominantly focusing on models underpinned by the transformer architecture, as established by [1], and further developed through the decoder-only variations by [2]. Contemporary efforts in this field primarily aim to enhance model capabilities by scaling up both the architecture and data volumes utilized during training. However, the exploration into reduce these model sizes while preserving their efficacy remains scant. In this study, we introduce three modifications to the decoder-only transformer architecture, namely ParallelGPT (pgpt), LinearGPT (lgpt), and ConvGPT (cgpt). These variants demonstrate comparable performance to the conventional architecture in language generation, yet benefit from reduced model sizes and faster training processes. We open-source the model weights and the complete codebase for these implementation for further research.

Keywords

Cite

@article{arxiv.2404.14462,
  title  = {Towards smaller, faster decoder-only transformers: Architectural variants and their implications},
  author = {Sathya Krishnan Suresh and Shunmugapriya P},
  journal= {arXiv preprint arXiv:2404.14462},
  year   = {2024}
}

Comments

10 pages, 6 figures

R2 v1 2026-06-28T16:02:43.837Z