Utility of General and Specific Word Embeddings for Classifying Translational Stages of Research

AMIA Annu Symp Proc. 2018 Dec 5:2018:1405-1414. eCollection 2018.

Abstract

Conventional text classification models make a bag-of-words assumption reducing text into word occurrence counts per document. Recent algorithms such as word2vec are capable of learning semantic meaning and similarity between words in an entirely unsupervised manner using a contextual window and doing so much faster than previous methods. Each word is projected into vector space such that similar meaning words such as "strong" and "powerful" are projected into the same general Euclidean space. Open questions about these embeddings include their utility across classification tasks and the optimal properties and source of documents to construct broadly functional embeddings. In this work, we demonstrate the usefulness of pre-trained embeddings for classification in our task and demonstrate that custom word embeddings, built in the domain and for the tasks, can improve performance over word embeddings learnt on more general data including news articles or Wikipedia.

MeSH terms

  • Algorithms*
  • Bayes Theorem
  • Decision Trees
  • Humans
  • Logistic Models
  • Natural Language Processing*
  • Semantics
  • Support Vector Machine