Yunjie He
2026
SCAIR: Schema-Conditioned Agentic Iterative Reasoning for Enterprise Knowledge Graphs
Prateek Chaturvedi | Yuqicheng Zhu | Hongkuan Zhou | Dongzhuoran Zhou | Yunjie He | Steffen Staab | Fei Du | Jie Tang | Evgeny Kharlamov
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026)
Prateek Chaturvedi | Yuqicheng Zhu | Hongkuan Zhou | Dongzhuoran Zhou | Yunjie He | Steffen Staab | Fei Du | Jie Tang | Evgeny Kharlamov
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026)
Knowledge Graph–based Retrieval-Augmented Generation (KG-RAG) enables natural language interaction with structured enterprise knowledge, yet existing agentic approaches that perform well on public benchmarks often fail to generalize to real-world enterprise Knowledge Graphs (KGs), which are dense, schema-driven, and operationally constrained. To address these limitations, we propose SCAIR (Schema-Conditioned Agentic Iterative Reasoning), a training-free framework that integrates structured planning with controlled iterative reasoning by injecting schema-conditioned structural priors and enforcing schema-aware traversal during multi-hop reasoning. Experiments on an enterprise-oriented benchmark constructed from a real-world Configuration Management DataBase (CMDB) demonstrate that SCAIR substantially improves performance over existing KG-RAG methods. Crucially, our study highlights that reliable enterprise graph reasoning cannot rely on generic agentic designs; instead, it must explicitly incorporate the target domain’s structural and operational constraints into the reasoning process. We demonstrate that by aligning agent design with business logic, substantial performance gains can be achieved without the need for costly model retraining.
2025
Conformalized Answer Set Prediction for Knowledge Graph Embedding
Yuqicheng Zhu | Nico Potyka | Jiarong Pan | Bo Xiong | Yunjie He | Evgeny Kharlamov | Steffen Staab
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)
Yuqicheng Zhu | Nico Potyka | Jiarong Pan | Bo Xiong | Yunjie He | Evgeny Kharlamov | Steffen Staab
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)
Knowledge graph embeddings (KGE) apply machine learning methods on knowledge graphs (KGs) to provide non-classical reasoning capabilities based on similarities and analogies. The learned KG embeddings are typically used to answer queries by ranking all potential answers, but rankings often lack a meaningful probabilistic interpretation - lower-ranked answers do not necessarily have a lower probability of being true. This limitation makes it difficult to quantify uncertainty of model’s predictions, posing challenges for the application of KGE methods in high-stakes domains like medicine. We address this issue by applying the theory of conformal prediction that allows generating answer sets, which contain the correct answer with probabilistic guarantees. We explain how conformal prediction can be used to generate such answer sets for link prediction tasks. Our empirical evaluation on four benchmark datasets using six representative KGE methods validates that the generated answer sets satisfy the probabilistic guarantees given by the theory of conformal prediction. We also demonstrate that the generated answer sets often have a sensible size and that the size adapts well with respect to the difficulty of the query.
2024
Predictive Multiplicity of Knowledge Graph Embeddings in Link Prediction
Yuqicheng Zhu | Nico Potyka | Mojtaba Nayyeri | Bo Xiong | Yunjie He | Evgeny Kharlamov | Steffen Staab
Findings of the Association for Computational Linguistics: EMNLP 2024
Yuqicheng Zhu | Nico Potyka | Mojtaba Nayyeri | Bo Xiong | Yunjie He | Evgeny Kharlamov | Steffen Staab
Findings of the Association for Computational Linguistics: EMNLP 2024
Knowledge graph embedding (KGE) models are often used to predict missing links for knowledge graphs (KGs). However, multiple KG embeddings can perform almost equally well for link prediction yet give conflicting predictions for unseen queries. This phenomenon is termed predictive multiplicity in the literature. It poses substantial risks for KGE-based applications in high-stake domains but has been overlooked in KGE research. We define predictive multiplicity in link prediction, introduce evaluation metrics and measure predictive multiplicity for representative KGE methods on commonly used benchmark datasets. Our empirical study reveals significant predictive multiplicity in link prediction, with 8% to 39% testing queries exhibiting conflicting predictions. We address this issue by leveraging voting methods from social choice theory, significantly mitigating conflicts by 66% to 78% in our experiments.