๐ค AI Summary
To address the challenges of poor robustness, high energy consumption, and difficulty in adapting large language models (LLMs) in low-supervision relation extraction (RE), this paper proposes a fine-tuning-free, modular, and energy-efficient RE framework. Methodologically, it innovatively integrates supervised multi-label contrastive learning with a Bayesian k-nearest neighbors (kNN) classifier to effectively mitigate noise inherent in distant supervision. It further introduces two fine-grained evaluation metricsโClass-wise Semantic Discriminability (CSD) and Precision-at-R (P@R)โand releases Wiki20d, a benchmark dataset designed to reflect realistic deployment scenarios. Experiments demonstrate that our approach achieves or surpasses state-of-the-art performance across five mainstream benchmarks while substantially reducing computational energy consumption. Ablation studies validate the advantages of its minimalist architecture in terms of robustness, zero-shot transferability, and seamless plug-and-play integration with LLMs, establishing an efficient and practical paradigm for automated knowledge graph expansion.
๐ Abstract
The growing demand for efficient knowledge graph (KG) enrichment leveraging external corpora has intensified interest in relation extraction (RE), particularly under low-supervision settings. To address the need for adaptable and noise-resilient RE solutions that integrate seamlessly with pre-trained large language models (PLMs), we introduce SCoRE, a modular and cost-effective sentence-level RE system. SCoRE enables easy PLM switching, requires no finetuning, and adapts smoothly to diverse corpora and KGs. By combining supervised contrastive learning with a Bayesian k-Nearest Neighbors (kNN) classifier for multi-label classification, it delivers robust performance despite the noisy annotations of distantly supervised corpora. To improve RE evaluation, we propose two novel metrics: Correlation Structure Distance (CSD), measuring the alignment between learned relational patterns and KG structures, and Precision at R (P@R), assessing utility as a recommender system. We also release Wiki20d, a benchmark dataset replicating real-world RE conditions where only KG-derived annotations are available. Experiments on five benchmarks show that SCoRE matches or surpasses state-of-the-art methods while significantly reducing energy consumption. Further analyses reveal that increasing model complexity, as seen in prior work, degrades performance, highlighting the advantages of SCoRE's minimal design. Combining efficiency, modularity, and scalability, SCoRE stands as an optimal choice for real-world RE applications.