Swedish spaCy models

The National Library of Sweden / KB Lab releases two pretrained multitask models compatible with the NLP python package spaCy.
A language specific model for Swedish is not included in the core models as of the latest release (v2.3.2), so we publish our own models trained within the spaCy framework.
The models include a part-of-speech tagger, dependency parser and named entity recognition. We trained two separate models for Universal POS tags (UPOS) and Language Specific POS tags (XPOS), as spaCy does not allow for joint training of both.

** UPDATE decembar 2023 **: The models are only available here https://huggingface.co/KBLab/swedish-spacy-pipeline/tree/main

** UPDATE February 2021 **: We are adding two transformer-based models trained in spaCy 3.0. They are available to download at the same link given below.
One model is a complete pipeline with UPOS tagger, parser, sentencer, ner and lemmatizer (sv_pipeline-0.0.0.tar.gz). Unfortunately the lemmatizer is not yet trainable in spaCy, so the performance is as good as the quality of the rules/lookup tables available for Swedish (i.e. not very good). If you need a Swedish lemmatizer we advise you for the moment to have a look at Stanza, efselab or lemmy.
The other model is a XPOS tagger in case you need language-specific part-of-speech tags (sv_tagger-0.0.0.tar.gz).

The training data is the same as the 2.3.2 models. Here are the performance scores of the new models on the same test sets:

XPOS tagger (accuracy): 97.96
UPOS tagger (accuracy): 98.40
Parser (UAS/LAS): 93.51/90.74
Sentencer (F score): 94.73
NER (F score): 90.06

The models can be installed and loaded into spaCy as follows:

$ pip install sv_pipeline-0.0.0.tar.gz

>>> import spacy
>>> nlp = spacy.load("sv-pipeline")

N.B. Make sure to install the spacy-transformers extension, or the models won't work.

Training data and performance

We trained two separate tagger models for UPOS and XPOS, but the parser and NER are the same for both models. The models are initialized with FastText word embeddings.

UPOS tagger

The UPOS tagger was trained using two Swedish treebanks from Universal Dependencies, UD_Swedish-Talbanken and UD_Swedish-LinES. The treebanks contain mainly news and non-fiction.

Evaluation result on the joint Talbanken and LinES test sets: 96.37

XPOS tagger

Since the Talbanken and LinES treebanks are annotated with different XPOS tagsets we could not use them to train the XPOS model. Instead we used Talbanken and the Stockholm-Umeå Corpus v3.0 (SUC 3.0), which is a balanced corpus containing different types of news text, as well as fictional and non-fictional prose.

Evaluation result on the Talbanken test set plus a held-out portion of SUC 3.0: 96.84

Dependency parser

The parser is trained on the Talbanken and LinES treebanks, and is equivalent in both models.

Evaluation results on the joint Talbanken and LinES test sets:
Unlabelled Attachment Score (UAS) 87.03
Labelled Attachment Score (LAS) 82.20

NER

Named Entity Recognition was trained on the Stockholm Umeå Corpus v3.0 (SUC 3.0). Unlike pos tags, entities in SUC 3.0 were not annotated by human annotators, but automatically generated using Sparv. This means that they cannot be considered a gold standard.

Evaluation results on a held-out portion of SUC 3.0:
Precision 86.27
Recall 84.48
F score 85.37

Usage

Make sure that you have git and git-lfs installed on your system. To download the models you can go to https://git.kb.se/kblabb/static and use the following commands to only get the files that you are interested in:

$ GIT_LFS_SKIP_SMUDGE=1 git clone https://git.kb.se/kblabb/static
$ cd static
$ git lfs pull --include your_file.zip

In order to load the models into spaCy and use them you can look at the snippet below, or go to the spaCy official homepage for much more in-depth information.

import spacy
nlp = spacy.load("model_name")
doc = nlp("Jag gillar London och Berlin.")
width = 15
for token in doc:
    print(f"{token.text: <{width}} {token.tag_: <{width}} {token.dep_: <{width}}")
    
for ent in doc.ents:
    print(ent.text, ent.label_)

Expected output:

Jag             PRON            nsubj          
gillar          VERB            ROOT           
London          PROPN           obj            
och             CCONJ           cc             
Berlin          PROPN           conj           
.               PUNCT           punct          
London LOC
Berlin LOC

IMPORTANT:
In the UPOS model the part-of-speech tags are accessible via token.tag_, while token.pos_ contains only unknown tags (X).
In the XPOS model the trained pos tags can be accessed via token.tag_, whereas token.pos_ contains UPOS tags generated through the Swedish tagmap that maps XPOS to UPOS, so they are not trained. This means that in the XPOS model, if the XPOS tag is incorrect, the UPOS tag is probably also incorrect.

Acknowledgments

Resources from Universal Dependencies were used to train the tagger and parser.
Resources from Stockholms University, Umeå University and Språkbanken at Gothenburg University were used to train the models for NER.
A script by @EmilStenstrom was used to convert SUC 3.0 data into IOB format (which is compatible with the spaCy converter for NER). A modified version of the script was used to extract XPOS tags from SUC 3.0 in .conllu format in order to train one of the pos taggers.
Many thanks to Joakim Nivre and Sara Stymne at Uppsala University for input regarding the choice of training data and licensing issues.

dpriskorn/swedish-spacy