Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Towards Hybrid-grained Feature Interaction Selection for Deep Sparse Network

About

Deep sparse networks are widely investigated as a neural network architecture for prediction tasks with high-dimensional sparse features, with which feature interaction selection is a critical component. While previous methods primarily focus on how to search feature interaction in a coarse-grained space, less attention has been given to a finer granularity. In this work, we introduce a hybrid-grained feature interaction selection approach that targets both feature field and feature value for deep sparse networks. To explore such expansive space, we propose a decomposed space which is calculated on the fly. We then develop a selection algorithm called OptFeature, which efficiently selects the feature interaction from both the feature field and the feature value simultaneously. Results from experiments on three large real-world benchmark datasets demonstrate that OptFeature performs well in terms of accuracy and efficiency. Additional studies support the feasibility of our method.

Fuyuan Lyu, Xing Tang, Dugang Liu, Chen Ma, Weihong Luo, Liang Chen, Xiuqiang He, Xue Liu• 2023

Related benchmarks

TaskDatasetResultRank
CTR PredictionCriteo
AUC0.8116
282
CTR PredictionAvazu
AUC79.25
144
Click-Through Rate PredictionKDD12
AUC0.7982
28
CTR PredictionIndustrial
AUC78.93
15
Showing 4 of 4 rows

Other info

Code

Follow for update