Skip to main navigation Skip to search Skip to main content

TIV: Thought Injection via Vectors for Efficient Reasoning in Large Reasoning Model

  • Yi Cao
  • , Weijie Shi
  • , Wei Jie Xu
  • , Yucheng Shen
  • , Yue Cui
  • , Hanghui Guo
  • , Shimin Di
  • , Ziyi Liu
  • , Jiaming Li
  • , Alexander Zhou
  • , Jia Zhu
  • , Jiajie Xu

Research output: Journal article publication › Conference article › Academic research › peer-review

Abstract

Large Reasoning Models (LRMs) have recently demonstrated impressive performance across a range of reasoning tasks by generating intermediate thoughts. However, these models can suffer from overthinking—generating excessive tokens that contribute little to final accuracy while increasing inference cost. To mitigate this, we propose TIV (Thought Injection via Vectors), an innovative framework that compresses token-level reasoning into compact vectors without sacrificing performance. Rather than generating explicit thoughts, TIV injects learnable vectors into the post-attention hidden states of the final token across Transformer layers, enabling implicit and lightweight reasoning. We further introduce a two-stage reinforcement learning strategy: the first stage calibrates the model’s reasoning distribution, and the second distills it into a vector-based policy optimized for both accuracy and brevity. Experiments on three reasoning benchmarks show that TIV preserves over 99% of the original accuracy while reducing output length by more than 65% on average, reaching up to 80% in some cases. Moreover, TIV consistently achieves superior trade-offs between accuracy and efficiency compared to existing methods, distinguishing itself as a state-of-the-art (SOTA) approach for efficient reasoning in LRMs.

Original languageEnglish
Pages (from-to)30148-30155
Number of pages8
JournalProceedings of the AAAI Conference on Artificial Intelligence
Volume40
Issue number36
DOIs
Publication statusPublished - Mar 2026
Event40th AAAI Conference on Artificial Intelligence, AAAI 2026 - Singapore, Singapore
Duration: 20 Jan 2026 → 27 Jan 2026

ASJC Scopus subject areas

  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'TIV: Thought Injection via Vectors for Efficient Reasoning in Large Reasoning Model'. Together they form a unique fingerprint.

Cite this