Tensorized Embedding Layers for Efficient Model Compression
Not on the map. Why?
Abstract (arXiv)
The embedding layers transforming input words into real vectors are the key components of deep neural networks used in natural language processing. However, when the vocabulary is large, the corresponding weight matrices can be enormous, which precludes their deployment in a limited resource setting. We introduce a novel way of parametrizing embedding layers based on the Tensor Train (TT) decomposition, which allows compressing the model significantly at the cost of a negligible drop or even a slight gain in performance. We evaluate our method on a wide range of benchmarks in natural language processing and analyze the trade-off between performance and compression ratios for a wide range of architectures, from MLPs to LSTMs and Transformers.
Key facts. Cited by 27 papers in the map (data through 7 Oct 2026). First posted on arXiv 30 Jan 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 5; 2023: 5; 2024: 6; 2025: 1; 2026: 2.
27 papers.
Sort:
- 2021TT-Rec: Tensor Train Compression for Deep Learning Recommendation Modelscited by 58
- 2021Redesigning the Transformer Architecture with Insights from Multi-particle Dynamical Systemscited by 29
- 2021Enabling Lightweight Fine-tuning for Pre-trained Language Model Compression based on Matrix Product Operatorscited by 28
- 2021KroneckerBERT: Learning Kronecker Decomposition for Pre-trained Language Models via Knowledge Distillationcited by 19
- 2022Exploring Extreme Parameter Compression for Pre-trained Language Modelscited by 15
- 2023Tensor Decomposition for Model Reduction in Neural Networks: A Reviewcited by 14
- 2024CoMERA: Computing- and Memory-Efficient Training via Rank-Adaptive Tensor Optimizationcited by 14
- 2024Motion meets Attention: Video Motion Promptscited by 12
- 2021Binary Code based Hash Embedding for Web-scale Applicationscited by 11
- 2023Efficient GPT Model Pre-training using Tensor Train Matrix Representationcited by 8
- 2022Knowledge Distillation of Russian Language Models with Reduction of Vocabularycited by 7
- 2022tntorch: Tensor Network Learning with PyTorchcited by 7
15 more
- 2024Super Tiny Language Modelscited by 7
- 2022Nimble GNN Embedding with Tensor-Train Decompositioncited by 6
- 2024TQCompressor: improving tensor decomposition methods in neural networks via permutationscited by 5
- 2024A Matrix Product State Model for Simultaneous Classification and Generationcited by 5
- 20213U-EdgeAI: Ultra-Low Memory Training, Ultra-Low BitwidthQuantization, and Ultra-Low Latency Accelerationcited by 3
- 2023Self-Supervised Learning for Data Scarcity in a Fatigue Damage Prognostic Problemcited by 3
- 2024Geometry is All You Need: A Unified Taxonomy of Matrix and Tensor Factorization for Compression of Generative Language Modelscited by 3
- 2021From Fully Trained to Fully Random Embeddings: Improving Neural Machine Translation with Compact Word Embedding Tablescited by 2
- 2022Deep tensor networks with matrix product operatorscited by 2
- 2026Compressing Transformer Language Models via Matrix Product Operator Decomposition: A Case Study on PicoGPTcited by 2
- 2021Learning Robust and Lightweight Model through Separable Structured Transformationscited by 0
- 2023Review of compressed embedding layers and their applications for recommender systemscited by 0
- 2023Enhancing Cross-Category Learning in Recommendation Systems with Multi-Layer Embedding Trainingcited by 0
- 2025SCRec: A Scalable Computational Storage System with Statistical Sharding and Tensor-train Decomposition for Recommendation Modelscited by 0
- 2026Pre-registered tests of solid-state-physics-inspired LLM compression: a cluster-level negative result at small-language-model scalecited by 0
- 2026Pre-registered tests of solid-state-physics-inspired LLM compression: a cluster-level negative result at small-language-model scale28 Sep 2026
- 2026Compressing Transformer Language Models via Matrix Product Operator Decomposition: A Case Study on PicoGPT30 Mar 2026
- 2025SCRec: A Scalable Computational Storage System with Statistical Sharding and Tensor-train Decomposition for Recommendation Models1 Apr 2025
- 2024Geometry is All You Need: A Unified Taxonomy of Matrix and Tensor Factorization for Compression of Generative Language Models3 Oct 2024
- 2024Motion meets Attention: Video Motion Prompts3 Jul 2024
- 2024A Matrix Product State Model for Simultaneous Classification and Generation25 Jun 2024
- 2024CoMERA: Computing- and Memory-Efficient Training via Rank-Adaptive Tensor Optimization23 May 2024
- 2024Super Tiny Language Models23 May 2024
- 2024TQCompressor: improving tensor decomposition methods in neural networks via permutations29 Jan 2024
- 2023Enhancing Cross-Category Learning in Recommendation Systems with Multi-Layer Embedding Training27 Sep 2023
- 2023Review of compressed embedding layers and their applications for recommender systems23 Jun 2023
- 2023Efficient GPT Model Pre-training using Tensor Train Matrix Representation5 Jun 2023
15 more
- 2023Tensor Decomposition for Model Reduction in Neural Networks: A Review26 Apr 2023
- 2023Self-Supervised Learning for Data Scarcity in a Fatigue Damage Prognostic Problem20 Jan 2023
- 2022Deep tensor networks with matrix product operators16 Sep 2022
- 2022tntorch: Tensor Network Learning with PyTorch22 Jun 2022
- 2022Nimble GNN Embedding with Tensor-Train Decomposition21 Jun 2022
- 2022Exploring Extreme Parameter Compression for Pre-trained Language Models20 May 2022
- 2022Knowledge Distillation of Russian Language Models with Reduction of Vocabulary4 May 2022
- 2021Learning Robust and Lightweight Model through Separable Structured Transformations27 Dec 2021
- 2021Redesigning the Transformer Architecture with Insights from Multi-particle Dynamical Systems30 Sep 2021
- 2021KroneckerBERT: Learning Kronecker Decomposition for Pre-trained Language Models via Knowledge Distillation13 Sep 2021
- 2021Binary Code based Hash Embedding for Web-scale Applications24 Aug 2021
- 2021Enabling Lightweight Fine-tuning for Pre-trained Language Model Compression based on Matrix Product Operators4 Jun 2021
- 20213U-EdgeAI: Ultra-Low Memory Training, Ultra-Low BitwidthQuantization, and Ultra-Low Latency Acceleration11 May 2021
- 2021From Fully Trained to Fully Random Embeddings: Improving Neural Machine Translation with Compact Word Embedding Tables18 Apr 2021
- 2021TT-Rec: Tensor Train Compression for Deep Learning Recommendation Models25 Jan 2021
- 2021TT-Rec: Tensor Train Compression for Deep Learning Recommendation Modelscited by 58
- 2021Redesigning the Transformer Architecture with Insights from Multi-particle Dynamical Systemscited by 29
- 2021Enabling Lightweight Fine-tuning for Pre-trained Language Model Compression based on Matrix Product Operatorscited by 28
- 2021KroneckerBERT: Learning Kronecker Decomposition for Pre-trained Language Models via Knowledge Distillationcited by 19
- 2022Exploring Extreme Parameter Compression for Pre-trained Language Modelscited by 15
- 2023Tensor Decomposition for Model Reduction in Neural Networks: A Reviewcited by 14
- 2024CoMERA: Computing- and Memory-Efficient Training via Rank-Adaptive Tensor Optimizationcited by 14
- 2024Motion meets Attention: Video Motion Promptscited by 12
- 2021Binary Code based Hash Embedding for Web-scale Applicationscited by 11
- 2023Efficient GPT Model Pre-training using Tensor Train Matrix Representationcited by 8
- 2022Knowledge Distillation of Russian Language Models with Reduction of Vocabularycited by 7
- 2022tntorch: Tensor Network Learning with PyTorchcited by 7
12 most cited of 27 All 27 (JSON)
- 2026Pre-registered tests of solid-state-physics-inspired LLM compression: a cluster-level negative result at small-language-model scale28 Sep 2026
- 2026Compressing Transformer Language Models via Matrix Product Operator Decomposition: A Case Study on PicoGPT30 Mar 2026
- 2025SCRec: A Scalable Computational Storage System with Statistical Sharding and Tensor-train Decomposition for Recommendation Models1 Apr 2025
- 2024Geometry is All You Need: A Unified Taxonomy of Matrix and Tensor Factorization for Compression of Generative Language Models3 Oct 2024
- 2024Motion meets Attention: Video Motion Prompts3 Jul 2024
- 2024A Matrix Product State Model for Simultaneous Classification and Generation25 Jun 2024
- 2024CoMERA: Computing- and Memory-Efficient Training via Rank-Adaptive Tensor Optimization23 May 2024
- 2024Super Tiny Language Models23 May 2024
- 2024TQCompressor: improving tensor decomposition methods in neural networks via permutations29 Jan 2024
- 2023Enhancing Cross-Category Learning in Recommendation Systems with Multi-Layer Embedding Training27 Sep 2023
- 2023Review of compressed embedding layers and their applications for recommender systems23 Jun 2023
- 2023Efficient GPT Model Pre-training using Tensor Train Matrix Representation5 Jun 2023
12 newest of 27 All 27 (JSON)