Ontology Refinement with Embeddings and LLMs: A Case Study on CASTON

Hui Yang, Ziyu Li, Xuan Liu, Anders Jarfors, He Tan +1 more
10/15/2026
cs.AI

Official code: HuiYang1997/OntRefine (1 star, last commit 2026-07-02).

Abstract

Ontologies are widely used to represent and organize complex knowledge in domains such as biomedicine and healthcare. However, constructing reliable ontologies remains challenging, as class hierarchies, whether manually or automatically constructed, may contain subtle subsumption errors that require careful refinement. We study this problem in CASTON, a casting-domain ontology, and present a lightweight but scalable refinement workflow that combines a fine-tuned ontology embedding model with Large Language Models (LLMs). The embedding model is used to extract a limited number of candidate axioms for refinement together with possible replacement candidates, and an LLM then returns repair suggestions. In a small expert evaluation, the default configuration using a CASTON-fine-tuned OnT embedding model and Qwen3-8B produced reliable high-confidence repairs: all six high-confidence replacements were accepted by both domain experts. The implementation is available at https://github.com/HuiYang1997/OntRefine.

View on arXivView PDF

Code Implementations(1)

10Shell, PythonJun 27, 20263 months ago

Cite this paper

@article{yang2026ontology,
  title  = {Ontology Refinement with Embeddings and LLMs: A Case Study on CASTON},
  author = {Hui Yang and Ziyu Li and Xuan Liu and Anders Jarfors and He Tan and Jiaoyan Chen},
  year   = {2026},
  eprint = {2610.21466},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2610.21466v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2610.21466.svg)](https://codeofpaper.com/paper/2610.21466)

Discussion