COSMOS: Connectivity-Oriented Submodular Maximization for Optimal Subgraph Retrieval
Boci Peng, Xiao Liu, Boren Hu, Yun Zhu, Xuanbo Fan, Yanwei Yue, Chunyu Yang, Yan Zhang
Abstract
Retrieving coherent evidence subgraphs is critical for Knowledge Base Question Answering (KBQA). Existing paradigms often treat facts independently, rely on biased heuristics, or employ myopic search, failing to optimize collective subgraph utility. In this paper, we propose COS-MOS (Connectivity-Oriented Submodular Maximization for Optimal Subgraph Retrieval), a unified framework that formalizes evidence retrieval as a constrained submodular maximization problem. This formulation mathematically captures the trade-off between information relevance and structural complexity. To tractably solve this combinatorial challenge, COSMOS employs a decompose-andconquer strategy, which first performs a seedguided greedy expansion to maximize local semantic utility, followed by a topology-aware component aggregation to bridge disjoint evidence clusters via Maximum Spanning Tree aggregation. Guided by theoretical bounds, we introduce Structure-Aware Contrastive Tuning to align semantic space with KG topology. Experimental results on WebQSP, CWQ, and M 3 GQA benchmarks demonstrate that COS-MOS achieves state-of-the-art performance. collaboratively created graph database for structuring human knowledge. In
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 10f719e7-8f3c-4bbe-bcc5-b69033eb72feBuilds on7
- Sequence-to-Sequence Knowledge Graph Completion and Question AnsweringApoorv Saxena, Adrian Kochsiek, Rainer GemullaACL 2022 · 183 citations
- StructGPT: A General Framework for Large Language Model to Reason over Structured DataJinhao Jiang, Kun Zhou, Zican Dong, Keming Ye et al.EMNLP 2023 · 173 citations
- UniKGQA: Unified Retrieval and Reasoning for Solving Multi-hop Question Answering Over Knowledge GraphJinhao Jiang, Kun Zhou, Xin Zhao, Ji-Rong WenICLR 2023 · 28 citations
- Direct Fact Retrieval from Knowledge Graphs without Entity LinkingJinheon Baek, Alham Fikri Aji, Jens Lehmann, Sung Ju HwangACL 2023 · 9 citations
- HyKGE: A Hypothesis Knowledge Graph Enhanced RAG Framework for Accurate and Reliable Medical LLMs ResponsesXinke Jiang, Ruizhe Zhang, Yongxin Xu, Rihong Qiu et al.ACL 2025
Related papers
- DSM: Question Generation over Knowledge Base via Modeling Diverse Subgraphs with Meta-learnerShasha Guo, Jing Zhang, Yanling Wang, Qianyi Zhang et al.EMNLP 2022 · 9 citations
- STEM: Structure-Tracing Evidence Mining for Knowledge Graphs-Driven Retrieval-Augmented GenerationPeng Yu, En Xu, Bin Chen, Haibiao Chen et al.ACL 2026
- Temporal Evidence Chain for Temporal Knowledge Graph Question Answering with Large Language ModelsShihao Liu, Xiaofei Zhou, Bo Wang, Geyuan ZhangACL 2026
- Enhancing Complex Question Answering over Knowledge Graphs through Evidence Pattern RetrievalWentao Ding, Jinmao Li, Liangchuan Luo, Yuzhong QuWWW 2024 · 32 citations
- Knowledge Base Question Answering by Case-based Reasoning over SubgraphsRajarshi Das, Ameya Godbole, Ankita Naik, Elliot Tower et al.ICML 2022 · 66 citations
