cd /news/large-language-models/repair-before-reinforce-context-augm… · home topics large-language-models article
[ARTICLE · art-128724] src=arxiv.org ↗ pub= topic=large-language-models verified=true sentiment=↑ positive

Repair Before Reinforce: Context-Augmented Knowledge Graph Reasoning for Multi-Hop Question Answering

A context-augmented training framework for multi-hop question answering, validated on disease-specific knowledge graphs for Gastroparesis and Diabetes extracted with GraphMERT, consistently improved multi-hop performance over KG-only supervision when applied to the Qwen3-14B model, according to an arXiv paper (2609.12230v1). The framework attaches supporting triples from the same source text chunk to each primary KG triple to form a context graph, and an LLM-judged, history-aware adaptive repair pipeline brought the models to 100% accuracy on the cleaned retained one-hop validation sets. Reinforcement learning initialized from repaired supervised fine-tuning checkpoints produced larger and more stable gains on 3-hop, 4-hop, and 5-hop tasks across both diseases.

by read1 min views1 publishedSep 14, 2026

arXiv:2609.12230v1 Announce Type: new Abstract: Question-answering often requires reasoning across multiple connected facts rather than retrieving a single isolated relation. Knowledge graphs (KGs) provide a structured way to represent such facts, but training large language models (LLMs) only on isolated KG head-relation-tail triples may limit their ability to learn the surrounding context needed for multi-hop reasoning. In this work, we propose a context-augmented training framework for multi-hop question-answering. Although generally applicable, we validate the framework in the context of disease-specific KGs, extracted using a reliable KG extraction framework called GraphMERT, for Gastroparesis and Diabetes. For each primary KG triple, we attach supporting triples extracted from the same source text chunk to form a context graph (CG). This creates two supervision settings: KG-grounded supervision, which uses only the target KG triple or path, and CG-grounded supervision, which uses the target KG triple or path together with supporting context triples. We train the Qwen3-14B model using supervised fine-tuning (SFT) under both settings, producing KGModel and CGModel variants. To strengthen the lower-hop factual foundation of the models, we introduce an LLM-judged, history-aware adaptive repair pipeline that identifies unresolved one-hop failures, continually fine-tunes on targeted repair examples, and removes or quarantines problematic noisy triples. This repair stage enables the models to reach 100% accuracy on the cleaned retained one-hop validation sets. Finally, we employ reinforcement learning (RL) using lower-hop question-answer items and evaluate generalization on harder 3-hop, 4-hop, and 5-hop tasks. Across both diseases, context-augmented supervision consistently improves multi-hop performance over KG-only supervision. RL initialized from repaired SFT checkpoints yields larger and more stable gains.

── more in #large-language-models 4 stories · sorted by recency
── more on @graphmert 3 stories trending now
sponsored brought to you by zahid.host 4,200+ EU-deployed projects
reading about agents? ship yours in a single git push.

Run your AI side-project on zahid.host

EU-based hosting, git-push deploys, automatic HTTPS, no cold starts. Free tier with a custom domain — perfect for shipping the agent you just read about.

$git push zahid main
Live at https://your-agent.zahid.host
Get free account → Pricing
from €0/mo · no card required
LIVE [news/repair-before-reinfo…] indexed:0 read:1min 2026-09-14 ·