NeurIPS 2023poster3 citations

Towards Hybrid-grained Feature Interaction Selection for Deep Sparse Network

Fuyuan Lyu, Xing Tang, Dugang Liu, Chen Ma, Weihong Luo, Liang Chen, xiuqiang He, Xue Liu

Abstract

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. All source code are publicly available\footnote{https://anonymous.4open.science/r/OptFeature-Anonymous}.

Feature Interaction SearchDeep Sparse Network
BibTeX
@inproceedings{
lyu2023towards,
title={Towards Hybrid-grained Feature Interaction Selection for Deep Sparse Network},
author={Fuyuan Lyu and Xing Tang and Dugang Liu and Chen Ma and Weihong Luo and Liang Chen and xiuqiang He and Xue Liu},
booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
year={2023},
url={https://openreview.net/forum?id=xxfHMqNcum}
}