Vocabulary Learning via Optimal Transport for Neural Machine Translation
Not on the map. Why?
Abstract (arXiv)
The choice of token vocabulary affects the performance of machine translation. This paper aims to figure out what is a good vocabulary and whether one can find the optimal vocabulary without trial training. To answer these questions, we first provide an alternative understanding of the role of vocabulary from the perspective of information theory. Motivated by this, we formulate the quest of vocabularization -- finding the best token dictionary with a proper size -- as an optimal transport (OT) problem. We propose VOLT, a simple and efficient solution without trial training. Empirical results show that VOLT outperforms widely-used vocabularies in diverse scenarios, including WMT-14 English-German and TED's 52 translation directions. For example, VOLT achieves almost 70% vocabulary size reduction and 0.5 BLEU gain on English-German translation. Also, compared to BPE-search, VOLT reduces the search time from 384 GPU hours to 30 GPU hours on English-German translation. Codes are available at https://github.com/Jingjing-NLP/VOLT .
Key facts. Cited by 58 papers in the map (data through 7 Oct 2026). First posted on arXiv 31 Dec 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 19; 2023: 11; 2024: 17; 2025: 8.
58 papers.
Sort:
By area:
- 2023Do All Languages Cost the Same? Tokenization in the Era of Commercial Language Modelscited by 186
- 2021Between words and characters: A Brief History of Open-Vocabulary Modeling and Tokenization in NLPcited by 130
- 2024Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabulariescited by 113
- 2024LLMs are Good Sign Language Translatorscited by 76
- 2022PLOT: Prompt Learning with Optimal Transport for Vision-Language Modelscited by 64
- 2022Impact of Tokenization on Language Models: An Analysis for Turkishcited by 62
- 2023Recent Advances on Federated Learning: A Systematic Surveycited by 40
- 2022Square One Bias in NLP: Towards a Multi-Dimensional Exploration of the Research Manifoldcited by 29
- 2022Rethinking and Refining the Distinct Metriccited by 27
- 2022Revisiting Sliced Wasserstein on Images: From Vectorization to Convolutioncited by 25
- 2022Hierarchical Sliced Wasserstein Distancecited by 23
- 2022Partially Does It: Towards Scene-Level FG-SBIR with Partial Inputcited by 20
38 more
- 2022How Robust is Neural Machine Translation to Language Imbalance in Multilingual Tokenizer Training?cited by 20
- 2022Overcoming Catastrophic Forgetting beyond Continual Learning: Balanced Training for Neural Machine Translationcited by 18
- 2023Wasserstein-Fisher-Rao Embedding: Logical Query Embeddings with Local Comparison and Global Transportcited by 16
- 2024Scaffold-BPE: Enhancing Byte Pair Encoding for Large Language Models with Simple and Effective Scaffold Token Removalcited by 16
- 2021The Volctrans GLAT System: Non-autoregressive Translation Meets WMT21cited by 14
- 2022Toward Interpretable Semantic Textual Similarity via Optimal Transport-based Contrastive Sentence Learningcited by 13
- 2024Lexically Grounded Subword Segmentationcited by 13
- 2025Protein Structure Tokenization: Benchmarking and New Recipecited by 13
- 2025DistDF: Time-Series Forecasting Needs Joint-Distribution Wasserstein Alignmentcited by 13
- 2024Memory-based Cross-modal Semantic Alignment Network for Radiology Report Generationcited by 11
- 2021Local Structure Matters Most: Perturbation Study in NLUcited by 10
- 2022OTExtSum: Extractive Text Summarisation with Optimal Transportcited by 8
- 2023T3L: Translate-and-Test Transfer Learning for Cross-Lingual Text Classificationcited by 8
- 2024MEDVOC: Vocabulary Adaptation for Fine-tuning Pre-trained Language Models on Medical Text Summarizationcited by 8
- 2024Language Generation with Strictly Proper Scoring Rulescited by 8
- 2022Learning Multiscale Transformer Models for Sequence Generationcited by 7
- 2022Linguistically inspired roadmap for building biologically reliable protein language modelscited by 7
- 2022Subword-Delimited Downsampling for Better Character-Level Translationcited by 7
- 2022Pretraining without Wordpieces: Learning Over a Vocabulary of Millions of Wordscited by 6
- 2024Hierarchical Hybrid Sliced Wasserstein: A Scalable Metric for Heterogeneous Joint Distributionscited by 6
- 2025FreeMesh: Boosting Mesh Generation with Coordinates Mergingcited by 6
- 2022Too Brittle To Touch: Comparing the Stability of Quantization and Distillation Towards Developing Lightweight Low-Resource MT Modelscited by 5
- 2024LBPE: Long-token-first Tokenization to Improve Large Language Modelscited by 5
- 2024SceneLLM: Implicit Language Reasoning in LLM for Dynamic Scene Graph Generationcited by 5
- 2024Computing Approximate Graph Edit Distance via Optimal Transportcited by 5
- 2025A Partition Cover Approach to Tokenizationcited by 5
- 2024UniBridge: A Unified Approach to Cross-Lingual Transfer Learning for Low-Resource Languagescited by 4
- 2024Second Language (Arabic) Acquisition of LLMs via Progressive Vocabulary Expansioncited by 4
- 2025Evaluation of LLMs in Medical Text Summarization: The Role of Vocabulary Adaptation in High OOV Settingscited by 4
- 2022GeONet: a neural operator for learning the Wasserstein geodesiccited by 3
- 2023On the Off-Target Problem of Zero-Shot Multilingual Neural Machine Translationcited by 3
- 2024Target-Aware Language Modeling via Granular Data Samplingcited by 3
- 2023TOT: Topology-Aware Optimal Transport For Multimodal Hate Detectioncited by 2
- 2024Infusing clinical knowledge into tokenisers for language modelscited by 2
- 2023Extrapolating Multilingual Understanding Models as Multilingual Generatorscited by 1
- 2023MobileNMT: Enabling Translation in 15MB and 30mscited by 1
- 2023SelfSeg: A Self-supervised Sub-word Segmentation Method for Neural Machine Translationcited by 1
- 2023Compressed online Sinkhorncited by 1
50 most cited of 58 All 58 (JSON)
- 2025DistDF: Time-Series Forecasting Needs Joint-Distribution Wasserstein Alignment28 Oct 2025
- 2025Conditional Unigram Tokenization with Parallel Data10 Jul 2025
- 2025Evaluation of LLMs in Medical Text Summarization: The Role of Vocabulary Adaptation in High OOV Settings27 May 2025
- 2025FreeMesh: Boosting Mesh Generation with Coordinates Merging19 May 2025
- 2025Self-Vocabularizing Training for Neural Machine Translation18 Mar 2025
- 2025Partial Distribution Alignment via Adaptive Optimal Transport7 Mar 2025
- 2025Protein Structure Tokenization: Benchmarking and New Recipe28 Feb 2025
- 2025A Partition Cover Approach to Tokenization8 Jan 2025
- 2024Computing Approximate Graph Edit Distance via Optimal Transport25 Dec 2024
- 2024Second Language (Arabic) Acquisition of LLMs via Progressive Vocabulary Expansion16 Dec 2024
- 2024SceneLLM: Implicit Language Reasoning in LLM for Dynamic Scene Graph Generation15 Dec 2024
- 2024LBPE: Long-token-first Tokenization to Improve Large Language Models8 Nov 2024
38 more
- 2024Adaptive BPE Tokenization for Enhanced Vocabulary Adaptation in Finetuning Pretrained Language Models4 Oct 2024
- 2024Target-Aware Language Modeling via Granular Data Sampling23 Sep 2024
- 2024Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabularies18 Jul 2024
- 2024Infusing clinical knowledge into tokenisers for language models20 Jun 2024
- 2024Lexically Grounded Subword Segmentation19 Jun 2024
- 2024COT: A Generative Approach for Hate Speech Counter-Narratives via Contrastive Optimal Transport18 Jun 2024
- 2024UniBridge: A Unified Approach to Cross-Lingual Transfer Learning for Low-Resource Languages14 Jun 2024
- 2024Language Generation with Strictly Proper Scoring Rules29 May 2024
- 2024MEDVOC: Vocabulary Adaptation for Fine-tuning Pre-trained Language Models on Medical Text Summarization7 May 2024
- 2024Scaffold-BPE: Enhancing Byte Pair Encoding for Large Language Models with Simple and Effective Scaffold Token Removal27 Apr 2024
- 2024Hierarchical Hybrid Sliced Wasserstein: A Scalable Metric for Heterogeneous Joint Distributions23 Apr 2024
- 2024LLMs are Good Sign Language Translators1 Apr 2024
- 2024Memory-based Cross-modal Semantic Alignment Network for Radiology Report Generation31 Mar 2024
- 2023Compressed online Sinkhorn8 Oct 2023
- 2023Exploring Linguistic Similarity and Zero-Shot Learning for Multilingual Translation of Dravidian Languages10 Aug 2023
- 2023SelfSeg: A Self-supervised Sub-word Segmentation Method for Neural Machine Translation31 Jul 2023
- 2023T3L: Translate-and-Test Transfer Learning for Cross-Lingual Text Classification8 Jun 2023
- 2023MobileNMT: Enabling Translation in 15MB and 30ms7 Jun 2023
- 2023Do All Languages Cost the Same? Tokenization in the Era of Commercial Language Models23 May 2023
- 2023Extrapolating Multilingual Understanding Models as Multilingual Generators22 May 2023
- 2023On the Off-Target Problem of Zero-Shot Multilingual Neural Machine Translation18 May 2023
- 2023Wasserstein-Fisher-Rao Embedding: Logical Query Embeddings with Local Comparison and Global Transport6 May 2023
- 2023TOT: Topology-Aware Optimal Transport For Multimodal Hate Detection27 Feb 2023
- 2023Recent Advances on Federated Learning: A Systematic Survey3 Jan 2023
- 2022Subword-Delimited Downsampling for Better Character-Level Translation2 Dec 2022
- 2022Word-Level Representation From Bytes For Language Modeling23 Nov 2022
- 2022Too Brittle To Touch: Comparing the Stability of Quantization and Distillation Towards Developing Lightweight Low-Resource MT Models27 Oct 2022
- 2022Transferring Knowledge via Neighborhood-Aware Optimal Transport for Low-Resource Hate Speech Detection17 Oct 2022
- 2022PLOT: Prompt Learning with Optimal Transport for Vision-Language Models3 Oct 2022
- 2022GeONet: a neural operator for learning the Wasserstein geodesic28 Sep 2022
- 2022Hierarchical Sliced Wasserstein Distance27 Sep 2022
- 2022Linguistically inspired roadmap for building biologically reliable protein language models3 Jul 2022
- 2022Square One Bias in NLP: Towards a Multi-Dimensional Exploration of the Research Manifold20 Jun 2022
- 2022Learning Multiscale Transformer Models for Sequence Generation19 Jun 2022
- 2022How Robust is Neural Machine Translation to Language Imbalance in Multilingual Tokenizer Training?29 Apr 2022
- 2022OTExtSum: Extractive Text Summarisation with Optimal Transport21 Apr 2022
- 2022Impact of Tokenization on Language Models: An Analysis for Turkish19 Apr 2022
- 2022Revisiting Sliced Wasserstein on Images: From Vectorization to Convolution4 Apr 2022
50 newest of 58 All 58 (JSON)
- 2023Do All Languages Cost the Same? Tokenization in the Era of Commercial Language Modelscited by 186
- 2021Between words and characters: A Brief History of Open-Vocabulary Modeling and Tokenization in NLPcited by 130
- 2024Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabulariescited by 113
- 2024LLMs are Good Sign Language Translatorscited by 76
- 2022Impact of Tokenization on Language Models: An Analysis for Turkishcited by 62
- 2023Recent Advances on Federated Learning: A Systematic Surveycited by 40
- 2022Square One Bias in NLP: Towards a Multi-Dimensional Exploration of the Research Manifoldcited by 29
- 2022Rethinking and Refining the Distinct Metriccited by 27
- 2022Revisiting Sliced Wasserstein on Images: From Vectorization to Convolutioncited by 25
- 2022Hierarchical Sliced Wasserstein Distancecited by 23
- 2022Partially Does It: Towards Scene-Level FG-SBIR with Partial Inputcited by 20
- 2022How Robust is Neural Machine Translation to Language Imbalance in Multilingual Tokenizer Training?cited by 20
12 most cited of 55 All 55 (JSON)
- 2025DistDF: Time-Series Forecasting Needs Joint-Distribution Wasserstein Alignment28 Oct 2025
- 2025Conditional Unigram Tokenization with Parallel Data10 Jul 2025
- 2025Evaluation of LLMs in Medical Text Summarization: The Role of Vocabulary Adaptation in High OOV Settings27 May 2025
- 2025FreeMesh: Boosting Mesh Generation with Coordinates Merging19 May 2025
- 2025Self-Vocabularizing Training for Neural Machine Translation18 Mar 2025
- 2025Partial Distribution Alignment via Adaptive Optimal Transport7 Mar 2025
- 2025Protein Structure Tokenization: Benchmarking and New Recipe28 Feb 2025
- 2025A Partition Cover Approach to Tokenization8 Jan 2025
- 2024Computing Approximate Graph Edit Distance via Optimal Transport25 Dec 2024
- 2024Second Language (Arabic) Acquisition of LLMs via Progressive Vocabulary Expansion16 Dec 2024
- 2024LBPE: Long-token-first Tokenization to Improve Large Language Models8 Nov 2024
- 2024Adaptive BPE Tokenization for Enhanced Vocabulary Adaptation in Finetuning Pretrained Language Models4 Oct 2024
12 newest of 55 All 55 (JSON)