jma-informatique/sante-travail-ner-onnx

Santé Travail NER

Modèle de reconnaissance d'entités nommées spécialisé dans les textes de santé au travail.

Le modèle détecte des entités liées aux métiers, symptômes, blessures, expositions professionnelles, équipements de protection, tâches, lieux de travail, dangers et restrictions médicales.

Modèle de base

Ce modèle est un fine-tuning de :

• almanach/camembert-large

Le modèle a ensuite été exporté en ONNX puis quantifié en int8 pour faciliter l'inférence côté navigateur avec Transformers.js.

Types d'entités

Le modèle reconnaît les entités suivantes :

Entité	Description
JOB_ROLE	Métier, fonction ou poste de travail
INJURY	Blessure ou lésion
EXPOSURE_AGENT	Agent d'exposition professionnel
PPE	Équipement de protection individuelle
TASK	Tâche ou activité professionnelle
SYMPTOM	Symptôme déclaré
HEALTH_EFFECT	Effet sanitaire
DISEASE	Maladie ou pathologie
PREVENTIVE_MEASURE	Mesure de prévention
MEDICAL_RESTRICTION	Restriction ou recommandation médicale
WORKPLACE	Lieu ou environnement de travail
HAZARD	Danger ou facteur de risque

Résultats

Évaluation sur le jeu de test :

Métrique	Valeur
F1	0.9194
Précision	0.9114
Rappel	0.9276
Accuracy	0.9710
Loss	0.1003

Exemples

Exemple 1

Texte :

Le soudeur présente une toux persistante suite à l'inhalation de fumées de soudage.


| Entité           | Texte             |
| ---------------- | ----------------- |
| `JOB_ROLE`       | soudeur           |
| `SYMPTOM`        | toux persistante  |
| `EXPOSURE_AGENT` | fumées de soudage |