Biomedical Entity Linking with Contrastive Context Matching

Shogo Ujiie,Hayate Iso,Eiji Aramaki

DOI: https://doi.org/10.48550/arXiv.2106.07583

2021-06-14

Computation and Language

Abstract:We introduce BioCoM, a contrastive learning framework for biomedical entity linking that uses only two resources: a small-sized dictionary and a large number of raw biomedical articles. Specifically, we build the training instances from raw PubMed articles by dictionary matching and use them to train a context-aware entity linking model with contrastive learning. We predict the normalized biomedical entity at inference time through a nearest-neighbor search. Results found that BioCoM substantially outperforms state-of-the-art models, especially in low-resource settings, by effectively using the context of the entities.