Ontology Refinement with Embeddings and LLMs: A Case Study on CASTON
Authors: Hui Yang, Ziyu Li, Xuan Liu, Anders Jarfors, He Tan, Jiaoyan Chen
Organizations: University of Manchester, Manchester, UK · Department of Materials and Manufacturing, Jönköping University, Sweden · COMPtech i Skillingaryd AB, Sweden · Department of Mechanical Engineering, National University of Singapore, Singapore · Department of Computing, Jönköping University, Sweden
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.
Figures & tables
Figure 1: Overview of the ontology-refinement workflow.
Figure 2: Prompt used for the LLM validation stage.
Setting
Confidence
Rows
Accepted (+2/+1)
Mixed
Rejected (-2/-1)
Acceptance rate
Default: Qwen3-8B + CASTON OnT-FT
High
6
6 (6/0)
0
0
100%
Default: Qwen3-8B + CASTON OnT-FT
Medium
2
0 (0/0)
2
0
0%
Ablation: Qwen3-8B + pretrained ensemble
High
7
4 (4/0)
0
3
57.1%
Ablation: Qwen3-8B + pretrained ensemble
Medium
4
1 (0/1)
0
3
25.0%
Auxiliary: GPT-4o mini + pretrained ensemble
High
16
5 (5/0)
1
10
31.3%
Auxiliary: GPT-4o mini + pretrained ensemble
Medium
5
1 (0/1)
0
4
20.0%
Table 1: Expert outcomes by setting and LLM confidence. Accepted rows have positive expert agreement; parentheses give the +2/+1 breakdown. Mixed rows are expert disagreements or unclear cases.
Ontologies are useful structures to organize and maintain information that can be understood both by humans and systems. However, since their manual crafting is a laborious task, many specific domains lack reference ontologies. The outstanding ability for understanding natural language demonstrated by the Large Language Models (LLMs) has motivated their application to aid on a variety of fields, including on ontology development. This work presents the experimentation with a technique that uses LLMs in the role of domain experts to build conceptual hierarchies for a given initial concept. Twenty ontologies automatically constructed for the domain of the Brazilian maritime territory (a.k.a the Blue Amazon) using GPT-3.5 and GPT-4 were then evaluated by human experts. The models were able to construct overall coherent conceptualizations of the domain, but none of the outputs was completely satisfactory as a representation of the context without refinement.
Vivian Magri Alcaldi Soares, Renata Wassermann
University of SÃo Paulo (USP) · Center for Artificial Intelligence (C4AI)
Ontology learning (OL) aims to automatically construct structured knowledge models from text, yet progress remains fragmented across methods, domains, and evaluation practices. Despite decades of research, OL lacks a shared infrastructure for systematic evaluation and ontology access. This absence has hindered progress and fragmented research, leaving the central challenges of OL largely unaddressed. We introduce OntoLearner, a modular, cross-domain, and first-of-its-kind framework that unifies ontology access, large language model (LLM)-driven learning pipelines, and standardized benchmarking. OntoLearner releases 180 machine-readable ontologies spanning 22 domains and provides pipeline-ready datasets with train/dev/test splits for three core OL tasks: term typing, taxonomy discovery, and non-taxonomic relation extraction. Using this infrastructure, we conduct a large-scale empirical study of OL, evaluating 22 retrieval models and 12 LLMs across domains and tasks. The results converge on a finding that reframes the central challenge of OL: failure modes scale with ontological complexity rather than model size or architectural sophistication. The primary bottleneck is not model capability, but a structural mismatch between how models encode knowledge and how ontologies organize it. These findings establish that effective OL is reachable through the cross-domain, multi-task benchmarking enabled by OntoLearner. OntoLearner is open-source (MIT license) at https://github.com/sciknoworg/OntoLearner/.
Hamed Babaei Giglou, Jennifer D'Souza, Andrei Aioanei +2
TIB – Leibniz Information Centre for Science and Technology, Hannover, Germany · 3IBM Research, New York, USA · 1TIB – Leibniz Information Centre for Science and Technology, Hannover, Germany +1
Knowledge Organization Systems like Ontologies and taxonomies are fundamental for structuring scientific knowledge, yet their manual curation presents a persistent bottleneck in knowledge management. While Large Language Models (LLMs) offer a scalable mechanism for automated ontology generation, their capacity to classify complex, domain-specific semantics requires systematic evaluation. In this paper, we assess the performance of five small, open-source LLMs (up to 9 billion parameters) in identifying semantic relationships between biomedical concepts. To support this evaluation, we introduce MeSH-Rel-4K, a dataset comprising 4K semantic relationships extracted from the Medical Subject Headings (MeSH). We analyse three adaptation strategies: standard prompting, Chain-of-Thought prompting, and fine-tuning. While parameter-constrained models traditionally struggle with the nuances of in-context logic, our results reveal that targeted fine-tuning increases the average F1-score by 34.1 percentage points. These results confirm that direct fine-tuning effectively exceeds the reasoning bottlenecks of smaller LLMs, providing an accurate, automated methodology for the construction and evolution of specialised biomedical ontologies.
Tanay Aggarwal, Angelo Salatino, Francesco Osborne +1
Knowledge Media Institute, The Open University, Milton Keynes, UK · Department of Business and Law, University of Milano-Bicocca, Milan, IT