LightThinker: Thinking Step-by-Step Compression
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Large language models (LLMs) have shown remarkable performance in complex reasoning tasks, but their efficiency is hindered by the substantial memory and computational costs associated with generating lengthy tokens. In this paper, we propose LightThinker, a novel method that enables LLMs to dynamically compress intermediate thoughts during reasoning. Inspired by human cognitive processes, LightThinker compresses verbose thought steps into compact representations and discards the original reasoning chains, thereby significantly reducing the number of tokens stored in the context window. This is achieved by training the model on when and how to perform compression through data construction, mapping hidden states to condensed gist tokens, and creating specialized attention masks. Additionally, we introduce the Dependency (Dep) metric to quantify the degree of compression by measuring the reliance on historical tokens during generation. Extensive experiments on four datasets and two models show that LightThinker reduces peak memory usage and inference time, while maintaining competitive accuracy. Our work provides a new direction for improving the efficiency of LLMs in complex reasoning tasks without sacrificing performance. Code is released at https://github.com/zjunlp/LightThinker.
Key facts. Cited by 104 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 32 of its 53 references that are in the map. First posted on arXiv 21 Feb 2025.
Builds on
32 of 53 references.
Closest ancestors · 5
Same direction, newest first
- 2025O1-Pruner: Length-Harmonizing Fine-Tuning for O1-Like Reasoning Pruning22 Jan
- 2024Compressed Chain of Thought: Efficient Reasoning Through Dense Representations17 Dec
- 2024C3oT: Generating Shorter Chain-of-Thought without Compromising Effectiveness16 Dec
- 2024A Comparative Study on Reasoning Patterns of OpenAI's o1 Model17 Oct
- 2024Concise Thoughts: Impact of Output Length on LLM Reasoning and Cost29 Jul
Other references · 27
- 2017Attention Is All You Needcited by 60,361
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,807
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
- 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 880
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
- 2024KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cachecited by 637
15 more
- 2024KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationcited by 591
- 2024PyramidKV: Dynamic KV Cache Compression based on Pyramidal Information Funnelingcited by 432
- 2023Learning to Compress Prompts with Gist Tokenscited by 390
- 2023Adapting Language Models to Compress Contextscited by 356
- 2022Reasoning with Language Model Prompting: A Surveycited by 252
- 2025Training Language Models to Reason Efficientlycited by 247
- 2024From Explicit CoT to Implicit CoT: Learning to Internalize CoT Step by Stepcited by 235
- 2024Long Context Compression with Activation Beaconcited by 104
- 2020Thinking Fast and Slow in AIcited by 70
- 2024Can Language Models Learn to Skip Steps?cited by 59
- 2024SepLLM: Accelerate Large Language Models by Compressing One Segment into One Separatorcited by 41
- 2024Break the Chain: Large Language Models Can be Shortcut Reasonerscited by 32
- 2024SCOPE: Optimizing Key-Value Cache Compression in Long-context Generationcited by 20
- 2023ChatGPT is a Remarkable Tool -- For Expertscited by 16
- 2024OneGen: Efficient One-Pass Unified Generation and Retrieval for LLMscited by 14
Built on by
Citing papers per year: 2025: 62; 2026: 42.
104 papers.
Sort:
By area:
- 2025Stop Overthinking: A Survey on Efficient Reasoning for Large Language Modelscited by 464
- 2025Towards Reasoning Era: A Survey of Long Chain-of-Thought for Reasoning Large Language Modelscited by 384
- 2025MEM1: Learning to Synergize Memory and Reasoning for Efficient Long-Horizon Agentscited by 238
- 2025A Survey on Test-Time Scaling in Large Language Models: What, How, Where, and How Well?cited by 223
- 2025DAST: Difficulty-Adaptive Slow-Thinking for Large Reasoning Modelscited by 158
- 2025ACON: Optimizing Context Compression for Long-horizon LLM Agentscited by 136
- 2025A Survey of Efficient Reasoning for Large Reasoning Models: Language, Multimodality, and Beyondcited by 132
- 2025Efficient Reasoning Models: A Surveycited by 95
- 2025AdaCoT: Pareto-Optimal Adaptive Chain-of-Thought Triggering via Reinforcement Learningcited by 87
- 2025A Survey on Latent Reasoningcited by 81
- 2025Efficient Inference for Large Reasoning Models: A Surveycited by 61
- 2025A Survey of LLM-Driven AI Agent Communication: Protocols, Security Risks, and Defense Countermeasurescited by 57
38 more
- 2025InftyThink: Breaking the Length Limits of Long-Context Reasoning in Large Language Modelscited by 54
- 2025Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodalitycited by 50
- 2025Speculative Thinking: Enhancing Small-Model Reasoning with Large Model Guidance at Inference Timecited by 48
- 2025SSR: Enhancing Depth Perception in Vision-Language Models via Rationale-Guided Spatial Reasoningcited by 46
- 2025Implicit Reasoning in Large Language Models: A Comprehensive Surveycited by 43
- 2025Towards Agentic RAG with Deep Reasoning: A Survey of RAG-Reasoning Systems in LLMscited by 42
- 2025Harnessing the Reasoning Economy: A Survey of Efficient Reasoning for Large Language Modelscited by 36
- 2025Retro-Search: Exploring Untaken Paths for Deeper and Efficient Reasoningcited by 34
- 2025ConCISE: Confidence-guided Compression in Step-by-step Efficient Reasoningcited by 33
- 2025Reasoning on a Budget: A Survey of Adaptive and Controllable Test-Time Compute in LLMscited by 31
- 2025R1-Compress: Long Chain-of-Thought Compression via Chunk Compression and Searchcited by 30
- 2025FlashThink: An Early Exit Method For Efficient Reasoningcited by 28
- 2025Promoting Efficient Reasoning with Verifiable Stepwise Rewardcited by 28
- 2025Let LRMs Break Free from Overthinking via Self-Braking Tuningcited by 27
- 2025ThinKV: Thought-Adaptive KV Cache Compression for Efficient Reasoning Modelscited by 21
- 2025SofT-GRPO: Surpassing Discrete-Token LLM Reinforcement Learning via Gumbel-Reparameterized Soft-Thinking Policy Optimizationcited by 21
- 2026Stabilizing Recurrent Dynamics for Test-Time Scalable Latent Reasoning in Looped Language Modelscited by 21
- 2026Render-of-Thought: Rendering Textual Chain-of-Thought as Images for Visual Latent Reasoningcited by 20
- 2025A Survey of Slow Thinking-based Reasoning LLMs using Reinforced Learning and Inference-time Scaling Lawcited by 19
- 2025Can Pruning Improve Reasoning? Revisiting Long-CoT Compression with Capability in Mind for Better Reasoningcited by 19
- 2025Towards Concise and Adaptive Thinking in Large Reasoning Models: A Surveycited by 18
- 2026MEMENTO: Teaching LLMs to Manage Their Own Contextcited by 18
- 2025Reasoning Path Compression: Compressing Generation Trajectories for Efficient LLM Reasoningcited by 17
- 2025ThinkLess: A Training-Free Inference-Efficient Method for Reducing Reasoning Redundancycited by 17
- 2025Not All Thoughts are Generated Equal: Efficient LLM Reasoning via Multi-Turn Reinforcement Learningcited by 16
- 2025Two Experts Are All You Need for Steering Thinking: Reinforcing Cognitive Effort in MoE Reasoning Models Without Additional Trainingcited by 16
- 2025LazyEviction: Lagged KV Eviction with Attention Pattern Observation for Efficient Long Reasoningcited by 16
- 2026Thinking in Streaming Videocited by 16
- 2025Generative AI Act II: Test Time Scaling Drives Cognition Engineeringcited by 15
- 2025Concise Reasoning, Big Gains: Pruning Long Reasoning Trace with Difficulty-Aware Promptingcited by 15
- 2025StreamingThinker: Large Language Models Can Think While Readingcited by 15
- 2025CoIn: Counting the Invisible Reasoning Tokens in Commercial Opaque LLM APIscited by 14
- 2025TrimR: Verifier-based Training-Free Thinking Compression for Efficient Test-Time Scalingcited by 14
- 2025Reasoning Model Unlearning: Forgetting Traces, Not Just Answers, While Preserving Reasoning Skillscited by 14
- 2025Exploring the System 1 Thinking Capability of Large Reasoning Modelscited by 10
- 2025Not All Tokens Matter: Towards Efficient LLM Reasoning via Token Significance in Reinforcement Learningcited by 10
- 2025A*-Thought: Efficient Reasoning via Bidirectional Compression for Low-Resource Settingscited by 9
- 2025Demystifying Hybrid Thinking: Can LLMs Truly Switch Between Think and No-Think?cited by 9
50 most cited of 104 All 104 (JSON)
- 2026ActiveMem: Dynamic Latent Memory Trees for Long-Horizon Agents27 Sep 2026
- 2026A*-Thought-V2: Efficient Latent Reasoning via Geometric Dynamics of LLM7 Sep 2026
- 2026BeaconKV: Key-Value Cache Compression Guided by Beacon Queries for Efficient Large Reasoning Model Inference4 Sep 2026
- 2026ChainPrune: Evaluating and Reducing Redundancy in Long Chain-of-Thought Reasoning22 Aug 2026
- 2026Training Continuous Chain of Thought Models: A Tale of Two Regimes18 Jul 2026
- 2026LaME: Learning to Think in Latent Space for Multimodal Embedding via Information Bottleneck11 Jun 2026
- 2026The Periodic Table of LLM Reasoning: A Structured Survey of Reasoning Paradigms, Methods, and Failure Modes9 Jun 2026
- 2026N-GRPO: Embedding-Level Neighbor Mixing for Enhanced Policy Optimization9 Jun 2026
- 2026HybridThinker: Efficient Chain-of-Thought Reasoning via Compressed Memory and Transient Thought Steps2 Jun 2026
- 2026Spectral-Progressive Thought Flow for Lightweight Multimodal Reasoning1 Jun 2026
- 2026Thinking Economically: A Hierarchical Framework for Adaptive-Complexity Reasoning in LLMs31 May 2026
- 2026LISA: Linear-Indexed Sparse Attention for Efficient Long-Context Reasoning29 May 2026
38 more
- 2026ThinkReset: Learnable Intermediate Interface Construction for Bounded-Context Long-Horizon Reasoning26 May 2026
- 2026Stabilizing Recurrent Dynamics for Test-Time Scalable Latent Reasoning in Looped Language Models26 May 2026
- 2026State commitment learning: training language models to distinguish computation from memory22 May 2026
- 2026Unified Data Selection for LLM Reasoning21 May 2026
- 2026RAG-Enhanced Large Language Models for Dynamic Content Expiration Prediction in Web Search13 May 2026
- 2026Path-Lock Expert: Separating Reasoning Mode in Hybrid Thinking via Architecture-Level Separation29 Apr 2026
- 2026MemoSight: Unifying Context Compression and Multi Token Prediction for Reasoning Acceleration16 Apr 2026
- 2026SWE-AGILE: A Software Agent Framework for Efficiently Managing Dynamic Reasoning Context13 Apr 2026
- 2026ZoomR: Memory Efficient Reasoning through Multi-Granularity Key Value Retrieval13 Apr 2026
- 2026MEMENTO: Teaching LLMs to Manage Their Own Context10 Apr 2026
- 2026SeLaR: Selective Latent Reasoning in Large Language Models9 Apr 2026
- 2026FoE: Forest of Errors Makes the First Solution the Best in Large Reasoning Models3 Apr 2026
- 2026Multimodal Model for Computational Pathology:Representation Learning and Image Compression19 Mar 2026
- 2026TRiMS: Real-Time Tracking of Minimal Sufficient Length for Efficient Reasoning via RL18 Mar 2026
- 2026Mitigating Overthinking in Large Reasoning Language Models via Reasoning Path Deviation Monitoring15 Mar 2026
- 2026Thinking in Streaming Video13 Mar 2026
- 2026SPOT: Span-level Pause-of-Thought for Efficient and Interpretable Latent Reasoning in Large Language Models6 Mar 2026
- 2026Learning from Partial Chain-of-Thought via Truncated-Reasoning Self-Distillation27 Feb 2026
- 2026BFS-PO: Best-First Search for Large Reasoning Models16 Feb 2026
- 2026Precedent-Informed Reasoning: Mitigating Overthinking in Large Reasoning Models via Test-Time Precedent Learning16 Feb 2026
- 2026Budget-Constrained Agentic Large Language Models: Intention-Based Planning for Costly Tool Use12 Feb 2026
- 2026A State-Transition Framework for Efficient LLM Reasoning1 Feb 2026
- 2026Capabilities and Fundamental Limits of Latent Chain-of-Thought1 Feb 2026
- 2026ImgCoT: Compressing Long Chain of Thought into Compact Visual Tokens for Efficient Reasoning of Large Language Model30 Jan 2026
- 2026VTC-R1: Vision-Text Compression for Efficient Long-Context Reasoning29 Jan 2026
- 2026Chain Of Thought Compression: A Theoretical Analysis29 Jan 2026
- 2026Render-of-Thought: Rendering Textual Chain-of-Thought as Images for Visual Latent Reasoning21 Jan 2026
- 2026Neural Chain-of-Thought Search: Searching the Optimal Reasoning Path to Enhance Large Language Models16 Jan 2026
- 2026Mid-Think: Training-Free Intermediate-Budget Reasoning via Token-Level Triggers11 Jan 2026
- 2026SyncThink: A Training-Free Strategy to Align Inference Termination with Reasoning Saturation7 Jan 2026
- 2025Reinforcement Learning for Chain of Thought Compression with One-Domain-to-All Generalization19 Dec 2025
- 2025Experience-Evolving Multi-Turn Tool-Use Agent with Hybrid Episodic-Procedural Memory8 Dec 2025
- 2025Think Before You Prune: Selective Self-Generated Calibration for Pruning Large Reasoning Models24 Nov 2025
- 2025Incorporating Self-Rewriting into Large Language Model Reasoning Reinforcement20 Nov 2025
- 2025Stealth Fine-Tuning: Efficiently Breaking Alignment in RVLMs Using Self-Generated CoT18 Nov 2025
- 2025BadThink: Triggered Overthinking Attacks on Chain-of-Thought Reasoning in Large Language Models13 Nov 2025
- 2025EffiReason-Bench: A Unified Benchmark for Evaluating and Advancing Efficient Reasoning in Large Language Models13 Nov 2025
- 2025SofT-GRPO: Surpassing Discrete-Token LLM Reinforcement Learning via Gumbel-Reparameterized Soft-Thinking Policy Optimization9 Nov 2025
50 newest of 104 All 104 (JSON)
- 2025Stop Overthinking: A Survey on Efficient Reasoning for Large Language Modelscited by 464
- 2025Towards Reasoning Era: A Survey of Long Chain-of-Thought for Reasoning Large Language Modelscited by 384
- 2025MEM1: Learning to Synergize Memory and Reasoning for Efficient Long-Horizon Agentscited by 238
- 2025A Survey on Test-Time Scaling in Large Language Models: What, How, Where, and How Well?cited by 223
- 2025DAST: Difficulty-Adaptive Slow-Thinking for Large Reasoning Modelscited by 158
- 2025A Survey of Efficient Reasoning for Large Reasoning Models: Language, Multimodality, and Beyondcited by 132
- 2025Efficient Reasoning Models: A Surveycited by 95
- 2025AdaCoT: Pareto-Optimal Adaptive Chain-of-Thought Triggering via Reinforcement Learningcited by 87
- 2025A Survey on Latent Reasoningcited by 81
- 2025Efficient Inference for Large Reasoning Models: A Surveycited by 61
- 2025InftyThink: Breaking the Length Limits of Long-Context Reasoning in Large Language Modelscited by 54
- 2025Speculative Thinking: Enhancing Small-Model Reasoning with Large Model Guidance at Inference Timecited by 48
12 most cited of 93 All 93 (JSON)
- 2026A*-Thought-V2: Efficient Latent Reasoning via Geometric Dynamics of LLM7 Sep 2026
- 2026BeaconKV: Key-Value Cache Compression Guided by Beacon Queries for Efficient Large Reasoning Model Inference4 Sep 2026
- 2026ChainPrune: Evaluating and Reducing Redundancy in Long Chain-of-Thought Reasoning22 Aug 2026
- 2026Training Continuous Chain of Thought Models: A Tale of Two Regimes18 Jul 2026
- 2026LaME: Learning to Think in Latent Space for Multimodal Embedding via Information Bottleneck11 Jun 2026
- 2026The Periodic Table of LLM Reasoning: A Structured Survey of Reasoning Paradigms, Methods, and Failure Modes9 Jun 2026
- 2026N-GRPO: Embedding-Level Neighbor Mixing for Enhanced Policy Optimization9 Jun 2026
- 2026HybridThinker: Efficient Chain-of-Thought Reasoning via Compressed Memory and Transient Thought Steps2 Jun 2026
- 2026Spectral-Progressive Thought Flow for Lightweight Multimodal Reasoning1 Jun 2026
- 2026Thinking Economically: A Hierarchical Framework for Adaptive-Complexity Reasoning in LLMs31 May 2026
- 2026LISA: Linear-Indexed Sparse Attention for Efficient Long-Context Reasoning29 May 2026
- 2026ThinkReset: Learnable Intermediate Interface Construction for Bounded-Context Long-Horizon Reasoning26 May 2026
12 newest of 93 All 93 (JSON)
- 2025ACON: Optimizing Context Compression for Long-horizon LLM Agentscited by 136
- 2025A Survey of LLM-Driven AI Agent Communication: Protocols, Security Risks, and Defense Countermeasurescited by 57
- 2026Budget-Constrained Agentic Large Language Models: Intention-Based Planning for Costly Tool Usecited by 8
- 2026ActiveMem: Dynamic Latent Memory Trees for Long-Horizon Agentscited by 0
- 2026ActiveMem: Dynamic Latent Memory Trees for Long-Horizon Agents27 Sep 2026
- 2026Budget-Constrained Agentic Large Language Models: Intention-Based Planning for Costly Tool Use12 Feb 2026
- 2025ACON: Optimizing Context Compression for Long-horizon LLM Agents1 Oct 2025
- 2025A Survey of LLM-Driven AI Agent Communication: Protocols, Security Risks, and Defense Countermeasures24 Jun 2025
- 2026Thinking in Streaming Videocited by 16
- 2025MILR: Improving Multimodal Image Generation via Test-Time Latent Reasoningcited by 8
- 2025Experience-Evolving Multi-Turn Tool-Use Agent with Hybrid Episodic-Procedural Memorycited by 6
- 2026Multimodal Model for Computational Pathology:Representation Learning and Image Compressioncited by 6
- 2026SWE-AGILE: A Software Agent Framework for Efficiently Managing Dynamic Reasoning Contextcited by 6
- 2026RAG-Enhanced Large Language Models for Dynamic Content Expiration Prediction in Web Searchcited by 1
- 2026RAG-Enhanced Large Language Models for Dynamic Content Expiration Prediction in Web Search13 May 2026
- 2026SWE-AGILE: A Software Agent Framework for Efficiently Managing Dynamic Reasoning Context13 Apr 2026
- 2026Multimodal Model for Computational Pathology:Representation Learning and Image Compression19 Mar 2026
- 2026Thinking in Streaming Video13 Mar 2026
- 2025Experience-Evolving Multi-Turn Tool-Use Agent with Hybrid Episodic-Procedural Memory8 Dec 2025
- 2025MILR: Improving Multimodal Image Generation via Test-Time Latent Reasoning26 Sep 2025
Similar, not cited
Close in wording, no citation either way
- 2026LightThinker++: From Reasoning Compression to Memory Management
- 2026Shorthand for Thought: Compressing LLM Reasoning via Entropy-Guided Supertokens
- 2025Making Slow Thinking Faster: Compressing LLM Chain-of-Thought via Step Entropy
- 2025How Well do LLMs Compress Their Own Chain-of-Thought? A Token Complexity Approach
- 2026Memory Augmentation Unlocks Efficient Chain-of-Thought Reasoning
- 2025Activation Steering for Chain-of-Thought Compression
- 2025Efficient Reasoning with Hidden Thinking
- 2025Think Silently, Think Fast: Dynamic Latent Compression of LLM Reasoning Chains