HyperAIHyperAI

Command Palette

Search for a command to run...

il y a 4 mois

Apprentissage contrastif guidé par pseudo-étiquettes pour la segmentation d’images médicales semi-supervisée

{Zhaozheng Yin Hritam Basak}

Apprentissage contrastif guidé par pseudo-étiquettes pour la segmentation d’images médicales semi-supervisée

Résumé

Bien que les travaux récents en apprentissage semi-supervisé (SemiSL) aient remporté un succès significatif dans la segmentation d’images naturelles, la tâche d’apprentissage de représentations discriminatives à partir d’étiquetages limités reste un problème ouvert dans le domaine des images médicales. Les cadres d’apprentissage contrastif (CL) reposent sur la notion de mesure de similarité, utile pour les problèmes de classification, mais ils échouent à transférer ces représentations de haute qualité vers une segmentation précise au niveau des pixels. À cet effet, nous proposons un nouveau cadre d’apprentissage contrastif basé sur des patches, en apprentissage semi-supervisé, dédié à la segmentation d’images médicales, sans recourir à aucune tâche prétexte explicite. Nous exploitons conjointement le pouvoir du CL et du SemiSL : les pseudo-étiquettes générées par le SemiSL apportent une guidance supplémentaire au CL, tandis que les informations de classe discriminatives apprises par le CL permettent d’obtenir une segmentation multi-classes précise. Par ailleurs, nous formulons une nouvelle fonction de perte qui encourage de manière synergique la séparabilité inter-classes et la compacité intra-classes des représentations apprises. Une nouvelle carte de disparité sémantique entre patches, fondée sur l’entropie moyenne des patches, est introduite afin de guider l’échantillonnage des positifs et des négatifs dans le cadre CL proposé. Une analyse expérimentale menée sur trois jeux de données publiques issus de modalités multiples démontre l’avantage de notre méthode par rapport aux approches de pointe. Le code est disponible à l’adresse suivante : https://github.com/hritam-98/PatchCL-MedSeg.

Benchmarks

BenchmarkMéthodologieMétriques
colorectal-gland-segmentation-on-cragPatchCL
Dice: 0.892
F1-score: 0.881
Hausdorff Distance (mm): 119.5
semi-supervised-medical-image-segmentation-on-2PatchCL
Dice (Average): 91.20
semi-supervised-medical-image-segmentation-on-3PatchCL
Dice (Average): 89.10
semi-supervised-semantic-segmentation-on-34PatchCL
Avg DSC: 0.919

Construire l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec du co-codage IA gratuit, un environnement prêt à l'emploi et les meilleurs prix GPU.

Co-codage IA
GPU prêts à utiliser
Meilleurs prix
Commencer

Hyper Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp