StarSpace: Embed All The Things!

Authors

  • Ledell Wu Facebook AI Research
  • Adam Fisch Facebook AI Research
  • Sumit Chopra Facebook AI Research
  • Keith Adams Facebook AI Research
  • Antoine Bordes Facebook AI Research
  • Jason Weston Facebook AI Research

DOI:

https://doi.org/10.1609/aaai.v32i1.11996

Keywords:

Nature Language Processing, Text Classification, Knowledge Representation, Recommender Systems

Abstract

We present StarSpace, a general-purpose neural embedding model that can solve a wide variety of problems: labeling tasks such as text classification,ranking tasks such as information retrieval/web search,collaborative filtering-based  or content-based recommendation,embedding of multi-relational graphs, and learning word, sentence or document level embeddings.In each case the model works by embedding those entities comprised of discrete features and comparing them against each other -- learning similarities dependent on the task.Empirical results on a number of tasks show that StarSpace is highly competitive with existing methods, whilst also being generally applicable to new cases where those methods are not.

Downloads

Published

2018-04-27

How to Cite

Wu, L., Fisch, A., Chopra, S., Adams, K., Bordes, A., & Weston, J. (2018). StarSpace: Embed All The Things!. Proceedings of the AAAI Conference on Artificial Intelligence, 32(1). https://doi.org/10.1609/aaai.v32i1.11996