Compare commits

..
5 Commits
Author SHA1 Message Date
ViperEkura e6be33aa53 doc: add SFT length-filter rationale — high PPL + high variance from A.3 IFD figure
- Add 15-token length floor to SFT samples, with explicit reference
  to Appendix A.3 / Figure 5 (ifd_length_grid)
- Short replies (<10 tokens) show both high per-token perplexity
  (L_uncond ~6-8, PPL ~400-3000 vs long replies ~2-3, PPL ~7-20)
  and wide variance in L_cond (span 0-17.5), which distorts
  downstream IFD-based difficulty estimates.
2026-07-24 06:37:35 +08:00
ViperEkura 0a1d0573ae doc: add SFT length filter (15-token floor) with IFD bias rationale
Per-field length filtering in SFT drops instruction--response pairs
with responses shorter than 15 tokens. Short replies exhibit high
per-token variance in both conditional and unconditional loss
(Appendix A.3 / Figure 5), which would distort downstream IFD-based
difficulty estimates.
2026-07-24 06:27:10 +08:00
ViperEkura 0c2bc916f2 fix: loss_compare caption token count (20B -> 5B)
The loss_compare.png x-axis only spans 0-5B tokens. The caption
incorrectly claimed ~20B tokens.
2026-07-24 06:21:52 +08:00
ViperEkura 4e70e827ff remove: drop ckpt_weight_density_per_run.png figure
The per-run weight density figure contained misleading legacy
iteration labels (500k/1M iter) that contradicted the paper's
stated token budgets (15B). Removing it avoids confusion; the
per-category density plot (ckpt_weight_density.png) and Table 7
remain as the primary evidence.
2026-07-24 06:20:08 +08:00
ViperEkura a7bbc7b29f fix: align SFT/DPO figures and text with actual training data
- SFT: 1,000 steps/WSD → ~3,800 steps/cosine; loss ~2.5→1.6 → ~2.1→1.5
- DPO: fix preference-loss narrative to match training-loss curve;
  correct initial grad-norm (~50 → ~200)
- Table 2: peak LR 1.5e-4 → 2.0e-4 (matches pt_metric.png)
- Clarify scheduling: pretraining=WSD, SFT+DPO=cosine
- Add disclaimer to ckpt_weight_density_per_run caption for legacy iter labels
2026-07-24 06:15:42 +08:00
3 changed files with 35 additions and 38 deletions
Binary file not shown.

Before

Width:  |  Height:  |  Size: 92 KiB

After

Width:  |  Height:  |  Size: 86 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 85 KiB

After

Width:  |  Height:  |  Size: 84 KiB

+35 -38
View File
@@ -35,9 +35,9 @@ storage backends, and a companion SFT pipeline ({\sc Alembic}) with
MinHash deduplication. The 24-layer GQA-SwiGLU decoder is trained with a MinHash deduplication. The 24-layer GQA-SwiGLU decoder is trained with a
hybrid Muon/AdamW optimizer and WSD scheduling under DDP/FSDP. hybrid Muon/AdamW optimizer and WSD scheduling under DDP/FSDP.
Supervised fine-tuning on deduplicated bilingual instructions reduces loss Supervised fine-tuning on deduplicated bilingual instructions reduces loss
from $\sim$2.5 to $\sim$1.6 over 1{,}000~steps; DPO alignment from $\sim$2.1 to $\sim$1.5 over $\sim$3{,}800~steps; DPO alignment
($\beta=0.1$, cosine schedule) on model-generated preference pairs ($\beta=0.1$, cosine schedule) on model-generated preference pairs
converges stably without over-optimisation. A BF16 stability analysis shows stable training-loss convergence without over-optimisation. A BF16 stability analysis
shows that GPT-2 residual scaling ($\sigma_0 = 0.02/\sqrt{2L}$) reduces shows that GPT-2 residual scaling ($\sigma_0 = 0.02/\sqrt{2L}$) reduces
per-block activation variance by a factor of 48, and post-training weight per-block activation variance by a factor of 48, and post-training weight
analysis across three checkpoints confirms that residual-scaled analysis across three checkpoints confirms that residual-scaled
@@ -154,8 +154,16 @@ pipeline proceeds as follows:
previously kept sample $\mathbf{s}'$. previously kept sample $\mathbf{s}'$.
\end{enumerate} \end{enumerate}
An optional LLM-as-Judge scoring module provides multi-dimensional A length filter is applied to SFT samples based on the IFD
quality scores that can be used to filter low-quality samples. length-bias analysis in Appendix~\ref{sec:ifd_bias}
(Figure~\ref{fig:length_bias}): instruction--response pairs
whose response contains fewer than 15 tokens are discarded,
because short replies exhibit both high per-token perplexity
($L_{\text{uncond}} \approx 6\text{--}8$, PPL~$\approx 400\text{--}3000$)
and wide variance in both $L_{\text{cond}}$ and $L_{\text{uncond}}$,
which would distort downstream IFD-based difficulty estimates.
The threshold is applied per-field, analogous to the pretraining
filter described above.
\subsection{DPO Data Generation} \subsection{DPO Data Generation}
@@ -285,7 +293,7 @@ Table~\ref{tab:train_params} lists the key hyperparameters.
\textbf{Hyperparameter} & \textbf{Value} \\ \textbf{Hyperparameter} & \textbf{Value} \\
\midrule \midrule
Precision & BF16 (weights + optimizer states) \\ Precision & BF16 (weights + optimizer states) \\
Optimizer & Hybrid Muon/AdamW$^a$, $\eta=1.5\times10^{-4}$ \\ Optimizer & Hybrid Muon/AdamW$^a$, $\eta=2.0\times10^{-4}$ \\
Betas & $(0.9, 0.95)$, weight decay $0.1$ \\ Betas & $(0.9, 0.95)$, weight decay $0.1$ \\
Gradient clip & Global L2, max norm $1.0$ \\ Gradient clip & Global L2, max norm $1.0$ \\
Scheduler & WSD (warmup 2\%, stable, decay) \\ Scheduler & WSD (warmup 2\%, stable, decay) \\
@@ -302,7 +310,7 @@ Sequence length & 2,048 tokens \\
\centering \centering
\includegraphics[width=0.50\linewidth]{data/loss_compare.png} \includegraphics[width=0.50\linewidth]{data/loss_compare.png}
\caption{Training loss curves: GPT-2 residual scaling vs.~Kaiming \caption{Training loss curves: GPT-2 residual scaling vs.~Kaiming
initialization over $\sim$20B tokens.} initialization over $\sim$5B tokens.}
\label{fig:loss} \label{fig:loss}
\end{figure} \end{figure}
@@ -338,21 +346,21 @@ training, ensuring stable weight updates in the final phase.
\begin{figure}[H] \begin{figure}[H]
\centering \centering
\includegraphics[width=0.85\linewidth]{data/sft_metric.png} \includegraphics[width=0.85\linewidth]{data/sft_metric.png}
\caption{SFT training metrics over 1{,}000 fine-tuning steps on a \caption{SFT training metrics over $\sim$3{,}800 fine-tuning steps on a
mixed Chinese--English instruction dataset: training loss, learning mixed Chinese--English instruction dataset: training loss, learning
rate, and gradient norm. The loss drops rapidly in the first 200 rate, and gradient norm. The smoothed loss decreases from $\sim$2.1 to
steps and then enters a slower decay phase. The learning rate $\sim$1.5, dropping rapidly in the first 500 steps and then entering a
follows a WSD schedule (2\% warmup, stable, decay). Gradient norms stabilize slower decay phase. The learning rate follows a cosine schedule with short linear warmup. Gradient norms stabilise after approximately 500 steps,
after approximately 300 steps, indicating that the fine-tuning indicating that the fine-tuning process has reached a stable optimization
process has reached a stable optimization regime.} regime.}
\label{fig:sft_metric} \label{fig:sft_metric}
\end{figure} \end{figure}
Figure~\ref{fig:sft_metric} shows the supervised fine-tuning metrics Figure~\ref{fig:sft_metric} shows the supervised fine-tuning metrics
for the 1K-step SFT checkpoint used in the IFD analysis for the SFT checkpoint used in the IFD analysis
(Appendix~\ref{app:ifd}). The training loss on the mixed (Appendix~\ref{app:ifd}). The training loss on the mixed
Chinese--English instruction dataset decreases from $\sim$2.5 to Chinese--English instruction dataset decreases from $\sim$2.1 to
$\sim$1.6 over 1{,}000 steps, with the gradient norm converging to a $\sim$1.5 over $\sim$3{,}800 steps, with the gradient norm converging to a
stable range after the warmup phase. stable range after the warmup phase.
\subsection{Direct Preference Optimization} \subsection{Direct Preference Optimization}
@@ -373,10 +381,9 @@ preferred response $y_w$ and dispreferred response $y_l$, the loss is:
\end{equation} \end{equation}
where $\beta$ controls the KL-divergence penalty against the reference. where $\beta$ controls the KL-divergence penalty against the reference.
We use $\beta=0.1$, a batch of 64 preference pairs per step, and the We use $\beta=0.1$, a batch of 64 preference pairs per step, and the
same hybrid Muon/AdamW optimizer. Unlike the pretraining and SFT same hybrid Muon/AdamW optimizer. The pretraining phase employs WSD (warmup--stable--decay) scheduling
phases, which both employ WSD (warmup--stable--decay) scheduling to to maintain a long high-learning-rate plateau. Both SFT and DPO
maintain a long high-learning-rate plateau, DPO alignment uses a alignment use cosine schedules with short linear warmup. The shorter ~3{,}000-step
cosine schedule with short linear warmup. The shorter ~3{,}000-step
alignment run does not benefit from an extended stable phase; instead, alignment run does not benefit from an extended stable phase; instead,
cosine decay lowers the learning rate steadily, which discourages cosine decay lowers the learning rate steadily, which discourages
over-optimisation away from the reference distribution and matches the over-optimisation away from the reference distribution and matches the
@@ -392,17 +399,14 @@ schedule, and gradient norm.}
\end{figure} \end{figure}
Figure~\ref{fig:dpo_metric} summarises the DPO training dynamics. Figure~\ref{fig:dpo_metric} summarises the DPO training dynamics.
The raw preference loss (left panel) starts near $0.69$ and is visibly The raw training loss (left panel) starts near $0.7$ and is visibly
noisy, a hallmark of pairwise preference sampling. The 100-step moving noisy; the smoothed curve reveals a steady downward trend that reaches
average reveals a steady downward trend that reaches a minimum of $\sim$0.1--0.15 by step 3{,}000 without rebound, indicating stable
$\sim$0.51 near step 1{,}200, after which it gently rebounds to convergence. The learning-rate schedule (centre panel) peaks at
$\sim$0.52 and oscillates, suggesting the policy has converged to a $5\times10^{-6}$ after a short linear warmup and then follows cosine
stable preference boundary rather than over-optimising away from the
reference distribution. The learning-rate schedule (centre panel) peaks
at $5\times10^{-6}$ after a short linear warmup and then follows cosine
decay to a floor of $\sim$0.5\,$\times\,$10$^{-6}$. Gradient norms (right panel) start decay to a floor of $\sim$0.5\,$\times\,$10$^{-6}$. Gradient norms (right panel) start
near 50 with occasional spikes above 55, then gradually decline and near 200 with occasional spikes above 250, then gradually decline and
stabilise in the 35--40 range after step 400, indicating consistent stabilise in the 35--50 range after step 1{,}000, indicating consistent
gradient magnitudes throughout alignment. gradient magnitudes throughout alignment.
% ====================================================================== % ======================================================================
@@ -551,12 +555,6 @@ spread.}
\label{fig:ckpt_weight_density} \label{fig:ckpt_weight_density}
\end{figure} \end{figure}
\begin{figure}[H]
\centering
\includegraphics[width=0.95\linewidth]{data/ckpt_weight_density_per_run.png}
\caption{Per-checkpoint weight density breakdowns.}
\label{fig:ckpt_weight_density_per_run}
\end{figure}
% ====================================================================== % ======================================================================
\section{Conclusion} \section{Conclusion}
@@ -577,10 +575,9 @@ non-scaled weights across all checkpoints
Supervised fine-tuning on deduplicated bilingual instructions (processed Supervised fine-tuning on deduplicated bilingual instructions (processed
by the companion {\sc Alembic} pipeline with MinHash deduplication) by the companion {\sc Alembic} pipeline with MinHash deduplication)
reduces training loss from $\sim$2.5 to $\sim$1.6 over 1{,}000~WSD-scheduled reduces training loss from $\sim$2.1 to $\sim$1.5 over $\sim$3{,}800~cosine-scheduled
steps. Subsequent DPO alignment on preference pairs ($\beta=0.1$, cosine steps. Subsequent DPO alignment on preference pairs ($\beta=0.1$, cosine
schedule) converges stably without over-optimisation, reaching a minimum schedule) shows stable training-loss convergence without over-optimisation.
preference loss near step~1{,}200.
Optimizer ablations (Figure~\ref{fig:ckpt_comparison}) demonstrate that Optimizer ablations (Figure~\ref{fig:ckpt_comparison}) demonstrate that
the hybrid Muon/AdamW configuration consistently outperforms pure AdamW the hybrid Muon/AdamW configuration consistently outperforms pure AdamW