Geometric Constraints for Small Language Models to Understand and Expand Scientific Taxonomies
Liri Fang, Dongqi Fu, Jiawei Han, Jingrui He, Vetle I Torvik
Abstract
Recent findings reveal that token embeddings of Large Language Models (LLMs) exhibit strong hyperbolicity. This insight motivates leveraging LLMs for scientific taxonomy tasks, where maintaining and expanding hierarchical knowledge structures is critical. Although potential, generally-trained LLMs face challenges in directly handling domain-specific taxonomies, including computational cost and hallucination. Meanwhile, Small Language Models (SLMs) provide a more economical alternative if empowered with proper knowledge transfer. In this work, we introduce SS-Mono (Structure-Semantic Monotonization), a novel pipeline that combines local taxonomy augmentation from LLMs, self-supervised fine-tuning of SLMs with geometric constraints, and LLM calibration. Our approach enables efficient and accurate taxonomy expansion across root, leaf, and intermediate nodes. Extensive experiments on both leaf and non-leaf expansion benchmarks demonstrate that a fine-tuned SLM (e.g., DistilBERT-base-110M) consistently outperforms frozen LLMs (e.g., GPT-4o, Gemma-2-9B) and domain-specific baselines. These findings highlight the promise of lightweight yet effective models for structured knowledge enrichment in scientific domains.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 06c97e24-c794-4827-8ba8-d2e991786ecdBuilds on20
- LoRA: Low-Rank Adaptation of Large Language ModelsEdward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu et al.ICLR 2022 · 18,833 citations
- Hyperbolic Neural Networks++Ryohei Shimizu, Yusuke Mukuta, Tatsuya HaradaICLR 2021 · 791 citations
- Harnessing Explanations: LLM-to-LM Interpreter for Enhanced Text-Attributed Graph Representation LearningXiaoxin He, Xavier Bresson, Thomas Laurent, Adam Perold et al.ICLR 2024 · 151 citations
- From Trees to Continuous Embeddings and Back: Hyperbolic Hierarchical ClusteringInes Chami, Albert Gu, Vaggos Chatziafratis, Christopher RéNeurIPS 2020 · 125 citations
- TaxoExpan: Self-supervised Taxonomy Expansion with Position-Enhanced Graph Neural NetworkJiaming Shen, Zhihong Shen, Chenyan Xiong, Chi Wang et al.WWW 2020 · 85 citations
Related papers
- BLEND: Balanced and Leaf-Enhanced Dual Fine-Tuning for Taxonomy CompletionPankaj, Dhruv Kumar, Vinayak Abrol, Vikram GoyalWWW 2026
- TaxoLLaMA: WordNet-based Model for Solving Multiple Lexical Semantic TasksViktor Moskvoretskii, Ekaterina Neminova, Alina Lobanova, Alexander Panchenko et al.ACL 2024 · 11 citations
- Compress and Mix: Advancing Efficient Taxonomy Completion with Large Language ModelsHongyuan Xu, Yuhang Niu, Yanlong Wen, Xiaojie YuanWWW 2025 · 6 citations
- OntoTune: Ontology-Driven Self-training for Aligning Large Language ModelsZhiqiang Liu, Chengtao Gan, Junjie Wang, Yichi Zhang et al.WWW 2025 · 11 citations
- TokAlign: Efficient Vocabulary Adaptation via Token AlignmentChong Li, Jiajun Zhang, Chengqing ZongACL 2025 · 7 citations
