Papers › Cautious Next Token Prediction

Cautious Next Token Prediction

3 Jul 2025arXiv:2507.03038archive 2025-07-28

Yizhou Wang, Lingzhi Zhang, Yue Bai, Mang Tik Chiu, Zhengmian Hu, Mingyuan Zhang, Qihua Dong, Yu Yin, Sohrab Amirghodsi, Yun Fu

Next token prediction paradigm has been prevailing for autoregressive models in the era of LLMs. The current default sampling choice for popular LLMs is temperature scaling together with nucleus sampling to balance diversity and coherence. Nevertheless, such approach leads to inferior performance in various NLP tasks when the model is not certain about testing questions. To this end, we propose a brand new training-free decoding strategy, dubbed as Cautious Next Token Prediction (CNTP). In the decoding process, if the model has comparatively high prediction entropy at a certain step, we sample multiple trials starting from the step independently and stop when encountering any punctuation. Then we select the trial with the lowest perplexity score viewed as the most probable and reliable trial path given the model's capacity. The trial number is negatively correlated with the prediction confidence, i.e., the less confident the model is, the more trials it should sample. This is consistent with human beings' behaviour: when feeling uncertain or unconfident, one tends to think more creatively, exploring multiple thinking paths, to cautiously select the path one feels most confident about. Extensive experiments on both LLMs and MLLMs show that our proposed CNTP approach outperforms existing standard decoding strategies consistently by a clear margin. Moreover, the integration of CNTP with self consistency can further improve over vanilla self consistency. We believe our proposed CNTP has the potential to become one of the default choices for LLM decoding. Code is available at https://github.com/wyzjack/CNTP.

PaperPDFCodeCode Syntology ran

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2507.03038")

Code

Syntology Ran 7 of 14 code samples harvested from 1 repository linked to this paper; 7 have no recorded run. Of those that ran: 7 ran with no contract checked.

By repository: official repository: 14 samples from 1 repository, 7 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

wyzjack/CNTP officialmentioned in paperjaxMIT report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

14 samples harvested; 7 ran; 0 honoured the contract we drafted; 7 have no recorded run. Read from Syntology's graph 2026-09-24; that is when this build read the record, not when the samples ran.

7ran
7unverified

Licence: 0 of the 14 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from wyzjack/CNTP. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

get_imports wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/dynamic_module_utils.py official repository ran MIT (permissive) · 1adcf2d05b2faed1 · report
get_relative_import_files wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/dynamic_module_utils.py official repository ran MIT (permissive) · 518f6a89865f25d2 · report
get_relative_imports wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/dynamic_module_utils.py official repository ran MIT (permissive) · 4af5faa701d0f2c9 · report
infer_metric_tags_from_eval_results wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modelcard.py official repository ran MIT (permissive) · dd8f7751ef859581 · report
parse_keras_history wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modelcard.py official repository ran MIT (permissive) · e6632a5722f0840d · report
quick_gelu wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_flax_utils.py official repository ran fingerprinted MIT (permissive) · a66d2b78670c85e1 · report
rename_key_and_reshape_tensor wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_flax_pytorch_utils.py official repository ran MIT (permissive) · 30d50ff2ced6b7a3 · report
convert_pytorch_state_dict_to_flax wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_flax_pytorch_utils.py official repository unverified MIT (permissive) · b1260d802f5e3997 · report
dtype_byte_size wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_flax_utils.py official repository unverified MIT (permissive) · 5e01aab2a01fb6db · report
flax_shard_checkpoint wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_flax_utils.py official repository unverified MIT (permissive) · 80d41fe6d81de3dd · report
is_hf_dataset wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modelcard.py official repository unverified MIT (permissive) · ec7b1528a936234d · report
prepare_fa2_from_position_ids wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_flash_attention_utils.py official repository unverified MIT (permissive) · a77b72cc8d5c90f0 · report
read_field wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_gguf_pytorch_utils.py official repository unverified MIT (permissive) · befefbd3afa0731c · report
reverse_permute_weights wyzjack/CNTP/custom_transformers_packages/gsm8k_strategyqa/transformers/modeling_gguf_pytorch_utils.py official repository unverified MIT (permissive) · bb3c4ecbbdf8a9c1 · report

Tasks

Prediction

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections