arXiv · 1906.09532
Smaller Text Classifiers with Discriminative Cluster Embeddings
Abstract
Word embedding parameters often dominate overall model sizes in neural methods for natural language processing. We reduce deployed model sizes of text classifiers by learning a hard word clustering in an end-to-end manner. We use the Gumbel-Softmax distribution to maximize over the latent clustering while minimizing the task loss. We propose variations that selectively assign additional parameters to words, which further improves accuracy while still remaining parameter-efficient.
Explore related subjects
Keep this discovery
Mingda Chen, Kevin Gimpel. 2019-06-23. Smaller Text Classifiers with Discriminative Cluster Embeddings. https://arxiv.org/abs/1906.09532
Cite the original work for its findings. Save a collection to share your selection of sources.