TensorGPT: Efficient Compression of the Embedding Layer in LLMs based on the Tensor-Train Decomposition
Authors: Mingxue Xu, Yao Lei Xu, Danilo P. Mandic
AI-generated Key Points
- High-dimensional token embeddings are crucial for capturing nuanced semantic information and improving complex language pattern modeling in Large Language Models (LLMs).
- The increased dimensionality poses challenges in terms of storage requirements and model parameters.
- A novel approach based on Tensor-Train Decomposition (TTD) is proposed to address the issue, treating each token embedding as a Matrix Product State (MPS) for efficient distributed computation.
- Experimental results on GPT-2 show impressive compression factors of up to 38.40 times for the embedding layer, with even a low compression factor of 3.31 times resulting in outperformance compared to the original GPT-2 model.
- The study "TensorGPT" by Mingxue Xu, Yao Lei Xu, and Danilo P. Mandic demonstrates how advancements in tensor decomposition techniques can significantly improve efficiency and performance in large-scale language models.
- Leveraging TTD for compressing token embeddings successfully addresses the challenge of high dimensionality while maintaining or surpassing performance levels of existing models like GPT-2.
- This research contributes to enhancing scalability and practicality of LLMs, opening up new possibilities for further exploration and optimization using advanced mathematical frameworks such as tensor decomposition methods.
Abstract: High-dimensional token embeddings underpin Large Language Models (LLMs), as they can capture subtle semantic information and significantly enhance the modelling of complex language patterns. However, the associated high dimensionality also introduces considerable model parameters, and a prohibitively high model storage. To address this issue, this work proposes an approach based on the Tensor-Train Decomposition (TTD), where each token embedding is treated as a Matrix Product State (MPS) that can be efficiently computed in a distributed manner. The experimental results on GPT-2 demonstrate that, through our approach, the embedding layer can be compressed by a factor of up to 38.40 times, and when the compression factor is 3.31 times, even produced a better performance than the original GPT-2 model.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.