Milan Straka commited on
Commit ·
cf37530
1
Parent(s): f4443d5
Add corpipe25-corefud1.3-large-251101 model.
Browse files
README.md
ADDED
|
@@ -0,0 +1,116 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: cc-by-nc-sa-4.0
|
| 3 |
+
language: [ca, cs, cu, de, en, es, fr, grc, hbo, hi, hu, ko, lt, no, pl, ru, tr]
|
| 4 |
+
tags: [ CorPipe, CorPipe 25, CRAC Shared Task, CRAC 2025, CorefUD, CorefUD 1.3, Coreference Resolution, Minnt]
|
| 5 |
+
base_model: google/mt5-large
|
| 6 |
+
---
|
| 7 |
+
|
| 8 |
+
# The `corpipe25-corefud1.3-large-251101` Model
|
| 9 |
+
|
| 10 |
+
The `corpipe25-corefud1.3-large-251101` is a `mT5-large`-based multilingual model for
|
| 11 |
+
coreference resolution usable in CorPipe 25 <https://github.com/ufal/crac2025-corpipe>.
|
| 12 |
+
It is released on [LINDAT/CLARIAH-CZ](https://hdl.handle.net/11234/1-6079) and on
|
| 13 |
+
[HuggingFace](https://huggingface.co/ufal/corpipe25-corefud1.3-large-251101) under the CC BY-NC-SA 4.0 license.
|
| 14 |
+
The model is downloaded automatically from HuggingFace when running prediction with
|
| 15 |
+
the `--load ufal/corpipe25-corefud1.3-large-251101` argument.
|
| 16 |
+
|
| 17 |
+
The model is language agnostic, so it can be in theory used to predict
|
| 18 |
+
coreference in any `mT5` language; for zero-shot cross-lingual evaluation,
|
| 19 |
+
please refer to the CRAC 2025 paper.
|
| 20 |
+
|
| 21 |
+
The model expects empty nodes to be already present on input, predicted by
|
| 22 |
+
https://github.com/ufal/crac2025_empty_nodes_baseline.
|
| 23 |
+
|
| 24 |
+
The model was trained using the following command (see the CorPipe 25 repository
|
| 25 |
+
for more information):
|
| 26 |
+
```sh
|
| 27 |
+
tbs="ca_ancora cs_pcedt cs_pdt cu_proiel de_potsdamcc en_gum en_litbank es_ancora fr_ancor fr_democrat grc_proiel hbo_ptnk hi_hdtb hu_korkor hu_szegedkoref ko_ecmt lt_lcc no_bokmaalnarc no_nynorsknarc pl_pcc ru_rucor tr_itcc"
|
| 28 |
+
|
| 29 |
+
python3 corpipe25.py --train --dev --treebanks $(for c in $tbs; do echo data/$c/$c-corefud-train.conllu; done) --batch_size=8 --learning_rate=6e-4 --learning_rate_decay --adafactor --encoder=google/mt5-large --exp=corpipe25-corefud1.3-large --compile
|
| 30 |
+
```
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
## CorefUD 1.3 Test Sets Results
|
| 34 |
+
|
| 35 |
+
The model achieves the following CorefUD 1.3 test set results (as reported in
|
| 36 |
+
the paper); segment size 2560 was used, with the exception for `cu_proiel` and
|
| 37 |
+
`grc_proiel` where it was 512:
|
| 38 |
+
|
| 39 |
+
| avg | ca | cs_pce| cs_pdt| cu | de_pot| en_gum| en_lit| es | fr_anc| fr_dem| grc | hbo_pt| hi | hu_kor| hu_sze| ko_emc| lt | no_bok| no_nyn| pl | ru | tr |
|
| 40 |
+
|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------|
|
| 41 |
+
| 72.84 | 80.1 | 74.6 | 78.0 | 58.5 | 67.2 | 73.3 | 77.4 | 82.0 | 72.1 | 68.5 | 71.2 | 67.9 | 76.3 | 67.3 | 68.0 | 69.8 | 74.4 | 75.2 | 74.0 | 77.5 | 81.2 | 67.7 |
|
| 42 |
+
|
| 43 |
+
|
| 44 |
+
## Running the Model on Plain Text
|
| 45 |
+
|
| 46 |
+
To run the model on plain text, first the plain text needs to be tokenized and
|
| 47 |
+
converted to CoNLL-U (and optionally parsed if you also want mention heads),
|
| 48 |
+
by using for example UDPipe 2:
|
| 49 |
+
|
| 50 |
+
```sh
|
| 51 |
+
curl -F data="Eve came home and Peter greeted her there. Then Peter and Paul set out to a trip and Eve waved them off." \
|
| 52 |
+
-F model=english -F tokenizer= -F tagger= -F parser= https://lindat.mff.cuni.cz/services/udpipe/api/process \
|
| 53 |
+
| python -X utf8 -c "import sys,json; sys.stdout.write(json.load(sys.stdin)['result'])" >input.conllu
|
| 54 |
+
```
|
| 55 |
+
|
| 56 |
+
Then the CoNLL-U file can be processed by CorPipe 25, by using for example
|
| 57 |
+
```sh
|
| 58 |
+
python3 corpipe25.py --load ufal/corpipe25-corefud1.3-large-251101 --exp . --epoch 0 --test input.conllu
|
| 59 |
+
```
|
| 60 |
+
which would generate the following predictions in `input.00.conllu`:
|
| 61 |
+
```
|
| 62 |
+
# generator = UDPipe 2, https://lindat.mff.cuni.cz/services/udpipe
|
| 63 |
+
# udpipe_model = english-ewt-ud-2.17-251125
|
| 64 |
+
# udpipe_model_licence = CC BY-NC-SA
|
| 65 |
+
# newdoc
|
| 66 |
+
# global.Entity = eid-etype-head-other
|
| 67 |
+
# newpar
|
| 68 |
+
# sent_id = 1
|
| 69 |
+
# text = Eve came home and Peter greeted her there.
|
| 70 |
+
1 Eve Eve PROPN NNP Number=Sing 2 nsubj _ Entity=(c1--1)
|
| 71 |
+
2 came come VERB VBD Mood=Ind|Number=Sing|Person=3|Tense=Past|VerbForm=Fin 0 root _ _
|
| 72 |
+
3 home home ADV RB _ 2 advmod _ Entity=(c2--1)
|
| 73 |
+
4 and and CCONJ CC _ 6 cc _ _
|
| 74 |
+
5 Peter Peter PROPN NNP Number=Sing 6 nsubj _ Entity=(c3--1)
|
| 75 |
+
6 greeted greet VERB VBD Mood=Ind|Number=Sing|Person=3|Tense=Past|VerbForm=Fin 2 conj _ _
|
| 76 |
+
7 her she PRON PRP Case=Acc|Gender=Fem|Number=Sing|Person=3|PronType=Prs 6 obj _ Entity=(c1--1)
|
| 77 |
+
8 there there ADV RB PronType=Dem 6 advmod _ Entity=(c2--1)|SpaceAfter=No
|
| 78 |
+
9 . . PUNCT . _ 2 punct _ _
|
| 79 |
+
|
| 80 |
+
# sent_id = 2
|
| 81 |
+
# text = Then Peter and Paul set out to a trip and Eve waved them off.
|
| 82 |
+
1 Then then ADV RB PronType=Dem 5 advmod _ _
|
| 83 |
+
2 Peter Peter PROPN NNP Number=Sing 5 nsubj _ Entity=(c4--1(c3--1)
|
| 84 |
+
3 and and CCONJ CC _ 4 cc _ _
|
| 85 |
+
4 Paul Paul PROPN NNP Number=Sing 2 conj _ Entity=(c5--1)c4)
|
| 86 |
+
5 set set VERB VBD Mood=Ind|Number=Plur|Person=3|Tense=Past|VerbForm=Fin 0 root _ _
|
| 87 |
+
6 out out ADP RP _ 5 compound:prt _ _
|
| 88 |
+
7 to to ADP IN _ 9 case _ _
|
| 89 |
+
8 a a DET DT Definite=Ind|PronType=Art 9 det _ Entity=(c6--2
|
| 90 |
+
9 trip trip NOUN NN Number=Sing 5 obl _ Entity=c6)
|
| 91 |
+
10 and and CCONJ CC _ 12 cc _ _
|
| 92 |
+
11 Eve Eve PROPN NNP Number=Sing 12 nsubj _ Entity=(c1--1)
|
| 93 |
+
12 waved wave VERB VBD Mood=Ind|Number=Sing|Person=3|Tense=Past|VerbForm=Fin 5 conj _ _
|
| 94 |
+
13 them they PRON PRP Case=Acc|Number=Plur|Person=3|PronType=Prs 12 obj _ Entity=(c4--1)
|
| 95 |
+
14 off off ADP RP _ 12 compound:prt _ SpaceAfter=No
|
| 96 |
+
15 . . PUNCT . _ 5 punct _ SpaceAfter=No
|
| 97 |
+
|
| 98 |
+
```
|
| 99 |
+
|
| 100 |
+
## How to Cite
|
| 101 |
+
|
| 102 |
+
```
|
| 103 |
+
@inproceedings{straka-2025-corpipe,
|
| 104 |
+
title = "{C}or{P}ipe at {CRAC} 2025: Evaluating Multilingual Encoders for Multilingual Coreference Resolution",
|
| 105 |
+
author = "Straka, Milan",
|
| 106 |
+
editor = "Ogrodniczuk, Maciej and Novak, Michal and Poesio, Massimo and Pradhan, Sameer and Ng, Vincent",
|
| 107 |
+
booktitle = "Proceedings of the Eighth Workshop on Computational Models of Reference, Anaphora and Coreference",
|
| 108 |
+
month = nov,
|
| 109 |
+
year = "2025",
|
| 110 |
+
address = "Suzhou, China",
|
| 111 |
+
publisher = "Association for Computational Linguistics",
|
| 112 |
+
url = "https://aclanthology.org/2025.crac-1.11/",
|
| 113 |
+
doi = "10.18653/v1/2025.crac-1.11",
|
| 114 |
+
pages = "130--139",
|
| 115 |
+
}
|
| 116 |
+
```
|
model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:72072e9928d1379d5f30d9c6c44c7e7174c4e3556d5752b57d75bc77f80033e8
|
| 3 |
+
size 2376264885
|
options.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"adafactor": true,
|
| 3 |
+
"batch_size": 8,
|
| 4 |
+
"compile": true,
|
| 5 |
+
"depth": 5,
|
| 6 |
+
"dev": [],
|
| 7 |
+
"encoder": "google/mt5-large",
|
| 8 |
+
"epochs": 15,
|
| 9 |
+
"exp": "mt5-large-s54",
|
| 10 |
+
"label_smoothing": 0.2,
|
| 11 |
+
"learning_rate": 0.0006,
|
| 12 |
+
"learning_rate_decay": true,
|
| 13 |
+
"load": [],
|
| 14 |
+
"logdir": "logs/mt5-large-s54-corpipe25--250624_153459-a=True,bs=8,c=True,d=5,d=,e=mt5-large,eb=10000,e=15,ls=0.2,lr=0.0006,lrd=True,r=50,se=0.5,sm=sentences,s=54,s=512,t=,t=True,t=ca_ancora-corefud-train.conllu,...,w=0.1",
|
| 15 |
+
"right": 50,
|
| 16 |
+
"sampling_exponent": 0.5,
|
| 17 |
+
"sampling_mode": "sentences",
|
| 18 |
+
"seed": 54,
|
| 19 |
+
"segment": 512,
|
| 20 |
+
"steps_per_epoch": 10000,
|
| 21 |
+
"test": [],
|
| 22 |
+
"threads": 8,
|
| 23 |
+
"train": true,
|
| 24 |
+
"treebanks": [
|
| 25 |
+
"data/ca_ancora/ca_ancora-corefud-train.conllu",
|
| 26 |
+
"data/cs_pcedt/cs_pcedt-corefud-train.conllu",
|
| 27 |
+
"data/cs_pdt/cs_pdt-corefud-train.conllu",
|
| 28 |
+
"data/cu_proiel/cu_proiel-corefud-train.conllu",
|
| 29 |
+
"data/de_potsdamcc/de_potsdamcc-corefud-train.conllu",
|
| 30 |
+
"data/en_gum/en_gum-corefud-train.conllu",
|
| 31 |
+
"data/en_litbank/en_litbank-corefud-train.conllu",
|
| 32 |
+
"data/es_ancora/es_ancora-corefud-train.conllu",
|
| 33 |
+
"data/fr_ancor/fr_ancor-corefud-train.conllu",
|
| 34 |
+
"data/fr_democrat/fr_democrat-corefud-train.conllu",
|
| 35 |
+
"data/grc_proiel/grc_proiel-corefud-train.conllu",
|
| 36 |
+
"data/hbo_ptnk/hbo_ptnk-corefud-train.conllu",
|
| 37 |
+
"data/hi_hdtb/hi_hdtb-corefud-train.conllu",
|
| 38 |
+
"data/hu_korkor/hu_korkor-corefud-train.conllu",
|
| 39 |
+
"data/hu_szegedkoref/hu_szegedkoref-corefud-train.conllu",
|
| 40 |
+
"data/ko_ecmt/ko_ecmt-corefud-train.conllu",
|
| 41 |
+
"data/lt_lcc/lt_lcc-corefud-train.conllu",
|
| 42 |
+
"data/no_bokmaalnarc/no_bokmaalnarc-corefud-train.conllu",
|
| 43 |
+
"data/no_nynorsknarc/no_nynorsknarc-corefud-train.conllu",
|
| 44 |
+
"data/pl_pcc/pl_pcc-corefud-train.conllu",
|
| 45 |
+
"data/ru_rucor/ru_rucor-corefud-train.conllu",
|
| 46 |
+
"data/tr_itcc/tr_itcc-corefud-train.conllu"
|
| 47 |
+
],
|
| 48 |
+
"warmup": 0.1
|
| 49 |
+
}
|
tags.txt
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
POP:1
|
| 3 |
+
POP:1,POP:1
|
| 4 |
+
POP:1,POP:1,POP:1
|
| 5 |
+
POP:1,POP:1,POP:1,POP:1
|
| 6 |
+
POP:1,POP:1,POP:1,POP:1,POP:1
|
| 7 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1
|
| 8 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1
|
| 9 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1
|
| 10 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1
|
| 11 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1
|
| 12 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1
|
| 13 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,PUSH,POP:1
|
| 14 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,PUSH,POP:1
|
| 15 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,PUSH,POP:1
|
| 16 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,PUSH,POP:1
|
| 17 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,POP:2,PUSH,POP:1
|
| 18 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,POP:1,PUSH,POP:1
|
| 19 |
+
POP:1,POP:1,POP:1,POP:1,POP:1,PUSH,POP:1
|
| 20 |
+
POP:1,POP:1,POP:1,POP:1,PUSH,POP:1
|
| 21 |
+
POP:1,POP:1,POP:1,POP:2
|
| 22 |
+
POP:1,POP:1,POP:1,POP:2,PUSH,POP:1
|
| 23 |
+
POP:1,POP:1,POP:1,PUSH,POP:1
|
| 24 |
+
POP:1,POP:1,POP:2
|
| 25 |
+
POP:1,POP:1,POP:2,POP:2
|
| 26 |
+
POP:1,POP:1,POP:2,PUSH,POP:1
|
| 27 |
+
POP:1,POP:1,POP:3
|
| 28 |
+
POP:1,POP:1,PUSH
|
| 29 |
+
POP:1,POP:1,PUSH,POP:1
|
| 30 |
+
POP:1,POP:1,PUSH,PUSH,POP:1
|
| 31 |
+
POP:1,POP:2
|
| 32 |
+
POP:1,POP:2,POP:2
|
| 33 |
+
POP:1,POP:2,POP:2,POP:2
|
| 34 |
+
POP:1,POP:2,PUSH,POP:1
|
| 35 |
+
POP:1,POP:3,PUSH,POP:1
|
| 36 |
+
POP:1,PUSH
|
| 37 |
+
POP:1,PUSH,POP:1
|
| 38 |
+
POP:1,PUSH,PUSH,POP:1
|
| 39 |
+
POP:2
|
| 40 |
+
POP:2,POP:2
|
| 41 |
+
POP:2,POP:2,POP:2
|
| 42 |
+
POP:2,POP:2,PUSH,POP:1
|
| 43 |
+
POP:2,PUSH,POP:1
|
| 44 |
+
POP:3
|
| 45 |
+
POP:3,PUSH,POP:1
|
| 46 |
+
POP:4
|
| 47 |
+
PUSH
|
| 48 |
+
PUSH,POP:1
|
| 49 |
+
PUSH,PUSH
|
| 50 |
+
PUSH,PUSH,POP:1
|
| 51 |
+
PUSH,PUSH,PUSH
|
| 52 |
+
PUSH,PUSH,PUSH,POP:1
|
| 53 |
+
PUSH,PUSH,PUSH,PUSH
|
| 54 |
+
PUSH,PUSH,PUSH,PUSH,POP:1
|
| 55 |
+
PUSH,PUSH,PUSH,PUSH,PUSH
|
| 56 |
+
PUSH,PUSH,PUSH,PUSH,PUSH,POP:1
|