Downloads · 30 days
0
SBB/ark-omikuji-fre-title-content
ark-omikuji-fre-title-content is a text classification model from SBB. Use it when you need a label for a piece of text. It is set up for annif. The card lists the license as apache-2.0.
An Annif model, trained on historical titles and additional catalogue metadata for automatic subject indexing tasks. It classifies a given text into one or multiple subjects from the “Alter Realkatalog” (ARK) classifi…
Downloads · 30 days
0
Access
Public
Updated Aug 14, 2024
Repo size
81.4 MB
Likes
0
Public
Click a slice to open those files.
.cbor50.7 MB · 62%
From the Hugging Face model README
An Annif model, trained on historical titles and additional catalogue metadata for automatic subject indexing tasks. It classifies a given text into one or multiple subjects from the “Alter Realkatalog” (ARK) classification system. The model was developed in the research project Human.Machine.Culture at Staatsbibliothek zu Berlin – Berlin State Library (SBB).
An Annif model, trained on historical titles and additional catalogue metadata for automatic subject indexing tasks. Subject indexing is a classical library task, aiming at describing the content of a resource. The model is intended to be used to automatically classify historical texts with a historical classification system developed in the 19th century to enrich those texts that have not been classified manually so far. For each input text, the model outputs one or multiple subjects from the ARK classification system. It is part of a collection of 5 models, created with the help of the Annif toolkit which addresses this task of automated subject indexing.
This model can directly be used to automatically classify historical texts with the ARK classification scheme. It is intended to be used together with the Annif automated subject indexing toolkit version 0.60.0-1.1.0.
Other/downstream uses outside of the Annif setting described above are not intended but also not excluded.
The model is not intended for use on contemporary texts, as language and concept drifts will probably influence the results negatively and some terms from the vocabulary are not appropriate for more recent publications.
Since we are dealing with historical texts and especially with a historical classification system such as the ARK, the classes suggested for an input text might not be suitable for today’s understanding or might even be of an unethical nature (for more information, see also the datasheet accompanying the Metadata of the “Alter Realkatalog” (ARK of Berlin State Library) and the Datasheet for Machine-Readable Vocabulary Files of the ARK (Alter Realkatalog)).
Another limitation when using the ARK as a vocabulary arises from its hierarchical structure: the system contains multiple classes that do not describe the same content (e.g. they have different IDs) but are labeled identical (same name). This is due to the fact that the manual inspection of the whole path to a class, including all the upper level classes leading to it, delivers additional information that allows for contextualization. As duplicate label names seem to be - as expected - a challenge for lexical methods, we decided to focus on statistical rather than lexical algorithms.
Considering that the ARK classification scheme consists of 225.691 classes in total and that there is only limited training material at hand plus an overall unbalanced distribution of classes, we might describe this task as an Extreme Multi-Label Classification (XMC) problem. We recommend being aware of this limitation and, if available, use additional training data to improve the current model’s performance (e.g. by running annif learn, see CLI commands documentation).
Training data include a selection of metadata fields that were obtained via CBS export:
The following title and content data fields have been extracted and combined from this dataset:
The vocabulary files themselves can be found here:
Training procedure includes loading the ARK vocabulary (see Datasheet for Machine-Readable Vocabulary Files of the ARK (Alter Realkatalog)) into Annif and training the Omikuji backend with the help of our training data. Further aspects on technical specifications can be found in the section Training hyperparameters.
Besides merging and transforming the data described under Training Data to fit the corpus formats accepted by Annif, no further preprocessing of natural language or similar has been performed.
Training takes from several minutes to a few hours on a V100, depending on the choice of dataset and algorithm as well as hyperparameter settings.
For some of the ARK Annif models, a slight hyperparameter optimization has been conducted to identify the final hyperparameter settings stated below.
hyperparameter configuration (as it needs to be stated in the Annif projects.cfg file):
[ark-omikuji-fre-title-content]
name=ARK-DE-18 Omikuji
language=fr
backend=omikuji
analyzer=snowball(french)
vocab=arktsv
cluster_k=2
collapse_every_n_layers=5
--limit 1, --threshold 0): 0.4228--limit 1, --threshold 0): 0.4044--limit 1, --threshold 0): 0.4103--limit 1, --threshold 0): 0.4169The dataset is described under Training Data. It was split into smaller subsets used for training, testing and validating (80%/10%/10% split).
Model performance has been evaluated based on the following metrics: Precision, Recall, F1 and NDCG. These are standard metrics for machine learning and more specifically automatic subject indexing tasks and are directly provided in Annif by calling the annif eval statement. Evaluation parameters (--limit = maximum number of results to return; --threshold = minimum confidence for a suggestion to be considered) have been optimized before using the validation dataset and affect the results accordingly. We also state F1@5 and NDCG@5 scores reached without any evaluation parameters.
Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).
See Annif and Omikuji repositories on Github. Omikuji is an implementation of Partitioned Label Trees (Prabhu et al., 2018):
To run this model, Annif version 0.60.0 or higher (min. up to 1.1.0) must be installed.
Sophie Schneider and Jörg Lehmann
Questions and comments about the model can be directed to Sophie Schneider at [email protected], questions and comments about the model card can be directed to Jörg Lehmann at [email protected].
Follow the Annif Getting Started page to set up and run Annif. Create a projects.cfg file (see section Training hyperparameters for details on the specific project configuration), load the ARK vocabulary (see Datasheet for Machine-Readable Vocabulary Files of the ARK (Alter Realkatalog)) via annif load-vocab command and copy the model folder over to data/projects.