English

SLiNT: Structure-aware Language Model with Injection and Contrastive Training for Knowledge Graph Completion

Computation and Language 2025-09-09 v1 Artificial Intelligence

Abstract

Link prediction in knowledge graphs requires integrating structural information and semantic context to infer missing entities. While large language models offer strong generative reasoning capabilities, their limited exploitation of structural signals often results in structural sparsity and semantic ambiguity, especially under incomplete or zero-shot settings. To address these challenges, we propose SLiNT (Structure-aware Language model with Injection and coNtrastive Training), a modular framework that injects knowledge-graph-derived structural context into a frozen LLM backbone with lightweight LoRA-based adaptation for robust link prediction. Specifically, Structure-Guided Neighborhood Enhancement (SGNE) retrieves pseudo-neighbors to enrich sparse entities and mitigate missing context; Dynamic Hard Contrastive Learning (DHCL) introduces fine-grained supervision by interpolating hard positives and negatives to resolve entity-level ambiguity; and Gradient-Decoupled Dual Injection (GDDI) performs token-level structure-aware intervention while preserving the core LLM parameters. Experiments on WN18RR and FB15k-237 show that SLiNT achieves superior or competitive performance compared with both embedding-based and generation-based baselines, demonstrating the effectiveness of structure-aware representation learning for scalable knowledge graph completion.

Keywords

Cite

@article{arxiv.2509.06531,
  title  = {SLiNT: Structure-aware Language Model with Injection and Contrastive Training for Knowledge Graph Completion},
  author = {Mengxue Yang and Chun Yang and Jiaqi Zhu and Jiafan Li and Jingqi Zhang and Yuyang Li and Ying Li},
  journal= {arXiv preprint arXiv:2509.06531},
  year   = {2025}
}

Comments

Accepted by EMNLP Findings 2025

R2 v1 2026-07-01T05:26:06.139Z