GRAFT: Gated Retrieval-Augmented Fine-Tuning for Relation Extraction

Yuhang Jiang, Ramakanth Kavuluru


Abstract
Even in the era of large language models (LLMs), biomedical relation extraction (RE) still plays a major role in timely creation of knowledge graphs that further guide biomedical knowledge discovery. The main task in RE is to extract a relation "as expressed" in an input text. At times, crucial definitional information or other auxiliary information about the entities involved may be missing from the input text. Augmenting it from other external textual sources appears helpful on the surface but can be harmful too, as these sources can overwhelm the signal in the original input, leading to false positives or false negatives. To counter this, we leverage a pre-trained biomedical text retriever to augment original inputs with additional instance-specific snippets. This is done through a gating mechanism that allows the retrieved snippets to enhance but not overwhelm the signal from the original input. We evaluate our approach on three standard biomedical relation extraction datasets (CDR, BioRED, and ChemProt) and show consistent improvements (up to 10 F1 points) compared with strong supervised baselines involving both encoder and decoder models. All our code and the datasets used are available for reuse: \url{https://github.com/bionlproc/GRAFT-RE}.
Anthology ID:
2026.bionlp-1.74
Volume:
BioNLP 2026
Month:
July
Year:
2026
Address:
San Diego, California
Editors:
Dina Demner-Fushman, Sophia Ananiadou, Kirk Roberts, Junichi Tsujii
Venues:
BioNLP | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
920–931
Language:
URL:
https://preview.aclanthology.org/ingest-acl-workshops/2026.bionlp-1.74/
DOI:
Bibkey:
Cite (ACL):
Yuhang Jiang and Ramakanth Kavuluru. 2026. GRAFT: Gated Retrieval-Augmented Fine-Tuning for Relation Extraction. In BioNLP 2026, pages 920–931, San Diego, California. Association for Computational Linguistics.
Cite (Informal):
GRAFT: Gated Retrieval-Augmented Fine-Tuning for Relation Extraction (Jiang & Kavuluru, BioNLP 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl-workshops/2026.bionlp-1.74.pdf