# Learning from Synthetic Data without Model Collapse in Iterative Instruction Tuning

> KITE prevents model collapse in synthetic-data instruction tuning by targeting weak skills via DINA-based diagnosis and curating examples near the model's semantic knowledge boundary, achieving sustained monotonic improvement across benchmarks.

- **Source:** [arXiv](https://arxiv.org/abs/2607.17043)
- **Published:** 2026-09-05
- **Permalink:** https://picx.dev/p/6wb4Oh
- **Whiteboard:** https://picx.dev/p/6wb4Oh/image

## Summary

## Summary (Overview)

- **Core Problem**: Model collapse in iterative instruction tuning with synthetic data is not uniform degradation but a **polarization of competence** — synthetic training reinforces already-strong skills while further degrading weak ones.
- **Key Contribution**: The paper proposes **KITE (Knowledge-boundary Instruction Tuning via Exploration)**, a two-stage framework combining failure-guided data generation with boundary-aware uncertainty curation.
- **Diagnostic Finding**: Using the DINA cognitive assessment model, the authors show that fine-tuning on accumulated synthetic data strengthens skills like "Data" and "Money & Finance" while weakening "Algebra" and "Time & Scheduling" — revealing actionable granularity for data curation.
- **Method**: Stage 1 generates candidate instructions targeting diagnosed weaknesses with rank-based noise injection; Stage 2 curates examples near the model's semantic knowledge boundary using a novel **Kernel Boundary Uncertainty (KBU)** score.
- **Results**: KITE consistently outperforms strong baselines (Self-Instruct, Few-shot, CDS, ToEdit) across 4 benchmarks and 5 open-source LLMs, with stable long-horizon improvement over 9 generations.

---

## Introduction and Theoretical Foundation

### Background and Motivation

Synthetic data has become essential in LLM post-training, but introduces a fundamental challenge: **model collapse** (Shumailov et al., 2024). As later-generation models train on data reflecting previous generations' outputs, the training distribution becomes distorted — losing diversity and fidelity to real-world data, reducing long-tail coverage, and harming held-out performance.

### Key Gap in Existing Work

The paper identifies two critical gaps in prior research:

1. **Objective mismatch**: Existing work focuses on *bounding* degradation (keeping it finite), but in iterative model evolution, the meaningful goal is ensuring each successive model *improves* over its predecessor.
2. **Granularity problem**: Aggregate-level collapse analysis (overall accuracy drop, long-tail narrowing) is too coarse to guide synthetic data construction. Collapse could stem from data being too narrow/repetitive (needs diversification) or from specific skill weaknesses (needs targeted construction).

### Theoretical Foundation

The paper leverages **DINA (Deterministic Inputs Noisy And gate)** model from cognitive assessment theory. DINA infers latent skill mastery from response patterns across many examples, treating each training instance as requiring a binary skill vector (from a Q-matrix). This provides structured weakness estimates beyond simple failure counting.

### The Polarization Finding

Using GSM8K as an illustrative case, the authors estimate skill mastery for both base and fine-tuned models:

> "Synthetic data improves several skills that were previously strong enough, including Data and Money & Finance. On the other hand, some weak skills not fully mastered by the base model become even weaker... such as Algebra and Time & Scheduling."

This demonstrates that collapse in instruction tuning is **polarization of competence**, not uniform degradation.

---

## Methodology

### Notation and Formulation

Let $M_t$ denote the model at evolution step $t$ with output distribution $p_{M_t}(y|x)$. The pipeline:

1. Start from verified training set $D_0^\star = \{(x_i, y_i^\star)\}_{i=1}^{n_0}$
2. At step $t$: generate candidate bank $\mathcal{B}_t = \{x_i\}_{i=1}^{N}$ with $N \gg B$
3. Select $D_t \subseteq \mathcal{B}_t$ with $|D_t| = B$ (labeling budget)
4. Label to $D_t^\star$, train next model on union of all verified data via SFT

Each $M_t$ is fine-tuned from the **same base model** on accumulated data.

### Stage 1: Failure-Guided Candidate Bank Construction

**1) Weakness profiling via DINA**: Each training instance has a binary skill requirement vector from a Q-matrix (constructed via LLM-assisted skill tagging). DINA estimates posterior mastery per skill; low-mastery skills become natural-language weakness descriptors $S_t = \{s_k\}_{k=1}^{K}$, used to instantiate question-generation prompts $\pi_Q(s)$.

**2) Rank-based noise injection**: To avoid easy/templated instructions, the paper perturbs token probabilities by rank:

$$
q_t^{(\alpha)}(v \mid c) = \operatorname{softmax}\left(\ell_t(c) + \alpha \log r_t(\cdot \mid c)\right)_v \propto q_t(v \mid c) \cdot r_t(v \mid c)^{\alpha} \tag{1}
$$

where $r_t(v \mid c) = 1 + |\{u : \ell_{t,u}(c) > \ell_{t,v}(c)\}|$ is the rank of token $v$. This smoothly upweights lower-ranked tokens, encouraging exploration without hard truncation.

### Stage 2: Uncertainty-Based Data Curation (KBU)

**Semantic uncertainty via kernel**: For each candidate $x$, sample $m$ answers $y^{(1)}, \ldots, y^{(m)} \sim M_t(\cdot \mid x)$, embed them as $h_i = \phi(y^{(i)})$, and define an RBF kernel:

$$
K_{ij} = \exp\left(-\frac{\|h_i - h_j\|^2}{2\sigma^2}\right), \quad \sigma^2 = \operatorname{median}\{\|h_i - h_j\|^2 : i < j\} \tag{2}
$$

**Likelihood weighting**: To suppress off-manifold low-probability generations, each sample is weighted by length-normalized log-likelihood:

$$
\ell_i = \frac{1}{|y^{(i)}|} \sum_k \log p_{M_t}\left(y_k^{(i)} \mid x, y_{<k}^{(i)}\right) \tag{3}
$$

$$
\tilde{w}_i = \frac{\exp(\gamma \ell_i)}{\sum_j \exp(\gamma \ell_j)} \tag{4}
$$

**KBU score**: Form unit-trace matrix $A = \frac{W K W}{\operatorname{tr}(W K W)}$ where $W = \operatorname{diag}(\sqrt{\tilde{w}_1}, \ldots, \sqrt{\tilde{w}_m})$, then compute Rényi-2 entropy:

$$
U_{\mathrm{KBU}}(x) = -\log \operatorname{tr}(A^2) \tag{5}
$$

**Selection criterion**: Choose candidates whose KBU falls within percentile range $(u_{\min}, u_{\max})$ — intermediate values indicate disagreement among plausible answers (near knowledge boundary), while too-low (too easy/confidently wrong) or too-high (beyond actionable boundary) values are filtered out.

---

## Empirical Validation / Results

### Main Results (Table 1)

| Model | Method | GSM8K | MMLU-Pro | MATH | GPQA | Average |
|-------|--------|-------|----------|------|------|---------|
| **Qwen-3-4B** | Initial | 93.63 | 69.55 | 80.80 | 40.40 | 71.10 |
| | Human data | 94.28 | 70.40 | 81.60 | 41.41 | 71.92 |
| | Self-Instruct | 93.79 | 70.12 | 81.00 | 40.90 | 71.45 |
| | Few-shot syn. | 93.58 | 70.01 | 80.80 | 40.40 | 71.20 |
| | CDS | 94.03 | 70.50 | 81.20 | 41.41 | 71.79 |
| | ToEdit | 94.50 | 70.40 | 81.60 | 41.41 | 71.98 |
| | **Ours (KITE)** | **95.04** | **71.87** | **82.60** | **41.92** | **72.86** |
| **Llama-3-8B** | Initial | 78.32 | 38.75 | 24.40 | 32.32 | 43.45 |
| | Human data | 79.98 | 39.38 | 26.00 | 32.83 | 44.55 |
| | **Ours (KITE)** | **81.12** | **39.85** | **26.60** | **33.33** | **45.23** |

**Key observations**:
- KITE achieves best average performance within every model group (Qwen-3-4B, Qwen-3-1.7B, Llama-3.2-3B, Llama-3-8B, Gemma-3-4B)
- Baselines remain below the Human data ceiling — Self-Instruct and Few-shot show limited gains; CDS improves over generic synthesis but below KITE; ToEdit remains consistently below KITE
- Gains on MMLU-Pro and GPQA are smaller (these benchmarks require broader knowledge harder to improve via instruction tuning alone)

### Ablation Study (Table 2, Llama-3-8B-Instruct)

| Ablation | GSM8K | MMLU-Pro | MATH | GPQA | Avg. |
|----------|-------|----------|------|------|------|
| w/o weakness profiling | 76.2 | 39.1 | 25.0 | 32.3 | 43.2 |
| w/o rank-noise | 80.4 | 39.9 | 26.2 | 33.3 | 44.9 |
| token entropy (vs KBU) | 79.9 | 39.3 | 25.0 | 31.3 | 43.9 |
| w/o likelihood weighting | 80.0 | 39.3 | 26.6 | 32.3 | 44.6 |
| Band-pass (0.4, 1.0) | 79.7 | 39.5 | 26.2 | 33.3 | 44.7 |
| Band-pass (0.0, 0.6) | 79.5 | 39.7 | 26.2 | 32.8 | 44.6 |
| **Full method** | **81.1** | **39.9** | **26.6** | **33.3** | **45.2** |

All components contribute: weakness profiling (+2.0 avg), rank-noise (+0.3), KBU over token entropy (+1.3), likelihood weighting (+0.6), and intermediate band-pass range (+0.5).

### Long-Horizon Evolution (9 Generations)

KITE on Llama-3.2-3B-Instruct shows **monotonic improvement** on every benchmark with no degradation-and-reversal signature of collapse — gains saturate gradually, indicating well-behaved self-improvement.

### Generalization to Non-Reasoning Tasks

On wikitext-2 recursive training:
- Standard decoding (low/high temperature) produces synthetic data concentrated in low-perplexity regions
- KITE yields broader distribution closer to human-written data
- High-temperature synthetic data causes test perplexity to increase monotonically across generations, while KITE keeps perplexity stable

### OOD Generalization

KITE trained on in-domain math subsets improves on unseen OlymMATH and AIME benchmarks, suggesting it enhances underlying reasoning skills rather than merely fitting benchmark distributions.

---

## Theoretical and Practical Implications

### Theoretical Implications

1. **Reframing model collapse**: The paper shifts the framing from "bounded degradation" to "sustained improvement," showing collapse manifests as *competence polarization* rather than uniform decay — a finer-grained characterization with direct implications for data construction.

2. **Actionable diagnosis**: DINA-based skill profiling provides a structured, interpretable diagnosis of *which* skills need attention, bridging cognitive assessment theory with LLM training.

3. **Semantic knowledge boundary**: The KBU score operationalizes the concept of a "knowledge boundary" — the region where multiple plausible answers coexist — as the sweet spot for learning signal, validated by the band-pass ablation results.

### Practical Implications

1. **Data curation pipeline**: KITE offers a practical two-stage recipe: diagnose weaknesses → generate targeted candidates with rank-noise → curate via likelihood-weighted semantic uncertainty.

2. **Verifier budget efficiency**: By generating a large candidate bank ($N \gg B$) and curating to budget $B$, KITE concentrates labeling effort on examples with genuine learning value.

3. **Beyond reasoning tasks**: The KBU curation component generalizes to non-reasoning settings (e.g., wikitext-2) even without weakness profiling, suggesting broad applicability.

---

## Conclusion

### Main Takeaways

The paper demonstrates that model collapse in synthetic-data instruction tuning manifests as **polarization of competence** — reinforcing strong skills while degrading weak ones. The proposed **KITE** framework directly counteracts both arms of polarization:

1. **Failure-guided generation** (Stage 1) targets weak skills via DINA-based diagnosis and rank-based noise injection
2. **Boundary-aware KBU curation** (Stage 2) selects examples near the semantic knowledge boundary where learning signal is maximal

### Future Directions

The authors acknowledge limitations and suggest:
- Scaling KITE to larger models and RL-style post-training
- Expert-validated Q-matrix construction for more calibrated skill diagnosis
- Per-skill re-diagnosis after KITE training to track skill-level evolution
- Broader task coverage beyond reasoning benchmarks

The key takeaway: **sustained self-improvement from synthetic data requires not just generating diverse data, but generating data targeted at current weaknesses and curated at the model's actionable knowledge boundary.**

---

_Markdown view of https://picx.dev/p/6wb4Oh, served by PicX — AI-generated visual whiteboard summaries of research papers._
