Files

171 lines
10 KiB
Markdown
Raw Permalink Normal View History

---
license: apache-2.0
base_model: Qwen/Qwen2.5-7B
language:
- fr
- en
library_name: transformers
pipeline_tag: text-generation
tags:
- qwen2.5
- sft
- dpo
- rlvr
- grpo
- trl
- instruction-tuning
datasets:
- allenai/tulu-3-sft-mixture
- allenai/tulu-3-pref-personas-instruction-following
- HuggingFaceH4/ultrafeedback_binarized
- openai/gsm8k
- allenai/RLVR-IFeval
---
# Qwen2.5-7B-base2instruct
Modèle conversationnel construit **à partir du modèle de base** [`Qwen/Qwen2.5-7B`](https://huggingface.co/Qwen/Qwen2.5-7B)
(non-instruct), via un pipeline **SFT → DPO → RLVR** entièrement reproduit **sur un seul GPU H100 80 Go**.
Objectif de l'expérience : reproduire nous-mêmes la transformation base→instruct, la comparer à l'instruct
officiel [`Qwen/Qwen2.5-7B-Instruct`](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct), et **en tirer une recette
et des leçons réutilisables**.
📦 **Code, scripts et recette complète** : https://github.com/AlexandreFenyo/qwen2.5-7b-base2instruct
---
## Résultats (lm-evaluation-harness, backend vLLM)
| modèle | IFEval (suivi d'instructions) | GSM8K (maths) | MMLU (connaissances) |
|---|---|---|---|
| base Qwen2.5-7B | 27.4 | 83.0 | 71.8 |
| + SFT (300k) | 51.2 | 77.6 | 69.2 |
| + DPO ciblé | 68.9 | 80.1 | 70.0 |
| **+ RLVR gradué (ce modèle)** | **75.0** | 79.7 | **70.2** |
| instruct officiel | 71.9 | 84.7 | 68.8 |
**Ce modèle dépasse l'instruct officiel sur IFEval (75.0 vs 71.9) et MMLU (70.2 vs 68.8)**, et reste en
retrait sur les maths (79.7 vs 84.7). ⚠️ Honnêteté : on a *spécialisé* le modèle vers le suivi d'instructions
(DPO et RLVR tous deux ciblés là-dessus) ; l'officiel est généraliste. On le bat donc sur l'axe optimisé,
pas « partout » (chat ouvert, sécurité, code, multi-tours non évalués ici).
---
# Un cours condensé : transformer un modèle de base en modèle instruct
## 0. L'idée fondamentale : savoir ≠ savoir obéir
Un modèle de **base** (pré-entraîné sur ~18 000 milliards de tokens) possède **déjà** la connaissance et
le raisonnement. La preuve : avant tout post-training, il fait **83 sur GSM8K** et **71.8 sur MMLU**. Ce
qui lui manque, c'est le **comportement d'assistant** : répondre à une consigne, en 0-shot, dans un format
de chat. Le post-training **n'ajoute pas de savoir — il installe un comportement**.
C'est visible dans le tableau : le base est fort en MMLU/GSM8K (évalués en *few-shot*, par complétion de
motif, sans obéissance requise) mais **faible en IFEval (27.4)** — précisément la compétence « obéir à une
consigne » qui lui manque. Tout le pipeline sert à combler ce 27 → 75.
## 1. SFT — apprendre le format chat (full fine-tuning)
- **Méthode** : *Supervised Fine-Tuning*. On entraîne le modèle à reproduire des réponses d'assistant au
format **ChatML** (`<|im_start|>role … <|im_end|>`), loss calculée **sur la réponse seule**
(`assistant_only_loss`).
- **Données** : [`allenai/tulu-3-sft-mixture`](https://huggingface.co/datasets/allenai/tulu-3-sft-mixture),
sous-ensemble de **300k** exemples.
- **Effet** : IFEval **27 → 51**. C'est le SFT qui « réveille » le comportement d'assistant et la plus
grande part du raisonnement latent du base.
- **Ce qui a marché** : plus de données SFT aide (180k→300k : +6 IFEval vs une version antérieure).
- **Ce qui a coincé** : **instabilité numérique**. Le SFT a divergé deux fois (gradient `NaN`) à cause
d'**overflows bf16 du forward** sur des échantillons pathologiques (logits → `inf`). Correctif :
un **garde-fou** qui neutralise les gradients non-finis avant le pas d'optimisation (`nan_to_num`) →
le batch fautif est sauté sans corrompre le modèle. *Leçon : sur de longs runs full-FT en bf16, prévoir
une protection anti-NaN.*
## 2. DPO — aligner sur des préférences (l'étape décisive)
- **Méthode** : *Direct Preference Optimization*. Apprentissage **hors-ligne** à partir de paires
`chosen`/`rejected` : le modèle apprend à préférer la bonne réponse à la mauvaise. Implémenté en **LoRA**
(référence = modèle adaptateur désactivé).
- **Ce qui a moins marché — DPO générique** : avec
[`ultrafeedback_binarized`](https://huggingface.co/datasets/HuggingFaceH4/ultrafeedback_binarized)
(préférences *générales* de qualité), IFEval **n'a pas bougé** (≈ +0). Le signal « réponse globalement
meilleure » ne dit rien sur « réponse qui respecte la consigne ».
- **Ce qui a été DÉCISIF — DPO ciblé** : avec
[`allenai/tulu-3-pref-personas-instruction-following`](https://huggingface.co/datasets/allenai/tulu-3-pref-personas-instruction-following)
(~20k paires où la réponse *choisie respecte une contrainte* et la *rejetée la viole*) + 10k génériques
pour la diversité, IFEval **51 → 69 (+17.7)** — quasi le niveau officiel, d'un coup.
- **Leçon centrale** : *même algorithme, données génériques vs ciblées → +17.7 d'écart.* Pour progresser
sur une capacité, il faut des données qui **ciblent exactement cette capacité**.
## 3. RLVR via GRPO — récompense vérifiable (amplifier)
- **Méthode** : *Reinforcement Learning with Verifiable Rewards*, algorithme **GRPO** (Group Relative
Policy Optimization). Le modèle **génère lui-même** ~8 réponses par prompt ; un **programme vérificateur**
note chacune ; l'avantage = écart à la moyenne du groupe → renforce les meilleures. En **LoRA**, génération
via transformers (vLLM 0.22 incompatible avec le GRPO de TRL 1.5.1).
- **Récompenses vérifiables** (sans reward model) :
- **maths** : réponse correcte (parse `\boxed{}`, [`openai/gsm8k`](https://huggingface.co/datasets/openai/gsm8k))
- **suivi d'instructions** : contraintes vérifiées par programme (24 validateurs maison : minuscules,
nombre de paragraphes, mots-clés, format… inspirés de
[`allenai/RLVR-IFeval`](https://huggingface.co/datasets/allenai/RLVR-IFeval)).
- **Ce qui a échoué — récompense BINAIRE** : 1 contrainte/prompt, reward 0/1. Résultat : **effondrement de
l'avantage GRPO**. Dans un groupe de 8, soit toutes réussissent (contrainte facile), soit toutes échouent
(dure) → écart-type nul → **gradient nul** (`frac_reward_zero_std → 1.0`). IFEval inchangé.
- **Ce qui a marché — récompense GRADUÉE** : prompts à **2-3 contraintes**, reward = **fraction satisfaite**
(0 / 0.33 / 0.66 / 1.0) → variance dans le groupe → gradient vivant → IFEval **69 → 75 (+6.1)**.
- **Leçon** : le RLVR **amplifie** une capacité que le modèle sait *déjà parfois* produire (ici, le DPO
l'avait installée) — « installer (DPO/SFT) puis amplifier (GRPO) ». Si le modèle ne sait pas du tout
faire, GRPO n'a rien à renforcer (pas de variance).
---
## Les 3 grandes leçons transférables
1. **Le savoir vient du base ; le post-training installe un comportement.** Reproduire une *capacité*
précise est bon marché ; ce qui est cher (pré-entraînement) est hérité gratuitement du base open-weights.
2. **Le signal doit cibler la capacité visée.** Vérifié 3 fois : SFT 180k→300k (+6), DPO générique→ciblé
(+17.7), RLVR binaire→gradué (+6). Le générique ne transfère pas.
3. **Installer puis amplifier.** SFT/DPO installent (imitation, hors-ligne) ; GRPO amplifie (exploration,
en-ligne). Le RLVR brille surtout là où existe un **vérificateur** (maths, code, contraintes) et où
l'exploration aide (raisonnement) — pas pour des comportements déjà bien couverts par des paires.
## Piège d'évaluation à connaître
En GSM8K `strict-match`, l'instruct officiel tombait à **21 %** — pur artefact (il n'émet pas le format
`#### N` et sa CoT verbeuse dépassait la limite de tokens). Mesuré équitablement (`flexible-extract`,
1024 tokens) : **84.7 %**. *Toujours vérifier ce que le parser attend avant de conclure.*
---
## Recette & stack
- **SFT** : full fine-tuning, ChatML, `assistant_only_loss`, bf16, SDPA, gradient checkpointing,
optim `adamw_8bit`, lr 4e-6, garde-fou anti-NaN.
- **DPO** : LoRA r=32, β=0.1, lr 5e-6, référence en direct (⚠️ `precompute_ref_log_probs=True` provoque
des NaN en bf16 — à désactiver).
- **RLVR** : GRPO-LoRA r=16, 8 générations/prompt, récompense graduée multi-contraintes, lr 2e-6, 2500 steps.
- **Stack** : PyTorch cu128 · transformers · **TRL 1.5.1** · PEFT · vLLM 0.22 (éval) · lm-eval 0.4.12 ·
suivi **Weights & Biases**.
## Datasets utilisés
| étape | dataset | rôle |
|---|---|---|
| SFT | [`allenai/tulu-3-sft-mixture`](https://huggingface.co/datasets/allenai/tulu-3-sft-mixture) | format chat + instructions |
| DPO | [`allenai/tulu-3-pref-personas-instruction-following`](https://huggingface.co/datasets/allenai/tulu-3-pref-personas-instruction-following) | préférences **ciblées** suivi d'instructions (décisif) |
| DPO | [`HuggingFaceH4/ultrafeedback_binarized`](https://huggingface.co/datasets/HuggingFaceH4/ultrafeedback_binarized) | préférences génériques (diversité) |
| RLVR | [`openai/gsm8k`](https://huggingface.co/datasets/openai/gsm8k) | récompense vérifiable maths |
| RLVR | [`allenai/RLVR-IFeval`](https://huggingface.co/datasets/allenai/RLVR-IFeval) | inspiration contraintes d'instructions |
| Éval | `google/IFEval`, `openai/gsm8k`, `cais/mmlu` | benchmarks |
## Utilisation
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
tok = AutoTokenizer.from_pretrained("fenyo/Qwen2.5-7B-base2instruct")
model = AutoModelForCausalLM.from_pretrained("fenyo/Qwen2.5-7B-base2instruct", torch_dtype="bfloat16", device_map="auto")
msgs = [{"role": "user", "content": "Explique en deux phrases pourquoi le ciel est bleu."}]
inputs = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt").to(model.device)
print(tok.decode(model.generate(inputs, max_new_tokens=200)[0][inputs.shape[1]:], skip_special_tokens=True))
```
Format **ChatML**, identique à l'instruct officiel.
## Limites
Modèle expérimental, **spécialisé suivi d'instructions** (peut être moins poli que l'officiel en chat
ouvert, sécurité, code, multi-tours — non évalués). En retrait sur les maths. Peut halluciner.
## Crédits
Base/instruct de référence : **Qwen2.5** (Alibaba, Apache-2.0). Données : **AllenAI Tülu-3**, HuggingFace H4,
OpenAI GSM8K. Entraînement : **TRL** (Hugging Face). Recette & code :
[github.com/AlexandreFenyo/qwen2.5-7b-base2instruct](https://github.com/AlexandreFenyo/qwen2.5-7b-base2instruct).