LoRi: Low-Rank Distillation for Implicit Reasoning

32d ago · Global · primary source: export.arxiv.org

A new distillation framework called LoRi aims to close the gap between implicit chain-of-thought reasoning and explicit prompting in large language models by exploiting low-rank structure in hidden-state trajectories, according to a paper submitted June 3, 2026 [1]. Implicit chain-of-thought (iCoT) methods seek to internalize reasoning steps within a model’s hidden states rather than generating them as text, but they have consistently lagged behind explicit chain-of-thought prompting in accuracy [1][2]. The authors of the LoRi paper report that the hidden-state trajectories produced during reasoning exhibit a low-rank structure, meaning their essential information can be captured in a compressed subspace [1][2]. LoRi transfers reasoning capability from a teacher model to a student by aligning their trajectories inside a shared low-rank tensor subspace, using both first- and second-order statistics to preserve the global structure of the reasoning process [1][2]. The framework was tested across multiple model families, including LLaMA and Qwen, at different scales on mathematical reasoning benchmarks [1][2]. Qwen is a family of large language models developed by Alibaba Cloud and distributed under open-source licenses including Apache 2.0 [8]. The paper states that LoRi consistently improves performance, with gains most pronounced on challenging multi-step tasks, approaching the accuracy of explicit chain-of-thought and surpassing prior iCoT distillation methods [1][2]. The work arrives amid broader efforts to improve multi-turn and underspecified interactions with language models. A separate recent study introduced a training framework called Found in Conversation, which uses view-asymmetric self-distillation to recover single-turn competence in multi-turn settings across LLaMA, Qwen, Phi, and OLMo backbones, recovering at least 92% of single-turn performance [5]. Distillation techniques have also been applied to other model behaviors. Researchers studying synthetic dishonesty fine-tuned honest and deceptive variants of five transformer models, including Qwen2.5-7B and Llama-3.1-8B, using Low-Rank Adaptation (LoRA), and found that linear probes could detect dishonesty with near-perfect AUC as early as layers 1–3 in most architectures [4]. The LoRi authors did not release a public benchmark comparison with such detection or editing methods, but their low-rank alignment approach adds to a growing toolkit of parameter-efficient techniques for shaping model internals [1][2][4]. The paper was submitted to arXiv on June 3, 2026, and is available as a preprint [1][2].

research-paper

Background sources we checked (9)
  • arxiv.org ↗ Implicit chain-of-thought (iCoT) methods aim to internalize reasoning in large language models, but often underperform explicit CoT prompting. We empirically find that hidden-state reasoning trajectories exhibit low-rank structure. Motivated by this observation, we propose a low-…
  • arxiv.org ↗ Knowledge editing aims to update or correct factual knowledge in a language model. A widely used approach, locate-then-edit, does this in two steps: it first localizes a fact within the model, then edits the weights there. To date, such methods have been developed exclusively on …
  • arxiv.org ↗ Deceptive alignment, in which models maintain accurate internal representations while deliberately producing false outputs, remains a central challenge in AI safety. While strategic deception is the primary long-term concern, synthetic dishonesty - induced via direct optimization…
  • arxiv.org ↗ Large Language Model (LLM) interactions are typically underspecified, with users clarifying all necessary details across multiple conversational turns. Yet recent work shows that LLMs perform far worse in this multi-turn setting than in a single turn with same information being a…
  • arxiv.org ↗ As jailbreaks, adversarially crafted inputs that bypass safety constraints, continue to be discovered in Large Language Models, practitioners increasingly rely on fine-tuning as a defensive strategy. Yet the theoretical foundations underlying this robustness fine-tuning remain un…
  • arxiv.org ↗ Emergent misalignment (EM), where fine-tuning on a narrow task (like insecure code) causes broad misalignment across unrelated domains, was first demonstrated by Betley et al. (2025). We conduct the most comprehensive EM study to date, reproducing the original GPT-4o finding and …
  • en.wikipedia.org ↗ Qwen (also known as Tongyi Qianwen, Chinese: 通义千问; pinyin: Tōngyì Qiānwèn) is a family of large language models developed by Alibaba Cloud. Many Qwen models are distributed under the free and open-source Apache 2.0 license, the source-available Qwen License, or the non-commercial…
  • en.wikipedia.org ↗ llama.cpp is an open-source software library that performs inference on various large language models such as Llama. It is co-developed alongside the GGML project, a general-purpose tensor library. Command-line tools are included with the library, alongside a server with a simple…
  • en.wikipedia.org ↗ Hangzhou DeepSeek Artificial Intelligence Basic Technology Research Co., Ltd., doing business as DeepSeek, is a Chinese artificial intelligence (AI) company that develops large language models (LLMs). Based in Hangzhou, Zhejiang, DeepSeek is owned and funded by High-Flyer, a Chin…

Sources

Spot something wrong? Report an issue