Chain of Code: Reasoning with a Language Model-Augmented Code Emulator
Abstract
Chain of Code (CoC) enhances language models' reasoning capabilities by using pseudocode and selective code emulation, enabling them to tackle complex semantic and logical tasks more effectively.
Code provides a general syntactic structure to build complex programs and perform precise computations when paired with a code interpreter - we hypothesize that language models (LMs) can leverage code-writing to improve Chain of Thought reasoning not only for logic and arithmetic tasks, but also for semantic ones (and in particular, those that are a mix of both). For example, consider prompting an LM to write code that counts the number of times it detects sarcasm in an essay: the LM may struggle to write an implementation for "detect_sarcasm(string)" that can be executed by the interpreter (handling the edge cases would be insurmountable). However, LMs may still produce a valid solution if they not only write code, but also selectively "emulate" the interpreter by generating the expected output of "detect_sarcasm(string)". In this work, we propose Chain of Code (CoC), a simple yet surprisingly effective extension that improves LM code-driven reasoning. The key idea is to encourage LMs to format semantic sub-tasks in a program as flexible pseudocode that the interpreter can explicitly catch undefined behaviors and hand off to simulate with an LM (as an "LMulator"). Experiments demonstrate that Chain of Code outperforms Chain of Thought and other baselines across a variety of benchmarks; on BIG-Bench Hard, Chain of Code achieves 84%, a gain of 12% over Chain of Thought. In a nutshell, CoC broadens the scope of reasoning questions that LMs can answer by "thinking in code".
Community
I love this.
It seems like a natural next step to create a language-agnostic psuedo-code compiler.
Language in. Binary out.
And vice versa.
Binary as an extra modality
Hi Authors,
First off - excellent work, congratulations.
Your video in https://chain-of-code.github.io/ has this prompt "Q: How many countries have I been to ? I have been to Mumbai, London, Washington, Grand Canyon ...".
You say "With direct prompting, the model predicts a number leading to mistakes".
I just tried the above question with direct prompting in Claude-3 and got the correct answer = 3.
Could you provide a more complex example, perhaps ?
"Based on the locations you mentioned:
Mumbai is a city in India
London is a city in the United Kingdom
Washington likely refers to Washington D.C. in the United States
Grand Canyon is a national park in the United States
So from the information provided, you have been to 3 different countries:
- India
- United Kingdom
- United States
Therefore, the total number of countries you have been to based on those locations is 3 countries."
Unlocking Advanced Reasoning: Chain of Code Explained
Links π:
π Subscribe: https://www.youtube.com/@Arxflix
π Twitter: https://x.com/arxflix
π LMNT (Partner): https://lmnt.com/
We have done an independent implemetation of this approach in our open-source optimizing inference proxy optillm - https://github.com/codelion/optillm/blob/main/optillm/plugins/coc_plugin.py
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper