$k$-Nearest Neighbor Augmented Neural Networks for Text Classification

Abstract

In recent years, many deep-learning based models are proposed for text classification. This kind of models well fits the training set from the statistical point of view. However, it lacks the capacity of utilizing instance-level information from individual instances in the training set. In this work, we propose to enhance neural network models by allowing them to leverage information from $k$-nearest neighbor (kNN) of the input text. Our model employs a neural network that encodes texts into text embeddings. Moreover, we also utilize $k$-nearest neighbor of the input text as an external memory, and utilize it to capture instance-level information from the training set. The final prediction is made based on features from both the neural network encoder and the kNN memory. Experimental results on several standard benchmark datasets show that our model outperforms the baseline model on all the datasets, and it even beats a very deep neural network model (with 29 layers) in several datasets. Our model also shows superior performance when training instances are scarce, and when the training set is severely unbalanced. Our model also leverages techniques such as semi-supervised training and transfer learning quite well.

References in corpus (9)

• Adam: A Method for Stochastic Optimization
• Neural Machine Translation by Jointly Learning to Align and Translate
• Convolutional Neural Networks for Sentence Classification
• Matching Networks for One Shot Learning
• A Convolutional Neural Network for Modelling Sentences
• Sentence Similarity Learning by Lexical Decomposition and Composition
• Efficient Character-level Document Classification by Combining Convolution and Recurrent Layers
• Multi-Perspective Context Matching for Machine Comprehension
• Semi-supervised Clustering for Short Text via Deep Representation Learning

Cited by in corpus (5)

• Deep Learning-based Sentiment Classification: A Comparative Survey
• Retrieval-Augmented Convolutional Neural Networks for Improved Robustness against Adversarial Examples
• On Convergence of Nearest Neighbor Classifiers over Feature Transformations
• Nearest Neighbour Few-Shot Learning for Cross-lingual Classification
• k-Nearest Neighbors by Means of Sequence to Sequence Deep Neural Networks and Memory Networks