phanerozoic commited on
Commit
52a3a00
·
verified ·
1 Parent(s): a517c96

Stage 4: ship ep3 checkpoint (peak F1 0.717 vs ep10 0.710)

Browse files
Files changed (1) hide show
  1. 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.710 on the 1000-image COCO val subset (Stage 0 classifier applied on top of the student's 40-D output, threshold swept).
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. F1 converges near 0.71. 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 1K
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.710
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.