Self-Alignment Pretraining for Biomedical Entity Representations
Not on the map. Why?
Abstract (arXiv)
Despite the widespread success of self-supervised learning via masked language models (MLM), accurately capturing fine-grained semantic relationships in the biomedical domain remains a challenge. This is of paramount importance for entity-level tasks such as entity linking where the ability to model entity relations (especially synonymy) is pivotal. To address this challenge, we propose SapBERT, a pretraining scheme that self-aligns the representation space of biomedical entities. We design a scalable metric learning framework that can leverage UMLS, a massive collection of biomedical ontologies with 4M+ concepts. In contrast with previous pipeline-based hybrid systems, SapBERT offers an elegant one-model-for-all solution to the problem of medical entity linking (MEL), achieving a new state-of-the-art (SOTA) on six MEL benchmarking datasets. In the scientific domain, we achieve SOTA even without task-specific supervision. With substantial improvement over various domain-specific pretrained MLMs such as BioBERT, SciBERTand and PubMedBERT, our pretraining scheme proves to be both effective and robust.
Key facts. Cited by 234 papers in the map (data through 13 Oct 2026). First posted on arXiv 22 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 23; 2022: 32; 2023: 48; 2024: 38; 2025: 39; 2026: 54.
234 papers.
Sort:
By area:
- 2021QA-GNN: Reasoning with Language Models and Knowledge Graphs for Question Answeringcited by 386
- 2022GreaseLM: Graph REASoning Enhanced Language Models for Question Answeringcited by 170
- 2023A Survey of Large Language Models for Healthcare: from Data, Technology, and Applications to Accountability and Ethicscited by 163
- 2022Deep Bidirectional Language-Knowledge Graph Pretrainingcited by 156
- 2021AMMUS : A Survey of Transformer-based Pretrained Models in Natural Language Processingcited by 131
- 2024A Comprehensive Survey of Scientific Large Language Models and Their Applications in Scientific Discoverycited by 96
- 2022RoentGen: Vision-Language Foundation Model for Chest X-ray Generationcited by 89
- 2021Fast, Effective, and Self-Supervised: Transforming Masked Language Models into Universal Lexical and Sentence Encoderscited by 88
- 2022Adapting Pretrained Vision-Language Foundational Models to Medical Imaging Domainscited by 85
- 2022BioBART: Pretraining and Evaluation of A Biomedical Generative Language Modelcited by 83
- 2023A Survey of GPT-3 Family Large Language Models Including ChatGPT and GPT-4cited by 81
- 2023Matching Patients to Clinical Trials with Large Language Modelscited by 69
38 more
- 2022Knowledge Injected Prompt Based Fine-tuning for Multi-label Few-shot ICD Codingcited by 40
- 2021Does the Magic of BERT Apply to Medical Code Assignment? A Quantitative Studycited by 39
- 2024BMRetriever: Tuning Large Language Models as Better Biomedical Text Retrieverscited by 38
- 2023GeoLM: Empowering Language Models for Geospatially Grounded Language Understandingcited by 37
- 2023OLaLa: Ontology Matching with Large Language Modelscited by 35
- 2022Generative Biomedical Entity Linking via Knowledge Base-Guided Pre-training and Synonyms-Aware Fine-tuningcited by 34
- 2021AMMU : A Survey of Transformer-based Biomedical Pretrained Language Modelscited by 32
- 2021TaCL: Improving BERT Pre-training with Token-aware Contrastive Learningcited by 32
- 2021Knowledge-Rich Self-Supervision for Biomedical Entity Linkingcited by 31
- 2022Summarizing Patients Problems from Hospital Progress Notes Using Pre-trained Sequence-to-Sequence Modelscited by 30
- 2023Leveraging Medical Knowledge Graphs Into Large Language Models for Diagnosis Prediction: Design and Application Studycited by 30
- 2023KnowGPT: Knowledge Graph based Prompting for Large Language Modelscited by 30
- 2021Learning Domain-Specialised Representations for Cross-Lingual Biomedical Entity Linkingcited by 27
- 2024Understand What LLM Needs: Dual Preference Alignment for Retrieval-Augmented Generationcited by 27
- 2024Language Models as Hierarchy Encoderscited by 26
- 2025Clinical ModernBERT: An efficient and long context encoder for biomedical textcited by 26
- 2021ConvFiT: Conversational Fine-Tuning of Pretrained Language Modelscited by 25
- 2023MKRAG: Medical Knowledge Retrieval Augmented Generation for Medical Question Answeringcited by 23
- 2021Entity Linking and Discovery via Arborescence-based Supervised Clusteringcited by 22
- 2021Rewire-then-Probe: A Contrastive Recipe for Probing Biomedical Knowledge of Pre-trained Language Modelscited by 21
- 2022BioLORD: Learning Ontological Representations from Definitions (for Biomedical Concepts and their Textual Descriptions)cited by 20
- 2021MirrorWiC: On Eliciting Word-in-Context Representations from Pretrained Language Modelscited by 18
- 2022Improving Word Translation via Two-Stage Contrastive Learningcited by 18
- 2023Overview of BioASQ 2023: The eleventh BioASQ challenge on Large-Scale Biomedical Semantic Indexing and Question Answeringcited by 18
- 2023GrapeQA: GRaph Augmentation and Pruning to Enhance Question-Answeringcited by 17
- 2021Cross-Domain Data Integration for Named Entity Disambiguation in Biomedical Textcited by 15
- 2024PromptLink: Leveraging Large Language Models for Cross-Source Biomedical Concept Linkingcited by 15
- 2021BERT might be Overkill: A Tiny but Effective Biomedical Entity Linker based on Residual Convolutional Neural Networkscited by 14
- 2021Mixture-of-Partitions: Infusing Large Biomedical Knowledge Graphs into BERTcited by 14
- 2022DR.BENCH: Diagnostic Reasoning Benchmark for Clinical Natural Language Processingcited by 14
- 2023Vision-Language Generative Model for View-Specific Chest X-ray Generationcited by 14
- 2023A Meta-Evaluation of Faithfulness Metrics for Long-Form Hospital-Course Summarizationcited by 14
- 2023Evaluation of large language models for discovery of gene set functioncited by 14
- 2024Ontology Embedding: A Survey of Methods, Applications and Resourcescited by 12
- 2025Large Language Models with Temporal Reasoning for Longitudinal Clinical Summarization and Predictioncited by 12
- 2025KERAP: A Knowledge-Enhanced Reasoning Approach for Accurate Zero-shot Diagnosis Prediction Using Multi-agent LLMscited by 12
- 2022Detection, Disambiguation, Re-ranking: Autoregressive Entity Linking as a Multi-Task Problemcited by 11
- 2022Embracing Ambiguity: Improving Similarity-oriented Tasks with Contextual Synonym Knowledgecited by 11
50 most cited of 234 All 234 (JSON)
- 2026Bridging the EHR Divide: Asymmetric Contrastive Learning for Cross-National Medical Representation Transfer4 Oct 2026
- 2026Can LLMs Reliably Annotate Bioassay Metadata to Improve Data Readiness?1 Oct 2026
- 2026Overview of BioASQ 2026: The fourteenth BioASQ Challenge on Large-Scale Biomedical Semantic Indexing and Question Answering30 Sep 2026
- 2026QureRadEmbed: Structuring Radiological Similarity through Attribute and Reasoning Supervision27 Sep 2026
- 2026Synthetic Hospital: An Open, Verifiable, Physician-Validated Longitudinal EHR Benchmark24 Sep 2026
- 2026BELXTR: Biomedical Entity Linking via Contextualized Token Retrieval22 Sep 2026
- 2026Ascent: An Agentic System over the Model Context Protocol for Real-World Clinical Data Analysis21 Sep 2026
- 2026ChemCLIR-Bench: Benchmarking Cross-Lingual Information Retrieval in Multilingual Chemical Patents19 Sep 2026
- 2026SNOMED CT Concept Recommendation from Masked Clinical Context15 Sep 2026
- 2026OntologyBench: Can Dense Retrieval Satisfy Structured Biomedical Constraints?8 Sep 2026
- 2026Import What You Need: Learning When and How to Augment EHR Graphs with External Knowledge1 Sep 2026
- 2026Bridging Lexical Divergence: LLM-Assisted, Cost-Efficient, Zero-shot Scientific Entity Linking31 Aug 2026
38 more
- 2026CoLa-ICD: A Knowledge-Enhanced Framework for Long-Tail Automated Medical Coding31 Aug 2026
- 2026From Location Phrases to Geographic Entities: Task-Adapted Retrieval for People Search29 Aug 2026
- 2026Aligning Biomedical Texts and Knowledge Graphs: A Systematic Comparison of Lightweight Alignment Strategies24 Aug 2026
- 2026CDEG: Learning Decision-Critical Evidence for Long-Horizon Diagnostic Agents24 Aug 2026
- 2026MedReaMM: Evaluating Large Multimodal Models on Expert-Level Clinical Diagnostic Synthesis23 Aug 2026
- 2026Clinically Structured Surrogate Rewards for Post-SFT Medical Image Captioning19 Aug 2026
- 2026A Cloud-Edge System for Multimodal Clinical Screening in Resource-Constrained Rural Settings13 Aug 2026
- 2026One Score, Two Decisions: Selective Prediction on the Rare-Disease Tail5 Aug 2026
- 2026Neighborhood-Aware Dual Biomedical Entity Linking4 Aug 2026
- 2026LLMBDC: Language Model for Biological Domains Oriented Clustering of Gene Ontology30 Jul 2026
- 2026VisKG-LM: Compiling Knowledge Graphs into Visual Memory for Multiple-Choice Question Answering29 Jul 2026
- 2026GraphRareBench: An Auditable Graph-Evidence Benchmark for Phenotype-Driven Rare-Disease Diagnosis27 Jul 2026
- 2026Medical-Checklist: Assessing the Comprehension of Medical Images by Multimodal Models24 Jul 2026
- 2026HERMES: Contrast-Aware Knowledge Graph Reasoning from Clinical Notes for Patient Outcome Prediction22 Jul 2026
- 2026OntoBook: Ontology-Grounded Synthetic Textbooks for Medical Encoder Pretraining21 Jul 2026
- 2026MedGuards: Multi-Agent System for Reliable Medical Error Detection and Correction24 Jun 2026
- 2026MMed-Bench-IR: A Heterogeneous Benchmark for Multilingual Medical Information Retrieval23 Jun 2026
- 2026RASC+: Retrieval-Constrained LLM Adjudication for Clinical Value Set Authoring22 Jun 2026
- 2026MedHal-Loc: Are "Explainable-by-Architecture" Medical Hallucination Detectors Faithful Localizers? A Localization Benchmark19 Jun 2026
- 2026Principles of Concept Representation in Sentence Encoders5 Jun 2026
- 2026Generalistic or Specific Embeddings, Which is Better? An Empirical Study on Search for Clinical Coding in Non-English Languages28 May 2026
- 2026MedCase-Structured: A Text-to-FHIR Dataset for Benchmarking Diagnostic Reasoning in Clinically Realistic EHR Settings28 May 2026
- 2026GRASP: Plan-Guided Graph Retrieval with Adaptive Fusion and Reranking on Semi-Structured Knowledge Bases28 May 2026
- 2026ChronoMedKG: A Temporally-Grounded Biomedical Knowledge Graph and Benchmark for Clinical Reasoning21 May 2026
- 2026BeLink: Biomedical Entity Linking Meets Generative Re-Ranking21 May 2026
- 2026Query-Conditioned Knowledge Alignment for Reliable Cross-System Medical Reasoning18 May 2026
- 2026Wasserstein Equilibrium Decoding for Reliable Medical Visual Question Answering18 May 2026
- 2026LongBEL: Long-Context and Document-Consistent Biomedical Entity Linking13 May 2026
- 2026Key Coverage Matters: Semi-Structured Extraction of OCR Clinical Reports10 May 2026
- 2026The Provenance Gap in Clinical AI: Evidence-Traceable Temporal Knowledge Graphs for Rare Disease Reasoning18 Apr 2026
- 2026Retrieve, Then Classify: Corpus-Grounded Automation of Clinical Value Set Authoring16 Apr 2026
- 2026BioELX: Context-Aware Cross-lingual Biomedical Entity Linking without Task-Specific Supervision9 Apr 2026
- 2026Graph-Aware Late Chunking for Retrieval-Augmented Generation in Biomedical Literature23 Mar 2026
- 2026MIRAGE: Knowledge Graph-Guided Cross-Cohort MRI Synthesis for Alzheimer's Disease Prediction2 Mar 2026
- 2026Asking the Right Questions: Ontology-Grounded Interpretable Embeddings for Biomedical Text2 Mar 2026
- 2026PVminer: A Domain-Specific Tool to Detect the Patient Voice in Patient Generated Data24 Feb 2026
- 2026CUICurate: A GraphRAG-based Framework for Automated Clinical Concept Curation for NLP applications20 Feb 2026
- 2026Think Like a Doctor: Conversational Diagnosis through the Exploration of Diagnostic Knowledge Graphs2 Feb 2026
50 newest of 234 All 234 (JSON)
- 2022GreaseLM: Graph REASoning Enhanced Language Models for Question Answeringcited by 170
- 2023GeoLM: Empowering Language Models for Geospatially Grounded Language Understandingcited by 37
- 2023GrapeQA: GRaph Augmentation and Pruning to Enhance Question-Answeringcited by 17
- 2022DR.BENCH: Diagnostic Reasoning Benchmark for Clinical Natural Language Processingcited by 14
- 2025Large Language Models with Temporal Reasoning for Longitudinal Clinical Summarization and Predictioncited by 12
- 2025KERAP: A Knowledge-Enhanced Reasoning Approach for Accurate Zero-shot Diagnosis Prediction Using Multi-agent LLMscited by 12
- 2024Thought Graph: Generating Thought Process for Biological Reasoningcited by 10
- 2022Structured Knowledge Grounding for Question Answeringcited by 2
- 2025Gene-R1: Reasoning with Data-Augmented Lightweight LLMs for Gene Set Analysiscited by 2
- 2026The Provenance Gap in Clinical AI: Evidence-Traceable Temporal Knowledge Graphs for Rare Disease Reasoningcited by 2
- 2026MedCase-Structured: A Text-to-FHIR Dataset for Benchmarking Diagnostic Reasoning in Clinically Realistic EHR Settingscited by 1
- 2026ChronoMedKG: A Temporally-Grounded Biomedical Knowledge Graph and Benchmark for Clinical Reasoningcited by 0
- 2026MedCase-Structured: A Text-to-FHIR Dataset for Benchmarking Diagnostic Reasoning in Clinically Realistic EHR Settings28 May 2026
- 2026ChronoMedKG: A Temporally-Grounded Biomedical Knowledge Graph and Benchmark for Clinical Reasoning21 May 2026
- 2026The Provenance Gap in Clinical AI: Evidence-Traceable Temporal Knowledge Graphs for Rare Disease Reasoning18 Apr 2026
- 2025Gene-R1: Reasoning with Data-Augmented Lightweight LLMs for Gene Set Analysis11 Sep 2025
- 2025KERAP: A Knowledge-Enhanced Reasoning Approach for Accurate Zero-shot Diagnosis Prediction Using Multi-agent LLMs3 Jul 2025
- 2025Large Language Models with Temporal Reasoning for Longitudinal Clinical Summarization and Prediction30 Jan 2025
- 2024Thought Graph: Generating Thought Process for Biological Reasoning11 Mar 2024
- 2023GeoLM: Empowering Language Models for Geospatially Grounded Language Understanding23 Oct 2023
- 2023GrapeQA: GRaph Augmentation and Pruning to Enhance Question-Answering22 Mar 2023
- 2022DR.BENCH: Diagnostic Reasoning Benchmark for Clinical Natural Language Processing29 Sep 2022
- 2022Structured Knowledge Grounding for Question Answering17 Sep 2022
- 2022GreaseLM: Graph REASoning Enhanced Language Models for Question Answering21 Jan 2022
- 2024Understand What LLM Needs: Dual Preference Alignment for Retrieval-Augmented Generationcited by 27
- 2023MKRAG: Medical Knowledge Retrieval Augmented Generation for Medical Question Answeringcited by 23
- 2024Retrieval augmented text-to-SQL generation for epidemiological question answering using electronic health recordscited by 8
- 2025CDE-Mapper: Using Retrieval-Augmented Language Models for Linking Clinical Data Elements to Controlled Vocabulariescited by 1
- 2026MedRAGChecker: Claim-Level Verification for Biomedical Retrieval-Augmented Generationcited by 1
- 2026Graph-Aware Late Chunking for Retrieval-Augmented Generation in Biomedical Literaturecited by 1
- 2026MMed-Bench-IR: A Heterogeneous Benchmark for Multilingual Medical Information Retrievalcited by 0
- 2026MMed-Bench-IR: A Heterogeneous Benchmark for Multilingual Medical Information Retrieval23 Jun 2026
- 2026Graph-Aware Late Chunking for Retrieval-Augmented Generation in Biomedical Literature23 Mar 2026
- 2026MedRAGChecker: Claim-Level Verification for Biomedical Retrieval-Augmented Generation10 Jan 2026
- 2025CDE-Mapper: Using Retrieval-Augmented Language Models for Linking Clinical Data Elements to Controlled Vocabularies7 May 2025
- 2024Understand What LLM Needs: Dual Preference Alignment for Retrieval-Augmented Generation26 Jun 2024
- 2024Retrieval augmented text-to-SQL generation for epidemiological question answering using electronic health records14 Mar 2024
- 2023MKRAG: Medical Knowledge Retrieval Augmented Generation for Medical Question Answering27 Sep 2023
- 2024GeneAgent: Self-verification Language Agent for Gene Set Knowledge Discovery using Domain Databasescited by 9
- 2025Agentic AI framework for End-to-End Medical Data Inferencecited by 8
- 2026MedGuards: Multi-Agent System for Reliable Medical Error Detection and Correctioncited by 0
- 2026Ascent: An Agentic System over the Model Context Protocol for Real-World Clinical Data Analysiscited by 0
- 2026Ascent: An Agentic System over the Model Context Protocol for Real-World Clinical Data Analysis21 Sep 2026
- 2026MedGuards: Multi-Agent System for Reliable Medical Error Detection and Correction24 Jun 2026
- 2025Agentic AI framework for End-to-End Medical Data Inference24 Jul 2025
- 2024GeneAgent: Self-verification Language Agent for Gene Set Knowledge Discovery using Domain Databases25 May 2024
- 2022RoentGen: Vision-Language Foundation Model for Chest X-ray Generationcited by 89
- 2022Adapting Pretrained Vision-Language Foundational Models to Medical Imaging Domainscited by 85
- 2026Wasserstein Equilibrium Decoding for Reliable Medical Visual Question Answeringcited by 0
- 2026MedReaMM: Evaluating Large Multimodal Models on Expert-Level Clinical Diagnostic Synthesiscited by 0
- 2026MedReaMM: Evaluating Large Multimodal Models on Expert-Level Clinical Diagnostic Synthesis23 Aug 2026
- 2026Wasserstein Equilibrium Decoding for Reliable Medical Visual Question Answering18 May 2026
- 2022RoentGen: Vision-Language Foundation Model for Chest X-ray Generation23 Nov 2022
- 2022Adapting Pretrained Vision-Language Foundational Models to Medical Imaging Domains9 Oct 2022
- 2021QA-GNN: Reasoning with Language Models and Knowledge Graphs for Question Answeringcited by 386
- 2023A Survey of Large Language Models for Healthcare: from Data, Technology, and Applications to Accountability and Ethicscited by 163
- 2022Deep Bidirectional Language-Knowledge Graph Pretrainingcited by 156
- 2021AMMUS : A Survey of Transformer-based Pretrained Models in Natural Language Processingcited by 131
- 2024A Comprehensive Survey of Scientific Large Language Models and Their Applications in Scientific Discoverycited by 96
- 2021Fast, Effective, and Self-Supervised: Transforming Masked Language Models into Universal Lexical and Sentence Encoderscited by 88
- 2022BioBART: Pretraining and Evaluation of A Biomedical Generative Language Modelcited by 83
- 2023A Survey of GPT-3 Family Large Language Models Including ChatGPT and GPT-4cited by 81
- 2023Matching Patients to Clinical Trials with Large Language Modelscited by 69
- 2022Knowledge Injected Prompt Based Fine-tuning for Multi-label Few-shot ICD Codingcited by 40
- 2021Does the Magic of BERT Apply to Medical Code Assignment? A Quantitative Studycited by 39
- 2024BMRetriever: Tuning Large Language Models as Better Biomedical Text Retrieverscited by 38
12 most cited of 206 All 206 (JSON)
- 2026Bridging the EHR Divide: Asymmetric Contrastive Learning for Cross-National Medical Representation Transfer4 Oct 2026
- 2026Can LLMs Reliably Annotate Bioassay Metadata to Improve Data Readiness?1 Oct 2026
- 2026Overview of BioASQ 2026: The fourteenth BioASQ Challenge on Large-Scale Biomedical Semantic Indexing and Question Answering30 Sep 2026
- 2026QureRadEmbed: Structuring Radiological Similarity through Attribute and Reasoning Supervision27 Sep 2026
- 2026Synthetic Hospital: An Open, Verifiable, Physician-Validated Longitudinal EHR Benchmark24 Sep 2026
- 2026BELXTR: Biomedical Entity Linking via Contextualized Token Retrieval22 Sep 2026
- 2026ChemCLIR-Bench: Benchmarking Cross-Lingual Information Retrieval in Multilingual Chemical Patents19 Sep 2026
- 2026SNOMED CT Concept Recommendation from Masked Clinical Context15 Sep 2026
- 2026OntologyBench: Can Dense Retrieval Satisfy Structured Biomedical Constraints?8 Sep 2026
- 2026Import What You Need: Learning When and How to Augment EHR Graphs with External Knowledge1 Sep 2026
- 2026Bridging Lexical Divergence: LLM-Assisted, Cost-Efficient, Zero-shot Scientific Entity Linking31 Aug 2026
- 2026CoLa-ICD: A Knowledge-Enhanced Framework for Long-Tail Automated Medical Coding31 Aug 2026
12 newest of 206 All 206 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025RELATE: Relation Extraction in Biomedical Abstracts with LLMs and Ontology Constraints
- 2022LinkBERT: Pretraining Language Models with Document Links
- 2026Learning from Medical Entity Trees: An Entity-Centric Medical Data Engineering Framework for MLLMs
- 2024LLMs are not Zero-Shot Reasoners for Biomedical Information Extraction
- 2023Inspire the Large Language Model by External Knowledge on BioMedical Named Entity Recognition
- 2024LLMs in Biomedicine: A study on clinical Named Entity Recognition
- 2026ANCHOR-RE: An Agentic Neuro-Symbolic Framework for Grounded Biomedical Relation Extraction
- 2026Efficient and Reliable Estimation of Named Entity Linking Quality: A Case Study on GutBrainIE