CATEGORICAL FEATURE SELECTION FOR RANKING MODELS
Machine Learning based ranking models are ubiquitous in powering recommendation engines at internet companies. These models typically use a combination of real-valued numerical and categorical features to generate predictions. Feature selection may be a widely encountered problem in this setting, that entails picking the optimal set of features as inputs to these models from a large pool of candidate real-valued and categorical features. A novel feature selection algorithm for categorical features building on stochastic neural networks is provided. It is shown empirically through results, the superiority of this algorithm over existing approaches. Study and proposal of best practices are also provided to practitioners to extract maximum value out of the new feature selection approach.
1 . A method comprising:
analyzing a set of categorical features associated with a plurality of users of a social network;
providing respective categorical features associated with corresponding embedding layers to corresponding stochastic gates;
determining scores associated with each of the categorical features provided to the stochastic gates; and
determining a subset of the categorical features to provide to a ranking model based on determined top scores associated with each of the categorical features.
2 . The method of claim 1 , wherein the categorical features having determined scores that are not within the top scores are prevented from being provided to the ranking model.
3 . The method of claim 1 , wherein the top scores are determined as being within a range of score values.
4 . The method of claim 1 , wherein the social network is associated with a stochastic neuron network.