HGNet: Scalable Foundation Model for Automated Knowledge Graph Generation from Scientific Literature
Organizations: BASIRA Lab, Imperial-X (I-X) and Department of Computing Imperial College London, London, United Kingdom
Abstract
Automated knowledge graph (KG) construction is essential for navigating the rapidly expanding body of scientific literature. However, existing approaches struggle to recognize long multi-word entities, often fail to generalize across domains, and typically overlook the hierarchical nature of scientific knowledge. While general-purpose large language models (LLMs) offer adaptability, they are computationally expensive and yield inconsistent accuracy on specialized tasks. As a result, current KGs are shallow and inconsistent, limiting their utility for exploration and synthesis. We propose a two-stage framework for scalable, zero-shot scientific KG construction. The first stage, Z-NERD, introduces (i) Orthogonal Semantic Decomposition (OSD), which promotes domain-agnostic entity recognition by isolating semantic "turns" in text, and (ii) a Multi-Scale TCQK attention mechanism that captures coherent multi-word entities through n-gram-aware attention heads. The second stage, HGNet, performs relation extraction with hierarchy-aware message passing, explicitly modeling parent, child, and peer relations. To enforce global consistency, we introduce two complementary objectives: a Differentiable Hierarchy Loss to discourage cycles and shortcut edges, and a Continuum Abstraction Field (CAF) Loss that embeds abstraction levels along a learnable axis in Euclidean space. This is the first approach to formalize hierarchical abstraction as a continuous property within standard Euclidean embeddings, offering a simpler alternative to hyperbolic methods. We release SPHERE (https://github.com/basiralab/SPHERE), a multi-domain benchmark for hierarchical relation extraction. Our framework establishes a new state of the art on SciERC, SciER, and SPHERE, improving NER by 8.08% and RE by 5.99% on out-of-distribution tests. In zero-shot settings, gains reach 10.76% for NER and 26.2% for RE.
Figures & tables
| Models | SciERC | SciER | BioRED | SemEval | CS | Physics | Bio | MS | ||||
| Sup | ZS | Sup | ZS | Sup | ZS | Sup | ZS | |||||
| Supervised Baselines | ||||||||||||
| SciBERT Ye et al. (2022) | 67.52 | 70.71 | 89.15 | 49.14 | 68.19 | 57.02 | 72.90 | 61.22 | 75.83 | 68.45 | 67.29 | 57.14 |
| PL-Marker Yan et al. (2023) | 70.32 | 74.04 | 86.41 | 47.69 | 68.64 | 56.39 | 72.83 | 60.51 | 75.78 | 66.17 | 66.72 | 57.92 |
| HGERE Yan et al. (2023) | 75.92 | 81.19 | 89.43 | 48.25 | 69.82 | 58.95 | 72.46 | 60.67 | 76.42 | 68.51 | 67.24 | 58.03 |
| UniversalNER-7b Zhou et al. (2024) | 66.09 | 73.13 | 88.46 | 47.60 | OOM | |||||||
| Models | SciERC | SciER | BioRED | SemEval 2017 | ||||||
| Hier. | Peer | Overall | Hier. | Peer | Overall | Overall | Hier. | Peer | Overall | |
| Supervised Models | ||||||||||
| PL-Marker Ye et al. (2022) | 35.60 | 44.97 | 41.63 | 40.25 | 61.84 | 56.78 | 29.87 | 32.96 | 43.40 | 37.19 |
| HGERE Yan et al. (2023) | 37.72 | 47.35 | 43.86 | 43.79 | 64.35 | 58.47 | 32.39 | 33.81 | 45.73 | 38.63 |
| PURE Zhong and Chen (2021) | 34.39 | 38.46 | 36.78 | 38.53 | 56.21 | 49.35 | 29.41 | 28.94 | 41.35 | 34.92 |
| Zero-Shot LLM Models | ||||||||||
| Models | Comp. Sci. | Physics | Biology | Mat. Sci. | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Hier. | Peer | All | Hier. | Peer | All | Hier. | Peer | All | Hier. | Peer | All | |
| Supervised Models | ||||||||||||
| PL-Marker Ye et al. (2022) | 51.98 | 57.04 | 55.29 | 50.22 | 56.48 | 53.51 | 52.35 | 53.76 | 53.03 | 52.96 | 53.27 | 53.12 |
| HGERE Yan et al. (2023) | 54.20 | 59.86 | 57.93 | 53.17 | 58.90 | 56.28 | 54.52 | 56.47 | 55.21 | 55.84 | 55.86 | 55.43 |
| Proposed Approaches | ||||||||||||
| HGNet (ours) | 77.40 | 81.36 | 79.51 | 76.93 | 83.47 | 80.60 | 82.53 | 84.29 | 83.74 | 81.91 | 85.64 | 83.65 |
| Models | Comp. Sci. | Mat. Sci. | ||||
|---|---|---|---|---|---|---|
| Hier. | Peer | All | Hier. | Peer | All | |
| PL-Marker Ye et al. (2022) | 28.72 | 28.41 | 28.56 | 33.10 | 34.22 | 33.85 |
| HGERE Yan et al. (2023) | 29.93 | 29.63 | 29.81 | 36.27 | 39.41 | 37.97 |
| HGNet (ours) | 59.36 | 64.07 | 62.60 | 69.92 | 71.33 | 70.62 |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | Z-NERD | HGNet |
|---|---|---|
| Encoder Base Model | SciBERT-base | SciBERT-base |
| Learning Rate | ||
| Batch Size | 16 | 8 |
| Optimizer | AdamW | AdamW |
| Dropout Rate | 0.1 | 0.2 |
| Max Sequence Length | 512 | 512 |
| Dataset | Domain | Docs | Relations | Graph Scope | Hierarchy Source |
|---|---|---|---|---|---|
| SciERC | CS (AI) | 500 | 4.6k | Local (Doc-Level) | Inferred from Text |
| BioRED | Biomed | 600 | 38k | Local (Doc-Level) | Inferred from Text |
| SciER | CS | 106 | 12k | Local (Doc-Level) | Inferred from Text |
| SPHERE | 4 Domains | 10,000 | 111,000 | Global (Corpus-Level) | Pre-defined Scaffold |
| Metric | Training Source | SciERC (Test) | SciER (Test) |
|---|---|---|---|
| HGERE Yan et al. (2023) | Full Supervised Training | 43.86% | 56.28% |
| HGERE (Zero-shot transfer) | SPHERE-CS Training Only | 25.62% | 28.34% |
| HGNet (Zero-shot transfer) | SPHERE-CS Training Only | 46.55% | 59.17% |
| Model | Geometry | SciERC (Overall) | SPHERE-CS (Overall) |
|---|---|---|---|
| HGCN Chami et al. (2019) | Hyperbolic ( ) | 45.82 | 66.35 |
| Order-Embeddings Vendrov et al. (2015) | Cone ( ) | 44.27 | 67.97 |
| HGNet (Ours) | Euclidean + CAF | 53.19 | 79.51 |
| Model | Prompting Strategy | SciERC | SciER |
|---|---|---|---|
| Llama-3-8B | Zero-Shot | 13.72 | 14.95 |
| 3-Shot CoT | 19.45 | 25.18 | |
| HGNet | Supervised | 53.19 | 62.36 |
| Model | Params | GFLOPs | Speed (doc/s) | Mem (GB) | Zero-Shot Gen. |
|---|---|---|---|---|---|
| Large Language Models | |||||
| Llama-3-70B | 70B | 140k | 0.5 | OOM | High |
| Llama-3-8B | 8B | 16k | 4.2 | 16.0+ | Moderate |
| Specialized SOTA | |||||
| PL-Marker | 220M | 44.0 | 12.4 | 7.2 | Low |
| HGERE | 220M | 22.5 | 14.1 | 9.5 | Low |
| Stage | Component | Params (M) | % of Total |
|---|---|---|---|
| Stage 1: Z-NERD | Specialized SciBERT Encoder | 109.5 | 37.4% |
| Multi-Scale TCQK (8 Heads) | 42.4 | 14.5% | |
| Stage 2: HGNet | Specialized SciBERT Encoder | 109.5 | 37.4% |
| Hierarchical GNN Layers | 31.6 | 10.7% | |
| Total | Full Two-Stage Pipeline | 293.0 | 100% |
KG Construction
Knowledge graph construction turns text, tables, images, and databases into structured entities and relationships that machines can query and reason over.
See all 186 papersAgents-K1: Towards Agent-native Knowledge Orchestration
cites edges, omitting key entities, claims, evidence, mechanisms, and method lineages essential for scientific reasoning. To this end, we introduce Agents-K1, an end-to-end knowledge orchestration pipeline that converts raw documents into agent-native scientific knowledge graphs. Agents-K1 integrates three components under a unifying theoretical foundation: a multimodal parser whose five-module schema captures entities, multimodal evidence, citations, and typed inter-entity relations across the full paper rather than abstracts alone; a 4B information-extraction backbone trained with GRPO under a rule-based reward; and a graphanything CLI, a tri-source agent interface that unifies web search, multimodal graph retrieval, and cross-document traversal. On top of this, we process 2.46 million scientific papers across six subjects to produce Scholar-KG, of which we release a one-million-paper subset, and the full Scholar-KG is accessible via the SCP link below. The same pipeline can be extended to general-domain corpora and to schema-conformant data synthesis. Extensive experiments demonstrate that Agents-K1 achieves superior performance in scientific information extraction, knowledge graph construction, and multi-hop scientific reasoning.