Multi-Label Text Classification with Transfer Learning for Policy Documents: The Case of the Sustainable Development Goals
2019 (English)Independent thesis Advanced level (degree of Master (Two Years)), 20 credits / 30 HE credits
Student thesis
Abstract [en]
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.
Place, publisher, year, edition, pages
2019. , p. 44
Keywords [en]
machine learning, deep neural networks, transfer learning, text classification, sustainable development goals, sdgs
National Category
Natural Language Processing
Identifiers
URN: urn:nbn:se:uu:diva-395186OAI: oai:DiVA.org:uu-395186DiVA, id: diva2:1360968
External cooperation
Aidin Niamir; Maral Dadvar
Subject / course
Language Technology
Educational program
Master Programme in Language Technology
Supervisors
Examiners
2019-10-152019-10-142025-02-07Bibliographically approved