Add W&B training curves to model card
Browse files
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
|
|
|
|
|
|
|
| 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 |
+

|
| 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
|