arXiv · 2501.00343
Chunk-Distilled Language Modeling
Abstract
We introduce Chunk-Distilled Language Modeling (CD-LM), an approach to text generation that addresses two challenges in current large language models (LLMs): the inefficiency of token-level generation, and the difficulty of adapting to new data and knowledge. Our method combines deep network-based LLMs with a straightforward retrieval module, which allows the generation of multi-token text chunks at a single decoding step. Our retrieval framework enables flexible construction of model- or domain-specific datastores, either leveraging the internal knowledge of existing models, or incorporating expert insights from human-annotated corpora. This adaptability allows for enhanced control over the language model's distribution without necessitating additional training. We present the CD-LM formulation along with performance metrics demonstrating its ability to improve language model performance and efficiency across a diverse set of downstream tasks. Code and data will be made publicly available.
Explore related subjects
Keep this discovery
Yanhong Li, Karen Livescu, Jiawei Zhou. 2024-12-31. Chunk-Distilled Language Modeling. https://arxiv.org/abs/2501.00343
Cite the original work for its findings. Save a collection to share your selection of sources.