Abstract

AutoEncoder is an unsupervised learning approach that can maps inputs to useful intermediate features, which can be used to build recommendation. Intermediate features of different entities obtained by AutoEncoder may have different weight for predicting users behavior. However, existing research typically uses a uniform weight on intermediate features to make a fast learning algorithm, this general approach may lead to the limited performance of the model. In this paper, we proposes a novel approach by using SGD to dynamically learn the intermediate features importance, which can integrate the intermediate features into matrix factorization framework seamlessly. In the previous works, the entities intermediate features learned by AutoEncoder are modeled as a whole. On this basis, we proposes to use attention parameters in entity intermediate feature to dynamically learn the intermediate features importance and build fine-grained model. By learning unique attention unit for each entity intermediate feature, the entities intermediate features are integrated into the matrix factorization framework better. Extensive experiments conducted over two real-world datasets demonstrate our proposed approach outperforms the compared models.

Full Text
Paper version not known

Talk to us

Join us for a 30 min session where you can share your feedback and ask us any queries you have

Schedule a call