CLAILGApr 22, 2024

SpaceByte: Towards Deleting Tokenization from Large Language Modeling

arXiv:2404.14408v329 citationsh-index: 1NIPS
Originality Incremental advance
AI Analysis

This addresses tokenization disadvantages like performance biases and adversarial vulnerability for large language model developers, though it appears incremental as it matches rather than surpasses tokenized models.

The paper tackles the performance gap between byte-level and subword tokenized language models by proposing SpaceByte, a byte-level decoder architecture with extra larger transformer blocks inserted after certain bytes like spaces. Their experiments show that SpaceByte roughly matches the performance of tokenized Transformer architectures while eliminating tokenization disadvantages.

Tokenization is widely used in large language models because it significantly improves performance. However, tokenization imposes several disadvantages, such as performance biases, increased adversarial vulnerability, decreased character-level modeling performance, and increased modeling complexity. To address these disadvantages without sacrificing performance, we propose SpaceByte, a novel byte-level decoder architecture that closes the performance gap between byte-level and subword autoregressive language modeling. SpaceByte consists of a byte-level Transformer model, but with extra larger transformer blocks inserted in the middle of the layers. We find that performance is significantly improved by applying these larger blocks only after certain bytes, such as space characters, which typically denote word boundaries. Our experiments show that for a fixed training and inference compute budget, SpaceByte outperforms other byte-level architectures and roughly matches the performance of tokenized Transformer architectures.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes