facebookresearch/deepconf
DeepConf: Deep Think with Confidence
What it solves
DeepConf addresses the challenge of improving the reasoning accuracy of Large Language Models (LLMs) in complex tasks like math, science, and coding. It aims to increase the reliability of answers by using parallel thinking (generating multiple reasoning paths) and efficiently managing the computational budget through confidence-based stopping.
How it works
DeepConf acts as a wrapper around LLM serving backends like vLLM. It implements a parallel thinking framework that can operate in two modes:
- Online Mode: Uses a set of warmup traces to establish a confidence threshold. It then applies early stopping to halt generation once a sufficient level of confidence is reached, saving compute.
- Offline Mode: Generates a batch of reasoning traces (a budget) all at once and applies various voting and aggregation strategies to determine the most likely correct answer.
Who it’s for
Developers and researchers working with reasoning-heavy LLMs who want to improve answer accuracy through ensemble methods (voting) and optimize inference costs using confidence-based early stopping.
Highlights
- vLLM Integration: Fully compatible with vLLM initialization parameters and serving backends.
- Confidence-Based Early Stopping: Reduces unnecessary computation in online mode by stopping when confidence thresholds are met.
- Flexible Voting Strategies: Supports multiple aggregation methods to select the final answer from parallel reasoning traces.
- Comprehensive Output: Provides detailed metadata, including confidence bars, token usage, and timing statistics for every run.
Related
- Project
- Project
- Project
- Project
- Dispatch