File size: 1,691 Bytes
81a9de1
 
 
 
8dbf153
 
 
 
81a9de1
 
8dbf153
81a9de1
8dbf153
81a9de1
8dbf153
81a9de1
8dbf153
81a9de1
8dbf153
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
---
base_model: zai-org/GLM-Z1-9B-0414
library_name: transformers
tags:
- eklav
- math-reasoning
- cot-distillation
- reasoning
---

# Eklav-9B-Math

Eklav trains a model to pick up a teacher's reasoning mid thought rather than imitate it end to end. The student sees a partial reasoning trace from the teacher, with the answer revealing tail removed, and learns to continue reasoning and produce the answer on its own. The model's own reasoning is conditioned on the teacher's partial trace during training rather than trained to reproduce it word for word. Same base model, same training data as standard full trace CoT distillation, only the training objective changes.

## Highlights

- -0.7% average pass@1 across 6 math benchmarks vs. standard full trace CoT SFT, same base model and training data (MMLU excluded, general knowledge check not a math benchmark)

## Model details

| | |
|---|---|
| Base model | [zai-org/GLM-Z1-9B-0414](https://huggingface.co/zai-org/GLM-Z1-9B-0414) |
| Task | Math reasoning |
| Training method | Eklav (hint conditioned SFT) |
| Format | Merged bf16 checkpoint |

## Results

Pass@1 (%), single evaluation run per benchmark.

| Benchmark | AIME 2025 | AIME 1983-2024 | GPQA-Diamond | GSM8K | MATH-500 | Omni-MATH | MMLU |
|---|---|---|---|---|---|---|---|
| CotGen | 35.2 | 59.3 | 49.8 | 95.5 | 91.2 | 39.8 | 81.0 |
| Eklav | 29.0 | 60.7 | 49.1 | 95.3 | 92.5 | 41.6 | 81.0 |

## Usage

```python
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "AdarshSingh7647/Eklav-9B-Math"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto", device_map="auto")
```