Please use this identifier to cite or link to this item:
https://doi.org/10.1145/3077136.3080777
Title: | Neural Factorization Machines for Sparse Predictive Analytics | Authors: | Xiangnan He Tat-Seng Chua |
Keywords: | Deep Learning Factorization Machines Neural Networks Recommendation Regression Sparse Data |
Issue Date: | 7-Aug-2017 | Publisher: | Association for Computing Machinery, Inc | Citation: | Xiangnan He, Tat-Seng Chua (2017-08-07). Neural Factorization Machines for Sparse Predictive Analytics. ACM SIGIR 2017 : 355-364. ScholarBank@NUS Repository. https://doi.org/10.1145/3077136.3080777 | Abstract: | Many predictive tasks of web applications need to model categorical variables, such as user IDs and demographics like genders and occupations. To apply standard machine learning techniques, these categorical predictors are always converted to a set of binary features via one-hot encoding, making the resultant feature vector highly sparse. To learn from such sparse data e'ectively, it is crucial to account for the interactions between features. Factorization Machines (FMs) are a popular solution for effciently using the second-order feature interactions. However, FM models feature interactions in a linear way, which can be insuficient for capturing the non-linear and complex inherent structure of real-world data. While deep neural networks have recently been applied to learn non-linear feature interactions in industry, such as the Wide&Deep by Google and DeepCross by Microsofft, the deep structure meanwhile makes them dificult to train. In this paper, we propose a novel model Neural Factorization Machine (NFM) for prediction under sparse settings. NFM seamlessly combines the linearity of FM in modelling second-order feature interactions and the non-linearity of neural network in modelling higher-order feature interactions. Conceptually, NFM is more expressive than FM since FM can be seen as a special case of NFM without hidden layers. Empirical results on two regression tasks show that with one hidden layer only, NFM significantly outperforms FM with a 7:3% relative improvement. Compared to the recent deep learning methods Wide&Deep and DeepCross, our NFM uses a shallower structure but o'ers better performance, being much easier to train and tune in practice. © 2017 Copyright held by the owner/author(s). | Source Title: | ACM SIGIR 2017 | URI: | https://scholarbank.nus.edu.sg/handle/10635/167447 | ISBN: | 9781450350228 | DOI: | 10.1145/3077136.3080777 |
Appears in Collections: | Elements Staff Publications |
Show full item record
Files in This Item:
File | Description | Size | Format | Access Settings | Version | |
---|---|---|---|---|---|---|
Neural Factorization Machines for Sparse Predictive Analytics.pdf | 3.55 MB | Adobe PDF | OPEN | None | View/Download |
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.