Stage 4: ship ep3 checkpoint (peak F1 0.717 vs ep10 0.710)
Browse files- stage_4/README.md +5 -5
stage_4/README.md
CHANGED
|
@@ -23,25 +23,25 @@ Compact ViT, defined in `student.py`:
|
|
| 23 |
|
| 24 |
## Result
|
| 25 |
|
| 26 |
-
F1 = 0.
|
| 27 |
|
| 28 |
```
|
| 29 |
epoch loss F1 P R
|
| 30 |
1 2.25 0.707 0.55 1.00
|
| 31 |
-
3 2.01 0.717 0.57 0.97
|
| 32 |
5 2.00 0.712 0.56 0.98
|
| 33 |
10 1.99 0.710 0.57 0.95
|
| 34 |
```
|
| 35 |
|
| 36 |
-
Loss plateaus around 2.0 after epoch 2.
|
| 37 |
|
| 38 |
## Comparison
|
| 39 |
|
| 40 |
```
|
| 41 |
-
model params F1 on COCO val
|
| 42 |
Stage 0 (EUPE-ViT-B + classifier) 85.64M 0.894
|
| 43 |
Stage 2 K=10 head prune + classifier 83.67M 0.916
|
| 44 |
-
Stage 4 student + classifier 3.27M 0.
|
| 45 |
```
|
| 46 |
|
| 47 |
26× smaller than the full EUPE-ViT-B pipeline, F1 drop of 0.18 from baseline. A proof that the 40-D target manifold is learnable by a compact specialist but not yet a drop-in replacement for the teacher.
|
|
|
|
| 23 |
|
| 24 |
## Result
|
| 25 |
|
| 26 |
+
F1 = **0.717 at epoch 3** (shipped as `student_final.safetensors`). The training loop evaluated every epoch and epoch 3 is the peak; later epochs drift slightly downward as loss saturates.
|
| 27 |
|
| 28 |
```
|
| 29 |
epoch loss F1 P R
|
| 30 |
1 2.25 0.707 0.55 1.00
|
| 31 |
+
3 2.01 0.717 0.57 0.97 <- shipped
|
| 32 |
5 2.00 0.712 0.56 0.98
|
| 33 |
10 1.99 0.710 0.57 0.95
|
| 34 |
```
|
| 35 |
|
| 36 |
+
Loss plateaus around 2.0 after epoch 2. Precision stays ~0.57 across all epochs while recall is >0.95, the student learns "when in doubt, call it person" and rarely misses a true positive but fires false on about half of person-negatives.
|
| 37 |
|
| 38 |
## Comparison
|
| 39 |
|
| 40 |
```
|
| 41 |
+
model params F1 on COCO val
|
| 42 |
Stage 0 (EUPE-ViT-B + classifier) 85.64M 0.894
|
| 43 |
Stage 2 K=10 head prune + classifier 83.67M 0.916
|
| 44 |
+
Stage 4 student + classifier 3.27M 0.717
|
| 45 |
```
|
| 46 |
|
| 47 |
26× smaller than the full EUPE-ViT-B pipeline, F1 drop of 0.18 from baseline. A proof that the 40-D target manifold is learnable by a compact specialist but not yet a drop-in replacement for the teacher.
|