about
Short-Text Classification Using Unsupervised Keyword Expansion (arxiv.org)
3 points by sel1 on Sep 19, 2019 | hide | past | pdf | discuss on HN

In plain words: Each short sentence is padded with extra related words invented inside a pre-trained language model's own memory, with no outside data or prior training. The added words were on-topic—3 to 10 per subject—and boosted accuracy when training examples were scarce.

Abstract

Short-text classification, like all data science, struggles to achieve high performance using limited data. As a solution, a short sentence may be expanded with new and relevant feature words to form an artificially enlarged dataset, and add new features to testing data. This paper applies a novel approach to text expansion by generating new words directly for each input sentence, thus requiring no additional datasets or previous training. In this unsupervised approach, new keywords are formed within the hidden states of a pre-trained language model and then used to create extended pseudo documents. The word generation process was assessed by examining how well the predicted words matched to topics of the input sentence. It was found that this method could produce 3-10 relevant new words for each target topic, while generating just 1 word related to each non-target topic. Generated words were then added to short news headlines to create extended pseudo headlines. Experimental results have shown that models trained using the pseudo headlines can improve classification accuracy when limiting the number of training examples.

Duncan Cameron-Steinke
arXiv:1909.07512 · cs.CL · submitted Sep 16, 2019
abstract · pdf · html

add comment on HN