Shinzmann commited on
Commit
fdfd364
·
verified ·
1 Parent(s): ea78107

Add W&B training curves to model card

Browse files
Files changed (1) hide show
  1. README.md +10 -1
README.md CHANGED
@@ -18,7 +18,9 @@ under 18, and an uncertainty, from a single face crop. **It is a signal, not a s
18
 
19
  - Backbone: EfficientNetV2-S (`tf_efficientnetv2_s`), ImageNet-pretrained.
20
  - Heads: age regression, under-18 logit, heteroscedastic (aleatoric) uncertainty.
21
- - Trained 30 epochs on an H200. Selection minimizes `MAE + 5 x MPTR@18` (a child-safety composite).
 
 
22
 
23
  ## Intended use
24
  Input: one detected, cropped, 224x224 RGB face. Output: `{estimated_age, p_under_18, uncertainty,
@@ -40,6 +42,13 @@ MAE by skin band: very_light 5.46, light 5.72, intermediate 5.50, tan 5.99, brow
40
  MAE by age band: 0-12 4.04, 13-15 6.10, 16-17 5.37, 18-20 4.85, 21-25 4.30, 26-35 5.22, 36-50 6.67,
41
  51+ 8.51. GPU eval latency p50 14.2 ms.
42
 
 
 
 
 
 
 
 
43
  ## Limitations and safety
44
  MAE is competitive, but **MPTR@18 is high**: about a third of true minors are scored as adults, and
45
  higher for dark and brown skin. So the CNN must not gate alone. Kámárí mitigates this with a
 
18
 
19
  - Backbone: EfficientNetV2-S (`tf_efficientnetv2_s`), ImageNet-pretrained.
20
  - Heads: age regression, under-18 logit, heteroscedastic (aleatoric) uncertainty.
21
+ - Trained 30 epochs on an H200 (batch 512, AdamW lr 3e-4 wd 1e-4, cosine schedule, bf16; ~15 min
22
+ wall-clock; 22,224 train / 2,529 val exact-age rows). Selection minimizes `MAE + 5 x MPTR@18`
23
+ (a child-safety composite). The full run is tracked in Weights & Biases (project `kamari`).
24
 
25
  ## Intended use
26
  Input: one detected, cropped, 224x224 RGB face. Output: `{estimated_age, p_under_18, uncertainty,
 
42
  MAE by age band: 0-12 4.04, 13-15 6.10, 16-17 5.37, 18-20 4.85, 21-25 4.30, 26-35 5.22, 36-50 6.67,
43
  51+ 8.51. GPU eval latency p50 14.2 ms.
44
 
45
+ ## Training curves
46
+ Pulled from the Weights & Biases run (project `kamari`, run `cnn-tf_efficientnetv2_s`). Training loss
47
+ is a heteroscedastic Gaussian NLL, so it is allowed to go negative as the variance head sharpens;
48
+ validation MAE falls to 5.74 and the safety metric (MPTR@18) is tracked alongside it.
49
+
50
+ ![CNN training curves](https://raw.githubusercontent.com/Mystique1337/kamari/main/docs/assets/training/cnn_training_curves.png)
51
+
52
  ## Limitations and safety
53
  MAE is competitive, but **MPTR@18 is high**: about a third of true minors are scored as adults, and
54
  higher for dark and brown skin. So the CNN must not gate alone. Kámárí mitigates this with a