Skip to content

Files

Latest commit

23fd7ff · Feb 3, 2023

History

History

NPCRF

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Feb 3, 2023
Nov 17, 2022
Nov 18, 2022

Neural Pairwise Conditional Random Field for Ultra-Fine Entity Typing

The NPCRF module proposed in the EMNLP 2022 paper: Modeling Label Correlations for Ultra-Fine Entity Typing with Neural Pairwise Conditional Random Field. NPCRF performs mean-field variational inference on a probabilistic model designed for better modeling label correlations in ultra-fine entity typing task.

Static Label Embeding

NPCRF requires static label embeddings, the preprocessed label embeddings (from GloVe for EN, Tencent for ZH) can be downloaded here: UFET, CFET, and you can place them in yoru folder and run the following config: (you need to reset your target_emb_dir in the config). Or you can provide the path of the glove embedding file (e.g., /path/to/your/glove.6B.300d.txt) and the code will generate label embedding for you.

Example of Training a model with NPCRF

python -m scripts.train -c examples/npcrf/configs/ufet_concat_npcrf.yaml

Benchmarks

UFET ma-F1 ma-P ma-R
NPCRF-roberta-large 47.1 49.5 44.9