Entity Embeddings of Categorical Variables
From MaRDI portal
Publication:87297
DOI10.48550/ARXIV.1604.06737arXiv1604.06737MaRDI QIDQ87297FDOQ87297
Publication date: 22 April 2016
Abstract: We map categorical variables in a function approximation problem into Euclidean spaces, which are the entity embeddings of the categorical variables. The mapping is learned by a neural network during the standard supervised training process. Entity embedding not only reduces memory usage and speeds up neural networks compared with one-hot encoding, but more importantly by mapping similar values close to each other in the embedding space it reveals the intrinsic properties of the categorical variables. We applied it successfully in a recent Kaggle competition and were able to reach the third position with relative simple features. We further demonstrate in this paper that entity embedding helps the neural network to generalize better when the data is sparse and statistics is unknown. Thus it is especially useful for datasets with lots of high cardinality features, where other methods tend to overfit. We also demonstrate that the embeddings obtained from the trained neural network boost the performance of all tested machine learning methods considerably when used as the input features instead. As entity embedding defines a distance measure for categorical variables it can be used for visualizing categorical data and for data clustering.
Cited In (1)
This page was built for publication: Entity Embeddings of Categorical Variables
Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q87297)
