In plain words
Some language models write by gradually changing noisy states into words. Lowering the temperature is meant to favor more likely answers, but standard tempering changes which answers are favored without changing the model’s original movements between noisy states.
The paper introduces Bayes-Consistent Tempering, which coordinates those changes. On one math test, its trained version solved 27.0% of problems at temperature 0.1, compared with 17.5% for standard tempering after the same amount of continued training. A problem counted as solved when a generated program returned the reference answer.
The research
When a flow language model is pushed toward more likely answers by lowering temperature, does its generation process still match the intermediate states it encounters?
Standard tempering changes the implied states but keeps the original sampling motions, leaving a mismatch between what the model was trained on and how it generates.
BCT aligns temperature-aware training and sampling with the same intermediate-state bridge, then tests whether this improves low-temperature generation.
The researchers compared BCT-L with standard tempering after both models continued from the same released S-arch checkpoint for 2,000 updates.
On GSM8K, where generated programs are checked against reference answers, BCT-L reached 27.0% mean accuracy at temperature 0.1 across three generation seeds, versus 17.5% for the continued standard-tempering model.
A control found that continuing the original training recipe alone changed standard-tempering accuracy by no more than one percentage point at tested temperatures on either S-FLM backbone.
Together, these results support the view that temperature-aware training contributes to BCT-L’s advantage in this setting, though the samplers also differ.
The GSM8K comparison changes both the continuation training objective and the inference sampler, so it does not isolate training as the cause of the gain. The evidence covers specific models, tasks and two bridge families; it does not establish results for other settings.
Analysis
Deep
- Original paper authors
- Daisuke Oba, Hiroki Furuta, Naoaki Okazaki
- Submitted to arXiv
- License
- CC BY 4.0