qwen3.5-9b-ontology-extraction-raft

A LoRA adapter for Qwen/Qwen3.5-9B that extracts a primitive ontology — [subject, relation, object] triples — from a raw text document.

Built by Semantic Swingers for the LLMs4OL 2026 shared task (Task A, "flagship": text → ontology triples, scored by graph similarity). If you have not heard of the challenge, the short version: given a document, produce the taxonomy/typing triples it implies. This adapter is the fine-tuned generator behind our submission.

Which adapter is this

• Regime: RA-FT (retrieval-aware).

• Use top_k = 10 at inference. RA-FT was trained with retrieved exemplars in the prompt, so it expects exemplars at inference. Using the wrong k understates the adapter.

• Hardware: CUDA GPU. The qwen3_5 hybrid needs CUDA fast-attention kernels; on Apple Silicon they fall back to CPU.

• graph_similarity (organizer v2, exact) on val_20: 0.7000

How to run it

Through our OntoLearner integration (the learner ships in the fork below):

from ontolearner.learner.text2onto import SemanticSwingersText2OntoLearner

learner = SemanticSwingersText2OntoLearner(
    adapter="datagero/qwen3.5-9b-ontology-extraction-raft",
    base_model_id="Qwen/Qwen3.5-9B",
    backend="peft",
    top_k=10,
)
learner.load()
# learner.fit(train_docs, task="text2onto"); learner.predict(eval_docs, task="text2onto")

How it was made (the training code is part of the integration)

This adapter was produced by the same package that serves it — training is a first-class part of the OntoLearner integration, not a separate script:

• Training code: semanticswingers_train.py — LoRA SFT with prompt masking (loss on completion tokens only) and, for RA-FT, leave-one-out exemplar retrieval so a training document never sees its own gold.
• Integration + reproduction: OntoLearner fork, PR #1 and the replication notebook therein (notebooks/pipeline_ontolearner.ipynb), which runs Tasks A/B/C end-to-end.

To reproduce: learner = SemanticSwingersText2OntoLearner(train_mode="raft", train_backend="peft", output_dir=...) then learner.fit(train_docs, task="text2onto").

Intended use & limitations

Research replication for the LLMs4OL 2026 shared task. Domain: general ontology construction from text; the training data is the challenge's Task A split. Not intended for production ontology engineering without validation. Inherits the base model's license and limitations.