Improving Multilabel Text Classification with Stacking and Recurrent Neural Networks

What is it about?

Multilabel text classification can be defined as a mapping function that categorizes a text in natural language into one or more labels defined by the scope of a problem. In this work we propose an architecture of stacked classifiers for multilabel text classification. The proposed models use an LSTM recurrent neural network in the first stage of the stack and different multilabel classifiers in the second stage. We evaluated our proposal in two datasets well-known in the literature (TMDB and EUR-LEX Subject Matters), and the results showed that the proposed stack consistently outperformed the baselines.

Featured Image

Photo by Kevin Ku on Unsplash

Photo by Kevin Ku on Unsplash