English

Error-Aware Curriculum Learning for Biomedical Relation Classification

Computation and Language 2025-07-22 v1

Abstract

Relation Classification (RC) in biomedical texts is essential for constructing knowledge graphs and enabling applications such as drug repurposing and clinical decision-making. We propose an error-aware teacher--student framework that improves RC through structured guidance from a large language model (GPT-4o). Prediction failures from a baseline student model are analyzed by the teacher to classify error types, assign difficulty scores, and generate targeted remediations, including sentence rewrites and suggestions for KG-based enrichment. These enriched annotations are used to train a first student model via instruction tuning. This model then annotates a broader dataset with difficulty scores and remediation-enhanced inputs. A second student is subsequently trained via curriculum learning on this dataset, ordered by difficulty, to promote robust and progressive learning. We also construct a heterogeneous biomedical knowledge graph from PubMed abstracts to support context-aware RC. Our approach achieves new state-of-the-art performance on 4 of 5 PPI datasets and the DDI dataset, while remaining competitive on ChemProt.

Keywords

Cite

@article{arxiv.2507.14374,
  title  = {Error-Aware Curriculum Learning for Biomedical Relation Classification},
  author = {Sinchani Chakraborty and Sudeshna Sarkar and Pawan Goyal},
  journal= {arXiv preprint arXiv:2507.14374},
  year   = {2025}
}

Comments

16 pages, 2 figures

R2 v1 2026-07-01T04:08:47.437Z