Contrastive learning: an efficient Domain Adaptation strategy for 2D mammography image classification - Centre Borelli UMR 9010 Accéder directement au contenu
Communication Dans Un Congrès Année : 2024

Contrastive learning: an efficient Domain Adaptation strategy for 2D mammography image classification

Vincent Jugnon
  • Fonction : Auteur
  • PersonId : 1078009
Agnès Desolneux
  • Fonction : Auteur
  • PersonId : 1114785
Mathilde Mougeot
  • Fonction : Auteur
  • PersonId : 1370321

Résumé

Effective computer aided breast-cancer diagnosis models using 2D mammography images must maintain consistent performance across varying image acquisition systems and post-processing techniques. Nevertheless, Deep Learning (DL) models have shown diminished performance with variations in image style and contrast. We propose two models trained for classifying respectively 2D mammography patches and complete images, using heterogeneous datasets distinguished by different image post-processing methods. We propose a Domain Adaptation (DA) methodology using Supervised Contrastive Learning (SCL) to achieve domain-invariant representations and improved class-separability. This approach is compared to a standard training using the Cross Entropy (CE) loss. The domain invariant models outperform those trained with CE in binary classification of full mammograms (cancer vs. no cancer), increasing the AUC from 0.745 to 0.816 in an independent test set. For patch classification, we show that the Domain Adaptation effectiveness varies with weight initialization and dataset size.
Fichier principal
Vignette du fichier
ISBI24_paper_871_preprint.pdf (6.62 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04577704 , version 1 (16-05-2024)

Identifiants

  • HAL Id : hal-04577704 , version 1

Citer

Gonzalo Iñaki Quintana, Vincent Jugnon, Laurence Vancamberg, Agnès Desolneux, Mathilde Mougeot. Contrastive learning: an efficient Domain Adaptation strategy for 2D mammography image classification. 2024 IEEE 21st International Symposium on Biomedical Imaging (ISBI), IEEE, May 2024, Athens, Greece. ⟨hal-04577704⟩
0 Consultations
0 Téléchargements

Partager

Gmail Mastodon Facebook X LinkedIn More