Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Last Layer Logits to Logic: Empowering LLMs with Logic-Consistent Structured Knowledge Reasoning

About

Large Language Models (LLMs) achieve excellent performance in natural language reasoning tasks through pre-training on vast unstructured text, enabling them to understand the logic in natural language and generate logic-consistent responses. However, the representational differences between unstructured and structured knowledge make LLMs inherently struggle to maintain logic consistency, leading to \textit{Logic Drift} challenges in structured knowledge reasoning tasks such as Knowledge Graph Question Answering (KGQA). Existing methods address this limitation by designing complex workflows embedded in prompts to guide LLM reasoning. Nevertheless, these approaches only provide input-level guidance and fail to fundamentally address the \textit{Logic Drift} in LLM outputs. Additionally, their inflexible reasoning workflows cannot adapt to different tasks and knowledge graphs. To enhance LLMs' logic consistency in structured knowledge reasoning, we specifically target the logits output from the autoregressive generation process. We propose the \textit{Logits-to-Logic} framework, which incorporates logits strengthening and logits filtering as core modules to correct logical defects in LLM outputs. Extensive experiments show that our approach significantly improves LLMs' logic consistency in structured knowledge reasoning and achieves state-of-the-art performance on multiple KGQA benchmarks.

Songze Li, Zhiqiang Liu, Zhaoyan Gong, Xiaoke Guo, Zhongpu Bo, Zhengke Gui, Lei Liang, Huajun Chen, Wen Zhang• 2025

Related benchmarks

TaskDatasetResultRank
Multi-hop Knowledge Graph Question AnsweringWebQSP
Hits@195.4
69
Multi-hop Knowledge Graph Question AnsweringGrailQA
Hits@182
68
Multi-hop Knowledge Graph Question AnsweringCWQ
Hits@180.8
64
Knowledge Base Question AnsweringWebQSP
Hits@196
53
Knowledge Base Question AnsweringCWQ
Hits@183.1
30
Single-Hop Knowledge Graph Question AnsweringSimple Questions
Hits@180.3
27
Multi-hop Knowledge Graph Question AnsweringQALD en 10
Hits@159.2
17
Knowledge Graph Question AnsweringCWQ
# Calls1
12
Slot FillingT-REx
Hit@187.5
8
Slot FillingRE Zero-Shot
Hit@191.3
8
Showing 10 of 11 rows

Other info

Follow for update