We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: Exponential Machines

Abstract: Modeling interactions between features improves the performance of machine learning solutions in many domains (e.g. recommender systems or sentiment analysis). In this paper, we introduce Exponential Machines (ExM), a predictor that models all interactions of every order. The key idea is to represent an exponentially large tensor of parameters in a factorized format called Tensor Train (TT). The Tensor Train format regularizes the model and lets you control the number of underlying parameters. To train the model, we develop a stochastic Riemannian optimization procedure, which allows us to fit tensors with 2^160 entries. We show that the model achieves state-of-the-art performance on synthetic data with high-order interactions and that it works on par with high-order factorization machines on a recommender system dataset MovieLens 100K.
Comments: ICLR-2017 workshop track paper
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:1605.03795 [stat.ML]
  (or arXiv:1605.03795v3 [stat.ML] for this version)

Submission history

From: Alexander Novikov [view email]
[v1] Thu, 12 May 2016 13:08:11 GMT (99kb,D)
[v2] Thu, 10 Nov 2016 10:24:08 GMT (182kb,D)
[v3] Fri, 8 Dec 2017 08:17:58 GMT (434kb,D)

Link back to: arXiv, form interface, contact.