Communication Dans Un Congrès Année : 2024

Diffusion model uniform manifold filtering for classification of small datasets with underrepresented classes: Application to chromosomal aberration microscopy detection

Résumé

A frequent problem in biomedical machine learning is the issue of imbalanced classes, especially when datasets are small, which restricts the performance of deep learning methods. To address this issue, generative models are often used to generate additional synthetic data. Specifically, image-to-image models can transform input images to match the characteristics of target images. However, training such models on small datasets can affect the quality of synthetic samples. We propose a new method to filter generative outputs of an image-to-image Brownian Bridge Diffusion Models (BBDMs) using Uniform Manifold Approximation and Projection (UMAP) dimension reduction of a real data classifier's feature space. We apply this methodology to filter synthetic chromosomal aberrations generated from the blue DAPI-colored channels in the context of cytogenetic Fluorescence In Situ Hybridization (FISH) microscopy. Our method shows that such filtered synthetic data significantly enhance classification performance compared to the state of the art CycleGAN and could potentially be applied to a variety of other generative models.
Fichier principal
Vignette du fichier
QTallon_BBM_FiSH.pdf (1.52 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04900898 , version 1 (20-01-2025)

Licence

Identifiants

  • HAL Id : hal-04900898 , version 1

Citer

Quentin Tallon, Juan Martinez, Eric Gregoire, Pascale Fernandez, Delphine Dugue, et al.. Diffusion model uniform manifold filtering for classification of small datasets with underrepresented classes: Application to chromosomal aberration microscopy detection. International Conference on Machine Vision, Oct 2024, Edinburg, United Kingdom. pp.14. ⟨hal-04900898⟩
0 Consultations
0 Téléchargements

Partager

More