English

Infinite-Dimensional Feature Interaction

Machine Learning 2024-11-05 v4

Abstract

The past neural network design has largely focused on feature representation space dimension and its capacity scaling (e.g., width, depth), but overlooked the feature interaction space scaling. Recent advancements have shown shifted focus towards element-wise multiplication to facilitate higher-dimensional feature interaction space for better information transformation. Despite this progress, multiplications predominantly capture low-order interactions, thus remaining confined to a finite-dimensional interaction space. To transcend this limitation, classic kernel methods emerge as a promising solution to engage features in an infinite-dimensional space. We introduce InfiNet, a model architecture that enables feature interaction within an infinite-dimensional space created by RBF kernel. Our experiments reveal that InfiNet achieves new state-of-the-art, owing to its capability to leverage infinite-dimensional interactions, significantly enhancing model performance.

Keywords

Cite

@article{arxiv.2405.13972,
  title  = {Infinite-Dimensional Feature Interaction},
  author = {Chenhui Xu and Fuxun Yu and Maoliang Li and Zihao Zheng and Zirui Xu and Jinjun Xiong and Xiang Chen},
  journal= {arXiv preprint arXiv:2405.13972},
  year   = {2024}
}

Comments

NeurIPS 2024

R2 v1 2026-06-28T16:36:17.182Z