Fusion Training for Mathematical Generalization in Large Language Models
Analyze & explore related research in Conceptio
Connect this record directly into your research workflow or AI agent. Explore citations across open repositories, query full text, and export structured proof bundles.
Fusion Training for Mathematical Generalization in Large Language Models Congfeng Cao1 , Pengyu Zhang2 , Jelke Bloem1 1
Institute for Logic, Language and Computation, University of Amsterdam 2 INDE Lab, University of Amsterdam,
arXiv:2608.09893v1 [cs.CL] 10 Aug 2026
Abstract Thinking Mode Fusion (TMF) enables large language models to support both concise responses and long-form reasoning by unifying a non-thinking mode and a thinking mode within a single model. However, its training dynamics, including the data ratio and training schedule between the two modes, remain underexplored. In this work, we present a systematic study of TMF by analyzing the effects of the training schedule and data ratio between thinking and non-thinking modes. Focusing on mathematical problem solving, we construct a benchmark with multiple thinking-to-non-thinking data ratios and three training schedules. Our results reveal an asymmetric interaction between the two modes: increasing the ratio of non-thinking supervision reduces the accuracy of the thinking mode. We further show that different training schedules modulate this trade-off and that the optimal schedule depends on the data ratio. Finally, we quantify a negative correlation between non-thinking and thinking mode supervision, highlighting an inherent tension between these two modes. These findings provide practical guidance for designing effective TMF training settings. All code and data are released to support further research at: Fusion Bench.
1
Introduction
Large language models (LLMs) have made substantial progress in long-form reasoning and have been increasingly applied to complex tasks via extended chains of thought (Wei et al., 2022; Kojima et al., 2022; Wang et al., 2023). However, many real-world queries are comparatively simple and do not require long-form reasoning. For such cases, long-form reasoning can be unnecessarily timeconsuming and computationally expensive. As a result, existing efforts have explored switching models between simple queries and complex problems (OpenAI, 2024, 2025), or adapting the length of model responses to improve reasoning
Thinking data
Non-thinking data
Data Ratio
Training Schedule
Thinking Mode Fusion
Asymmetric interaction between two modes
Training schedule and data ratio modulate this interaction
Figure 1: An overview of the experimental design for analyzing Thinking Mode Fusion under different training orders and data ratios.
efficiency (Arora and Zanette, 2025; Kimi et al., 2025; Shen et al., 2025; Wu et al., 2025; Zhang et al., 2025). Qwen3 series models address this limitation by introducing Thinking Mode Fusion (TMF) during post-training (Yang et al., 2025). TMF unifies a thinking mode and a non-thinking mode within a single model, enabling users to control reasoning behavior for both long-form reasoning and concise responses. TMF is implemented through two different chat templates, i.e., a thinking chat template and a non-thinking chat template, that support mode selection. This design reduces deployment complexity and inference costs while allowing the model to adjust its reasoning depth under different computational budgets. Despite Qwen3’s strong empirical performance, the mechanisms of TMF and the interaction between the two modes remain underexplored. In particular, it remains unclear how the training schedule, i.e., the training order of the two modes, and the data ratio, i.e., the combination of thinking and non-thinking data, affect overall model perfor-
mance. This raises a practical question: when both modes are trained within a single model, how do the training schedule and the data ratio affect the model’s performance? To investigate this question, we focus on mathematics, where problem difficulty varies widely, from short-solution questions to Olympiad-level tasks that require long-form reasoning. As shown in Figure 1, we systematically investigate the interaction between the two modes and analyze the effects of TMF under different training schedules and varying ratios of thinking to non-thinking data. To support this study, we construct a benchmark, Fusion Bench, in accordance with the Qwen3 design and consisting of two types of mathematical problems: simple problems for non-thinking mode training and Olympiad-level problems for thinking mode training under diverse data ratios and training mode combinations. Our results reveal an asymmetric interaction between the two modes: the trade-off between the two modes shows that thinking mode performance degrades as non-thinking supervision increases. Note that TMF unifies two modes (long-form reasoning and concise response) within a single model for a single task, which is different from multi-task settings such as math and code. Moreover, we show that the training schedule modulates this trade-off and that the optimal schedule depends on the data ratio. Finally, we quantify a negative correlation between non-thinking and thinking mode supervision, highlighting an inherent tension between these two modes. Our main contributions are as follows: • We conduct a systematic study of training schedules and data ratios in TMF, revealing a trade-off between thinking and non-thinking modes, where increased non-thinking supervision degrades thinking performance, and we quantify a negative correlation between the two modes. • We show that training schedules modulate this trade-off: while the optimal schedule depends on the data ratio, the Mix training schedule consistently achieves strong performance across both thinking and non-thinking modes when averaged over different ratios. • We construct and release Fusion Bench 1 , a benchmark comprising thinking and nonthinking datasets with varying data ratios and 1
https://anonymous.4open.science/r/Fusion-Bench-2D62
training schedules, enabling controlled and reproducible analysis of TMF.
2
Related Work
Thinking Mode Fusion and Hybrid Thinking. Large language models (LLMs) have shown that supervised fine-tuning (SFT) on long chain-ofthought (Wei et al., 2022) traces is able to improve performance on complex tasks. However, enhancing reasoning depth through specialized post-training increases computational costs due to the long reasoning responses. Models such as OpenAI o1 (OpenAI, 2024), DeepSeekR1 (DeepSeek-AI, 2025), and Gemini (Team, 2025) widely adopt hybrid thinking mechanisms that control whether the model engages in reasoning, thereby achieving a more efficient and flexible reasoning process. Existing efforts to improve reasoning efficiency focus on reducing the length of model responses (Arora and Zanette, 2025; Kimi et al., 2025; Shen et al., 2025). Wu et al. (2025) merge reasoning and non-reasoning models to reduce output length. AdaptThink proposes teaching reasoning models to choose the optimal thinking mode adaptively based on problem difficulty (Zhang et al., 2025). Qwen3 (Yang et al., 2025) introduced Thinking Mode Fusion (TMF), which allows a single model to support both longform reasoning and concise responses controlled by the chat format. Wang et al. (2025) further demystify hybrid thinking, revealing that current hybrid thinking LLMs only achieve partial mode separation, where reasoning behaviors often leak into the no-think mode. However, the mechanisms of TMF remain underexplored. Data Mixture. Prior studies have shown that data mixture plays a critical role in both pre-training and post-training stages. Ye et al. (2025a) outlined the quantitative predictability of model performance with respect to mixture proportions. Liu et al. (2025) propose RegMix to automatically identify a high-performing data mixture by formulating it as a regression task. Li et al. (2025) frame the selection of data proportions as an optimization problem to minimize validation loss across multiple tasks during SFT. While these studies provide insights into domainlevel mixing (e.g., balancing math, code, and chat), they lack an examination of the internal interference between different response formats within the same domain, which is a key focus of our study.
Math Datasets. We group math datasets into two categories that align with the thinking (with long reasoning) versus non-thinking (with short solutions) modes considered in this paper. On the simple data side, a complementary line of work focuses on datasets that admit short solutions or direct answer supervision, which better aligns with the non-thinking mode. Classic benchmarks such as GSM8K (Cobbe et al., 2021), which consists of grade-school math problems, are widely used for instruction tuning and evaluation, and are often employed in short-solution settings to encourage concise generation. Calc-X (Kadlčík et al., 2023) is a collection of simple arithmetic-focused math word problems, comprising over 300,000 curated samples. Patel et al. (2021) introduced MAWPS, an elementary-level benchmark dataset featuring concise solutions and final answers. In this work, we use GSM8K as the non-thinking mode base dataset, as it covers a diverse range of math problems and is widely adopted. On the long reasoning data side, recent work has released large-scale datasets that explicitly provide long reasoning traces, which are crucial for training and analyzing long-form mathematical reasoning. OpenMathReasoning (Moshkov et al., 2025) is a generated large-scale math reasoning dataset for training large language models (LLMs), containing 3.2 million long chain-of-thought (CoT) solutions. OpenMathInstruct-1 (Toshniwal et al., 2024) and OpenMathInstruct-2 (Toshniwal et al., 2025) provide 1.8 million and 14 million generated solutions, respectively, covering a wide range of mathematical difficulty levels. We also note curated math reasoning benchmarks such as OMNI-MATH (Gao et al., 2025) and LIMO (Ye et al., 2025b), which are commonly used to evaluate advanced reasoning capabilities. In addition, several datasets emphasize challenging questions paired with chain-of-thought annotations, including Skywork-MathQA (Zeng et al., 2024), which contains 2.5 million questionanswer pairs, and NuminaMath (LI et al., 2024), which consists of 860, 000 competition-level problems with long-form reasoning. For ease of evaluation, we choose OpenMathReasoning since it provides problems with exact and verifiable final answers, unlike many other datasets that include proof-based or open-ended reasoning tasks without checkable solutions. In this work, we sample data from the OpenMathReasoning subset generated by DeepSeek-R1 (DeepSeek-AI, 2025) and select examples with a high pass rate (greater than 0.96)
Table 1: Benchmark statistics for thinking and nonthinking training data under different data ratios. T denotes thinking mode data and NT denotes non-thinking mode data. We fix the thinking mode subset size (NT = 1500) and vary the non-thinking subset size (NN T ) accordingly. Mean denotes the mean length of the reasoning traces. Mode
Ratio (T : NT) Count Mean Min
Non-thinking
4:1 3:1 2:1 1:1 1:2 1:3 1:4
Thinking
-
375 500 750 1,500 3,000 4,500 6,000
671 672 681 692 692 695 694
330 330 330 298 298 298 298
Max 1,540 1,540 1,540 1,817 1,817 1,863 1,863
1,500 11,877 1,903 54,786
under the 72B model as the thinking-mode training dataset.
3
Data Processing
As mentioned above (Section 2), we select GSM8K (Cobbe et al., 2021) and OpenMathReasoning (Moshkov et al., 2025) as the base datasets for non-thinking and thinking modes, respectively, which are in accordance with the Qwen3 mode design for simple and complex problems. Non-thinking mode dataset. For the nonthinking mode, we use the GSM8K training set. For each experimental setting, we subsample GSM8K into a subset of size NN T according to a target ratio ρ between thinking (NT ) and non-thinking (NN T ) data. We define the ratio as ρ=
NT . NN T
(1)
We sweep ρ from 4 to 1/4 (i.e., non-thinking to thinking ratios from 4:1 to 1:4). Since the GSM8K training set contains only 7,470 examples, to satisfy our experimental design for the data ratio, we fix the amount of thinking mode data to 1,500 examples. With the thinking mode training data fixed at 1,500, the corresponding non-thinking training data size ranges from 375 to 6,000. Thinking mode dataset. To enable controlled comparisons across different fusion strategies and data ratios, we fix the number of thinking mode training examples to NT = 1,500 throughout our benchmark. The OpenMathReasoning dataset contains 3.2 million long chain-of-thought solutions; from this corpus, we sample 1,875 examples. We then randomly split these samples into a 8:2
Table 2: Statistics of test data for thinking and nonthinking evaluation. Mean denotes the mean length of the reasoning traces, Min denotes the minimum length of the reasoning traces, and Max denotes the maximum length of the reasoning traces. Mode
Count
Mean
Min
Max
Non-thinking Thinking
375 375
696 11,992
298 1,840
1,863 69,268
train/test partition. Detailed statistics of the thinking and non-thinking training data are presented in Table 1. For uniform evaluation, we sample 375 test examples from the GSM8K and OpenMathReasoning test sets for the non-thinking and thinking modes, respectively. Detailed statistics of the thinking and non-thinking test data are presented in Table 2. The thinking mode test data have a mean length of 11,992, with a maximum length of 69,268 and a minimum length of 1,840. In contrast, the nonthinking mode test data have a mean length of 696, with a maximum length of 1,863 and a minimum length of 298. After data selection, to align with the Qwen3 official chat template settings, we further process the GSM8K subset into non-thinking mode data and the OpenMathReasoning subset into thinking mode data. For non-thinking mode data, we insert the <think></think> tags before the solutions. For thinking mode data, we wrap the long-form reasoning solutions with the <think></think> tags. To simplify the evaluation process, we wrap all final answers in the \boxed{} tag. Examples of non-thinking and thinking data are shown in Appendix A.4.
4
Methodology
We study Thinking Mode Fusion (TMF) as a post-training problem for large language models (Qwen3 models), where the model support two distinct modes: (i) long-form reasoning for hard problems and (ii) concise responses for simple problems. Our goal is to evaluate how TMF is affected by two controllable factors during supervised finetuning (SFT): (a) training schedule (the ordering or interleaving of thinking and non-thinking mode training) and (b) data ratio (the relative amount of thinking versus non-thinking data, ranging from 1:4 to 4:1, while fixing the amount of thinking mode training data). We use the non-thinking and thinking test sets constructed in Section 3 to evaluate
the uniformity of model performance across modes. Figure 2 provides an overview of our experimental design. 4.1
Supervised Fine-Tuning with Two Modes
We construct two types of training data (thinking and non-thinking) and feed them into a unified supervised fine-tuning (SFT) framework with different training schedules and data ratios. As mentioned in Section 3, we convert examples into the model’s chat template, with the mode specified by the presence or absence of reasoning content within the <think></think> tags. For thinking-mode training, we use the constructed thinking-mode training data, in which long-form reasoning solutions are wrapped in the <think></think> tags. For non-thinking-mode training, we use the constructed non-thinking-mode training data, which contain short solutions following the <think></think> tags. After training, we extract final answers from the \boxed{} tag, as described in Section 3. Finally, we evaluate model performance in both thinking and non-thinking modes using the corresponding test sets in Table 2. 4.2
Data Ratio Control
As described in Section 3, we construct thinking mode and non-thinking mode training data with a wide range of ratios {1:4,1:3,1:2,1:1,2:1,3:1,4:1} by fixing the number of thinking mode training examples. 4.3
Training Schedule
Given a fixed thinking set DT and a ratio-specific (ρ) non-thinking subset DN T , TMF training reduces to constructing a single SFT stream S by a scheduling function (ρ) S = S CHEDULE DT ,DN T ;m ,
(2)
where m specifies how the two modes are arranged during training. We investigate three schedules that cover sequential curricula and interleaving. Sequential: T-NT. We place all thinking mode examples before all non-thinking mode examples: (ρ)
ST-NT = [DT ∥ DN T ].
(3)
This schedule tests whether updates from nonthinking data applied after reasoning training dilute or overwrite reasoning behaviors, and whether reasoning training provides transferable benefits to subsequent concise answering.
Data Ratio
Accuracy Thinking Mode
Thinking data 1:4
1:3
1:2
1:1
2:1
3:1
Non-Thinking Mode
4:1
Thinking Mode Fusion Training Schedule
Nonthinking data
Length
NT-T
Non-Thinking Data
Thinking Data
T-NT
Thinking Data
Non-Thinking Data
Mix
N
T
N
T
Length
Data Ratio
Figure 2: Overview of Thinking Mode Fusion training design. We construct two datasets: thinking mode and non-thinking mode. We vary the data ratio (T:NT from 1:4 to 4:1) and schedule the resulting SFT stream using three strategies: T-NT (first thinking then non-thinking), NT-T (non-thinking then thinking), and Mix (interleaved to match the target ratio). Each configuration fine-tunes Qwen3 model and is evaluated on both thinking and non-thinking benchmarks to quantify interference and trade-offs.
Sequential: NT-T.
We reverse the order: (ρ)
SNT-T = [DN T ∥ DT ].
(4)
This schedule investigates the opposite direction of interference: whether late-stage reasoning supervision disrupts concise response patterns learned from non-thinking data and whether nonthinking alignment persists under subsequent longreasoning updates. Interleaved: Mix. We interleave the two modes of data to the target ratio during training. In particular, we use a simple 1-to-n interleaving procedure: when ρ ≤ 1 (thinking is less frequent), we interleave 1 T (thinking mode) followed by n NT (nonthinking mode) examples, with n ≈ round(1/ρ); when ρ > 1 (thinking is more frequent), we interleave 1 NT followed by n T examples, with n ≈ round(ρ). This schedule makes the intended TMF usage where the model repeatedly alternates between the two modes, enabling us to test whether frequent mode switching alleviates the degradation observed under purely sequential curricula. 4.4
Full Factorial Design
Our TMF analysis is defined by the Cartesian product of data ratios and schedules: (ρ,m) ∈ {4,3,2,1, 12 , 31 , 14 }×{T-NT,NT-T,Mix}.
(5)
For each configuration (ρ, m), we build the corresponding SFT training S while keeping the base
model, supervision format, and optimization hyperparameters fixed. This controlled setup allows us to attribute performance changes to (i) how much thinking supervision is mixed with non-thinking data and (ii) how the two modes are scheduled during training, thereby directly quantifying interference and trade-offs induced by TMF. We finally evaluate both the accuracy and the reasoning length of model responses, as reasoning length has been shown to reflect reasoning quality in mathematical problem solving (Sprague et al., 2025; Jin et al., 2024; Tutek et al., 2025).
5
Experiment and Analysis
5.1
Experiment Setup
Our study aims to answer four research questions: RQ1: Do the non-thinking and thinking modes affect each other’s performance? Although Qwen3 adopts Thinking Mode Fusion to separate long-form and concise reasoning into different fine-tuning modes, training these two modes jointly may still lead to performance tradeoffs or interference in the mathematical domain. RQ2: How does the different training schedule affect Thinking Mode Fusion? Thinking Mode Fusion is introduced by first fine-tuning on concise reasoning traces and then fine-tuning on long-form reasoning traces (Yang et al., 2025). However, in practice, non-thinking and thinking modes can be trained in different orders. As de-