Source-linked AI summary

Decoupled Marginal Sharpening for Training-Free Inference-Time Scaling

Weixin Wang, Wei Deng, Anderson Schneider, Yuriy Nevmyvaka, Pan Xu, Andrew Bennett

arXiv:2610.20483v1cs.LG

TL;DR

Full-sequence sharpening can suppress answers supported by alternative reasoning paths, while finite-particle SMC can lose path diversity through repeated resampling. DMS separates reasoning-path sharpening from answer-marginal sharpening, with FAA-SMC supporting its reasoning population. DMS achieves the highest mean pass@1 across all twelve model–benchmark settings.

  • Problem

    Full-sequence sharpening can suppress answers supported by many reasoning paths, and finite-particle SMC resampling can cause path collapse.

    Source: §1 · p. 2, §3 · p. 4
  • Method

    DMS sharpens reasoning paths, aggregates conditional answer probabilities over them, then sharpens the answer marginal with a separate exponent; FAA-SMC approximates the reasoning population.

    Source: §1 · pp. 2–3
  • Results

    DMS achieves the highest mean pass@1 in all twelve model–benchmark settings, with gains over the strongest baseline ranging from 0.34–10.07 percentage points.

    Source: §1 · p. 3, §5.2 · p. 8
  • Takeaways & Limitations

    The comparisons support DMS’s complete procedure, combining a separately controlled reasoning distribution with marginal-probability weighting of complete answer candidates.

    Source: §5.2 · p. 9

Abstract

from arXiv · show

We study how to improve LLM reasoning at inference time without updating model weights or using external reward models. Recent reward-free power-sampling methods define the sequence-level power target $\pi_\alpha(y\mid x)\propto p_0(y\mid x)^\alpha$, where $\alpha>1$, which increases the relative probability of completions that are already likely under the base model. Existing methods use Metropolis--Hastings or Sequential Monte Carlo (SMC) to approximately sample from this target. However, two challenges arise at different levels. At the target level, full-sequence sharpening can concentrate probability on a few exact reasoning--answer trajectories, suppressing alternative paths that support the same answer. At the sampler level, finite-particle SMC can suffer from path collapse when resampling repeatedly duplicates a few high-weight prefixes. To address these challenges, we propose Decoupled Marginal Sharpening (DMS), which separately controls how strongly the model favors particular reasoning paths and how strongly it favors final answers after aggregating support across those paths. At the target level, DMS first samples from a sharpened reasoning distribution, then marginalizes the model's answer probabilities over the resulting weighted reasoning population and sharpens the induced answer marginal. Separate exponents allow mild reasoning sharpening to encourage exploration of diverse paths, followed by stronger answer-marginal sharpening to concentrate probability on answers with greater aggregate support across those paths. At the sampler level, we develop Fully Adapted Annealed SMC, which combines an annealed sequence of power targets with locally fully adapted proposals and predictive ancestor resampling. Experiments on MATH500, GPQA-Diamond, HumanEval, and AIME24/25 across three base models show that DMS achieves the highest mean pass@1 among the compared methods in all tasks.

1 Introduction

The introduction identifies target-level loss of support for answers reached by many reasoning paths and sampler-level path collapse, then presents DMS and FAA-SMC as responses. DMS achieves the highest mean pass@1 in all twelve model–benchmark settings.

1 Introduction

  • Reward-free power sampling sharpens the base model’s sequence probabilities, but exact tokenwise sampling is generally intractable, motivating approximate samplers such as SMC.The introduction also mentions Metropolis–Hastings and future-continuation estimates as alternatives.Source: §1 · p. 1
  • Full-sequence sharpening can favor individual high-likelihood reasoning–answer pairs over an answer supported by many alternative paths.The introduction also notes that higher sequence likelihood does not necessarily imply correctness.Source: §1 · p. 2
  • DMS sharpens reasoning paths, aggregates conditional answer probabilities over that population, and then sharpens the answer marginal with separate exponents.The reasoning exponent controls each path’s contribution; the answer exponent controls concentration on answers with greater aggregate support.Source: §1 · p. 2
  • FAA-SMC addresses finite-particle path collapse with annealed power targets, predictive ancestor resampling, and locally fully adapted proposals.The resulting weighted reasoning population supports an empirical answer marginal and cross-reasoning evaluation of complete answer candidates.Source: §1 · pp. 2–3
  • DMS achieves the highest mean pass@1 in all twelve model–benchmark settings, using a two-stage target and a practical sampling procedure.The procedure combines FAA-SMC for reasoning with cross-reasoning likelihood evaluation for answer candidates.Source: §1 · p. 3

2 Related Work

Related work spans reward-guided generation and reward-free sampling from model-intrinsic signals. DMS connects sequence-level sharpening with answer-level aggregation across reasoning paths.

2 Related Work

  • Reward-based inference methods guide generation with external rewards or verifiers, including twisted-SMC approaches and reward-guided SMC decoders.Their effectiveness depends on the availability and accuracy of the reward signals.Source: §2 · p. 3
  • Reward-free sequence-level power sampling has been approximated with Metropolis–Hastings, future-correction estimates, and SMC variants.Related work also studies particle allocation, ensemble distributions, and model-intrinsic scores.Source: §2 · p. 3
  • Self-consistency and marginal sharpening aggregate support across reasoning paths at the answer level; DMS independently sharpens the reasoning distribution used to form that marginal.Self-consistency selects the answer with greatest aggregate support, while marginal sharpening constructs an answer marginal under the original reasoning distribution.Source: §2 · p. 3

3 Preliminaries

The preliminaries define autoregressive sequence probabilities and sequence-level power sharpening, then describe SMC approximation and its finite-particle path-collapse risk.

3 Preliminaries

  • An autoregressive model assigns a sequence probability from successive token probabilities conditioned on the prompt and preceding tokens.Generation uses a maximum length T; early endings are padded with a fixed terminal symbol without changing sequence probability.Source: §3 · pp. 3–4
  • Sequence-level power sharpening favors higher-likelihood outputs when α > 1, but sampling token by token requires summing over future continuations, which is generally intractable.SMC instead uses tractable proposals and importance weights to correct toward the target.Source: §3 · p. 4
  • SMC represents an evolving target with weighted particles and uses proposals plus incremental importance weights to account for target changes.The terminal normalized target is the sequence-level power distribution.Source: §3 · p. 4
  • When effective sample size falls below a threshold, SMC resamples prefixes by normalized weight, potentially duplicating high-weight paths and discarding alternatives.With a finite particle budget, repeated resampling can cause path collapse; resetting weights does not recover discarded paths.Source: §3 · p. 4

4 Methodology: Decoupled Marginal Sharpening

DMS separates reasoning-path sharpening from answer-level sharpening, aggregating answer support across reasoning paths before concentrating probability on answers. FAA-SMC approximates the reasoning target, and weighted answer candidates are reweighted for final selection.

4.1 Two-Stage Sharpened Target

  • DMS separately sharpens reasoning paths and the answer marginal formed by aggregating conditional answer probabilities across those paths.Separate exponents allow the reasoning distribution to remain less concentrated while answer selection gives greater weight to answers with stronger aggregate support.Source: §4 · p. 4, §4.1 · p. 5

4.2 Reasoning Sampling with FAA-SMC

  • FAA-SMC anneals the reasoning-path power target and uses predictive ancestor resampling with locally fully adapted proposals to construct a weighted reasoning population.Annealing introduces sharpening gradually while preserving the final target; the proposal lets predictive weights be computed before token generation.Source: §4.2 · p. 6
  • Island-DMS runs independent particle populations, ensuring the pooled population retains at least one initial-particle ancestor from each island.The method divides K particles among L islands, each of which resamples only within its own population.Source: §4.2 · p. 7

4.3 Sampling from the Sharpened Answer Marginal

  • DMS samples answer candidates from the empirical marginal, scores each complete candidate under all retained reasoning paths, and reweights candidates for final selection.Candidate generation uses one selected path, but marginal scoring includes contributions from the entire reasoning population; importance weighting corrects the candidate proposal toward the sharpened answer target.Source: §4.3 · pp. 7–8

5 Experiments

Across the evaluated models and benchmarks, DMS achieves the highest mean pass@1 in every setting. The experiments compare DMS with decoding and reward-free baselines, and ablations examine its sharpening stages and island sampling.

5.1 Experimental Setup

  • The evaluation covers three base models, four benchmarks, and comparisons with standard decoding, low-temperature decoding, Power-SMC, and Marginal Sharpening.DMS uses 16 reasoning particles and 32 answer candidates per problem; AIME24/25 uses Island-DMS with eight two-particle islands.Source: §5.1 · p. 8

5.2 Main Results

  • DMS achieves the highest mean pass@1 in all twelve model–benchmark settings, improving over the strongest baseline by 0.34–10.07 percentage points.On MATH500, it exceeds Power-SMC by 8.00 and 10.07 points with Qwen3-4B and Qwen3-8B; on HumanEval, it exceeds Marginal Sharpening by 5.49 and 8.94 points.Source: §5.2 · pp. 8–9, Table 1 · p. 9

5.3 Ablation Study on AIME24/25

  • With αZ = 1, increasing αA from 1 to 4 raises expected accuracy from 66.48% to 76.89% on the AIME24/25 Qwen3-8B ablation.At fixed exponents and particle count, Island-DMS raises expected accuracy from 74.55% to 76.83% and oracle pass@32 from 78.33% to 81.67%; both it and answer sharpening alone obtain 77.22% pass@1.Source: §5.3 · p. 9, Table 2 · p. 9
  • Reasoning sharpening raises average candidate correctness from 66.48% to 71.77% but reduces oracle pass@32 from 82.78% to 78.33% when αA = 1.In this setting, answer sharpening alone achieves similar expected accuracy to Island-DMS, and additional reasoning sharpening is not required for the best observed pass@1.Source: §5.3 · p. 9, Table 2 · p. 9

6 Conclusion

DMS separates reasoning-path sharpening from answer-level sharpening, using FAA-SMC to sample reasoning paths and aggregate answer support. Across twelve benchmark settings, it achieves the highest mean pass@1 among the compared methods.

6 Conclusion

  • DMS achieves the highest mean pass@1 among compared methods in all twelve settings across MATH500, GPQA-Diamond, HumanEval, and AIME24/25, using three base models.The comparisons include Power-SMC and Marginal Sharpening.Source: §6 · p. 10
  • DMS sharpens the reasoning distribution, marginalizes answer probabilities over the resulting reasoning population, and then sharpens the induced answer marginal.Separate exponents control reasoning-path concentration and answer-level consensus.Source: §6 · p. 10
  • FAA-SMC combines annealed power targets, predictive resampling, and locally fully adapted proposals to address sampler-level path collapse.The conclusion also identifies improving answer-marginal estimation, reducing particle degeneracy, and adapting sharpening strengths and inference-time budgets as future work.Source: §6 · p. 10

A Additional Details for DMS

The appendix describes FAA-SMC as the sampler for DMS reasoning paths and outlines the answer-target representation and empirical answer-weight calculation. Annealing preserves the specified final power target while adapting intermediate targets.

A Additional Details for DMS

  • DMS applies FAA-SMC to the reasoning segment with terminal exponent α = αZ.The sampler uses an exponent schedule from β0 = 1 to βT = α and returns weighted sequences.Source: A Additional Details for DMS · p. 10
  • Annealing changes intermediate targets while preserving the specified final power target; terminated particles receive fixed terminal symbols and still receive corrections until the final exponent.Along a fixed sequence with positive proposal probability, the incremental importance ratios telescope.Source: A Additional Details for DMS · p. 10
  • The augmented answer target pairs reasoning paths with answers and applies an additional power factor shared across paths paired with the same answer.Summing over reasoning paths yields the answer target.Source: A Additional Details for DMS · p. 11
  • DMS estimates the answer marginal from a finite weighted reasoning population and computes candidate answer weights from each candidate’s conditional log probability under reasoning paths.Only answer tokens contribute to the score; reasoning supplies the conditioning context, and subtracting a maximum improves numerical stability without changing normalized weights.Source: A Additional Details for DMS · p. 11

B Experimental Setup and Implementation Details

The experiments evaluate three base models on four benchmarks with shared generation and evaluation rules across methods. DMS uses a finite reasoning-particle population and answer candidates, with an island implementation on AIME24/25.

B Experimental Setup and Implementation Details

  • Experiments evaluate Qwen3-4B, Qwen3-8B, and DeepSeek-R1-Distill-Llama-8B on MATH500, GPQA-Diamond, HumanEval, and AIME24/25, averaging results over three seeds.The study uses the models’ native chat templates and bfloat16 inference on one NVIDIA H100 80 GB GPU per task.Source: B Experimental Setup and… · p. 11
  • Within each model–benchmark setting, methods share the checkpoint, tokenizer, prompt, maximum trajectory length, answer extractor, and grader.Marginal Sharpening and DMS allocate separate reasoning and answer limits, unlike methods that generate a complete reasoning–answer trajectory within the full limit.Source: B Experimental Setup and… · p. 11
  • DMS defaults to K = 16 reasoning particles and M = 32 answer candidates, with predictive-weight ESS checks every 64 tokens and systematic resampling below K/2.Each complete answer candidate is cross-scored under retained reasoning contexts; the power-two annealing schedule begins after 1024 reasoning tokens.Source: B Experimental Setup and… · p. 12
  • Table 3 lists the reasoning and answer exponent pairs used in the main results in Table 1.The supplied passage identifies the table’s scope but does not provide its exponent values.Source: Table 3 · p. 12
  • On AIME24/25, all three models use eight islands of two particles, with resampling within islands and equal total weight assigned to each island.The pooled particle weights define the empirical answer marginal for candidate generation and selection.Source: B Experimental Setup and… · p. 12

C Sensitivity to the Two Sharpening Exponents

In standard DMS with Qwen3-8B on AIME24/25, answer sharpening improves expected accuracy across tested reasoning exponents. The strongest reported result uses unsharpened reasoning and a larger answer exponent, supporting independent choices of the two strengths.

C Sensitivity to the Two Sharpening Exponents

  • Setting αZ = 1.005 raises average candidate correctness from 66.48% to 71.77%, but its reasoning bank’s highest expected accuracy is 74.58%, below 77.13% with αZ = 1.Improving average candidate correctness does not necessarily improve accuracy after answer reweighting.Source: C Sensitivity to the Two Sharpening… · pp. 12–13
  • At αZ = 1, expected accuracy rises from 66.48% at αA = 1 to 76.89% at αA = 4 and 77.13% at αA = 8.Most of the improvement is obtained by αA = 4, with limited additional gains from stronger answer sharpening.Source: C Sensitivity to the Two Sharpening… · p. 12, Figure 3 · p. 13
  • The strongest result combines an unsharpened reasoning distribution with a larger answer exponent, supporting independent choices of reasoning and answer sharpening strengths.The experiment uses standard DMS with Qwen3-8B on AIME24/25 and averages expected accuracy over three seeds.Source: C Sensitivity to the Two Sharpening… · p. 13, Figure 3 · p. 13

D Annealing and Resampling Ablations

The ablation compares annealing and ancestor resampling in standard DMS, with all variants targeting the same terminal reasoning distribution. The reference sampler leads in expected accuracy, while disabling resampling preserves more ancestors and raises oracle coverage but lowers expected accuracy.

D Annealing and Resampling Ablations

  • All variants approximate the same terminal reasoning target, with completed traces continuing to receive annealing weight corrections.The reference uses a quadratic annealing schedule and resamples at 64-token checkpoints when predictive ESS falls below K/2.Source: D Annealing and Resampling Ablations · p. 13
  • The reference sampler exceeds the no-annealing and no-resampling variants in mean expected accuracy by 2.83 and 2.46 percentage points, respectively.The comparison uses non-island DMS with Qwen3-8B on AIME24/25, averaged over three seeds.Source: Table 4 · p. 13, D Annealing and Resampling Ablations · p. 13
  • Disabling resampling preserves all 16 initial ancestors and raises oracle pass@32 from 78.33% to 80.00%.The final reasoning weights become more concentrated, with normalized ESS falling to 0.252, while expected accuracy decreases to 72.09%.Source: D Annealing and Resampling Ablations · pp. 13–14
  • Final ESS and surviving ancestors measure different properties: resampling can balance weights without recovering discarded ancestors.Similar final ESS values can coexist with different numbers of surviving ancestors in the reference and no-annealing variants.Source: D Annealing and Resampling Ablations · p. 14

E Higher-Budget Reasoning-Sampler Diagnostic

A higher-budget replica-exchange sampler provides a diagnostic of reasoning-exponent sensitivity on HumanEval. Both exponent ladders show peak expected accuracy at moderate reasoning sharpening, alongside a slight decline in oracle coverage as sharpening increases.

E Higher-Budget Reasoning-Sampler Diagnostic

  • The diagnostic uses finite-chain replica-exchange samples rather than exact samples from the reasoning targets.Adjacent replicas exchange traces, while local Metropolis–Hastings updates revise previously generated reasoning traces.Source: E Higher-Budget Reasoning-Sampler… · p. 14
  • Expected accuracy peaks at αZ = 1.05 in the fine ladder at 96.59%, and at αZ = 1.5 in the extended ladder at 96.61%.The fine ladder rises from 95.06% at αZ = 1 to 96.59%, then reaches 96.39% at αZ = 1.1.Source: E Higher-Budget Reasoning-Sampler… · p. 14, Table 5 · p. 14
  • Oracle pass@32 decreases slightly from 98.17% to 97.97% as αZ increases within each ladder.The passage describes this pattern as consistent with sharpening reducing diversity and the chance of finding a correct answer through a less likely path.Source: E Higher-Budget Reasoning-Sampler… · p. 14

F Time-Cost Analysis

The time-cost analysis measures reasoning generation, answer generation, and cross-scoring per problem. DMS has higher measured latency than standard decoding but lower latency than Marginal Sharpening across all evaluated settings.

F Time-Cost Analysis

  • Across all twelve model–benchmark settings, DMS has lower measured latency than Marginal Sharpening.Non-island DMS requires 1.23–3.82 times standard-decoding latency; Island-DMS on AIME24/25 requires 2.48–4.30 times.Source: F Time-Cost Analysis · p. 15, Table 6 · p. 15
  • For Island-DMS with Qwen3-8B on AIME24/25, reasoning generation takes 705.38 s (80.63%) per problem, versus 25.24 s (2.89%) for answer generation and 144.23 s (16.49%) for cross-scoring.These timings reflect the evaluated implementations and configurations.Source: F Time-Cost Analysis · p. 15
Loading 2610.20483v1…