Multi-Label Text Classification with Transfer Learning for Policy Documents : The Case of the Sustainable Development Goals

• Main Author: Rodríguez Medina, Samuel
• Format: Others
• Language: English
• Published: Uppsala universitet, Institutionen för lingvistik och filologi 2019
• Subjects: machine learning; deep neural networks; transfer learning; text classification; sustainable development goals; sdgs; Language Technology (Computational Linguistics); Språkteknologi (språkvetenskaplig databehandling)
• Online Access: http://urn.kb.se/resolve?urn=urn:nbn:se:uu:diva-395186

We created and analyzed a text classification dataset from freely-available web documents from the United Nation's Sustainable Development Goals. We then used it to train and compare different multi-label text classifiers with the aim of exploring the alternatives for methods that facilitate the search of information of this type of documents. We explored the effectiveness of deep learning and transfer learning in text classification by fine-tuning different pre-trained language representations — Word2Vec, GloVe, ELMo, ULMFiT and BERT. We also compared these approaches against a baseline of more traditional algorithms without using transfer learning. More specifically, we used multinomial Naive Bayes, logistic regression, k-nearest neighbors and Support Vector Machines. We then analyzed the results of our experiments quantitatively and qualitatively. The best results in terms of micro-averaged F1 scores and AUROC are obtained by BERT. However, it is also interesting that the second best classifier in terms of micro-averaged F1 scores is the Support Vector Machines, closely followed by the logistic regression classifier, which both have the advantage of being less computationally expensive than BERT. The results also show a close relation between our dataset size and the effectiveness of the classifiers.