MeFEm: Medical Face Embedding model
Organizations: Sber AI Lab
Abstract
We present MeFEm, a vision model based on a modified Joint Embedding Predictive Architecture (JEPA) for biometric and medical analysis from facial images. Key modifications include an axial stripe masking strategy to focus learning on semantically relevant regions, a circular loss weighting scheme, and the probabilistic reassignment of the CLS token for high quality linear probing. Trained on a consolidated dataset of curated images, MeFEm outperforms strong baselines like FaRL and Franca on core anthropometric tasks despite using significantly less data. It also shows promising results on Body Mass Index (BMI) estimation, evaluated on a novel, consolidated closed-source dataset that addresses the domain bias prevalent in existing data. Model weights are available at https://huggingface.co/boretsyury/MeFEm , offering a strong baseline for future work in this domain.
Figures & tables
| Dataset name | Original Size | Media type | Usable size |
| FaceCaption-15M | 15 078 935 | images | 4 572 807 |
| AVSpeech | 0 2 621 845 | videos | 1 446 778 |
| SFHQ | 00 425 258 | images | 0 425 258 |
| Total | 6 444 843 | ||
| SHFQ | AVSpeech | FaceCaption | |
|---|---|---|---|
| Good samples | |||
| Bad samples |
Visual Representation Learning
Visual representation learning develops image and video features that capture useful structure for recognition, retrieval, spatial reasoning, and action.
See all 224 papers