Implicit Chain of Thought Reasoning via Knowledge Distillation
Authors: Yuntian Deng, Kiran Prasad, Roland Fernandez, Paul Smolensky, Vishrav Chaudhary, Stuart Shieber
AI-generated Key Points
- Authors propose an alternative approach to reasoning in language models (LMs)
- LMs can reason effectively with intermediate computation not expressed in natural language
- Internal hidden states of the language model used for implicit reasoning instead of explicit chain-of-thought reasoning steps
- Implicit reasoning steps distilled from a teacher model trained on explicit chain-of-thought reasoning
- Vertical reasoning among hidden states in different layers instead of horizontal reasoning by generating intermediate words
- Experiments conducted on multi-digit multiplication task and grade school math problem dataset
- Results show that this alternative reasoning method allows solving previously unsolvable tasks without explicit chain-of-thought reasoning
- Comparable speed achieved to no chain-of-thought reasoning
- Enhances language models' ability to reason by leveraging implicit reasoning through hidden states
- Opens up new possibilities for solving complex tasks effectively and efficiently.
Abstract: To augment language models with the ability to reason, researchers usually prompt or finetune them to produce chain of thought reasoning steps before producing the final answer. However, although people use natural language to reason effectively, it may be that LMs could reason more effectively with some intermediate computation that is not in natural language. In this work, we explore an alternative reasoning approach: instead of explicitly producing the chain of thought reasoning steps, we use the language model's internal hidden states to perform implicit reasoning. The implicit reasoning steps are distilled from a teacher model trained on explicit chain-of-thought reasoning, and instead of doing reasoning "horizontally" by producing intermediate words one-by-one, we distill it such that the reasoning happens "vertically" among the hidden states in different layers. We conduct experiments on a multi-digit multiplication task and a grade school math problem dataset and find that this approach enables solving tasks previously not solvable without explicit chain-of-thought, at a speed comparable to no chain-of-thought.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.