Text classification model SloBERTa-Trendi-Topics 1.0

  Authors

Čibej, Jaka ; et al.show everyone

Čibej, Jaka ; Kuzman, Taja ; Ljubešić, Nikola ; Kosem, Iztok ; Ponikvar, Primož ; Dobrovoljc, Kaja ; Krek, Simon

  Item identifier

http://hdl.handle.net/11356/1709

 Project URL

https://sled.ijs.si/

 Demo URL

https://huggingface.co/cjvt/sloberta-trendi-topics

 Referenced by

https://nl.ijs.si/jtdh22/pdf/JTDH2022_Kosem-et-al_Spremljevalni-korpus-Trendi.pdf

 Date issued

2022-10-28

 Type

toolService

 Language(s)

Slovenian

 Description

The SloBerta-Trendi-Topics model is a text classification model for categorizing news texts with one of 13 topic labels. It was trained on a set of approx. 36,000 Slovene texts from various Slovene news sources included in the Trendi Monitor Corpus of Slovene (http://hdl.handle.net/11356/1590) such as "rtvslo.si", "sta.si", "delo.si", "dnevnik.si", "vecer.com", "24ur.com", "siol.net", "gorenjskiglas.si", etc. The texts were semi-automatically categorized into 13 categories based on the sections under which they were published (i.e. URLs). The set of labels was developed in accordance with related categorization schemas used in other corpora and comprises the following topics: "črna kronika" (crime and accidents), "gospodarstvo, posel, finance" (economy, business, finance), "izobraževanje" (education), "okolje" (environment), "prosti čas" (free time), "šport" (sport), "umetnost, kultura" (art, culture), "vreme" (weather), "zabava" (entertainment), "zdravje" (health), "znanost in tehnologija" (science and technology), "politika" (politics), and "družba" (society). The categorization process is explained in more detail in Kosem et al. (2022): https://nl.ijs.si/jtdh22/pdf/JTDH2022_Kosem-et-al_Spremljevalni-korpus-Trendi.pdf The model was trained on the labeled texts using the SloBERTa 2.0 contextual embeddings model (http://hdl.handle.net/11356/1397; also available at HuggingFace: https://huggingface.co/EMBEDDIA/sloberta) and validated on a development set of 1,293 texts using the simpletransformers library and the following hyperparameters: Train batch size: 8 Learning rate: 1e-5 Max. sequence length: 512 Number of epochs: 2 The model achieves a macro-F1-score of 0.94 on a test set of 1,295 texts (best for "črna kronika", "politika", "šport", and "vreme" at 0.98, worst for "prosti čas" at 0.83). Please note that the fastText-Trendi-Topics 1.0 text classification model is also available (http://hdl.handle.net/11356/1710) that is faster and computationally less demanding, but achieves lower classification accuracy.

 Publisher

Jožef Stefan Institute

 Acknowledgement

Ministry of Culture of the Republic of Slovenia JR-infrastruktura-SJ-2021-2022 "SLED - Monitor corpus of Slovene and related resources"

ARRS (Slovenian Research Agency) P6-0411 "Language Resources and Technologies for Slovene"

University of Ljubljana I0-0022 "Network of Research Infrastructure Centres (MRIC)"

 Subject(s)

text classification topic classification SloBERTa Slovenian news articles

 Collection(s)

CLARIN.SI data & tools

Show full item record