Can Score-Based Generative Modeling Effectively Handle Medical Image Classification?

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sarker, Sushmita, Sarker, Prithul, Bebis, George, Tavakkoli, Alireza
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916628906639360
author Sarker, Sushmita
Sarker, Prithul
Bebis, George
Tavakkoli, Alireza
author_facet Sarker, Sushmita
Sarker, Prithul
Bebis, George
Tavakkoli, Alireza
contents The remarkable success of deep learning in recent years has prompted applications in medical image classification and diagnosis tasks. While classification models have demonstrated robustness in classifying simpler datasets like MNIST or natural images such as ImageNet, this resilience is not consistently observed in complex medical image datasets where data is more scarce and lacks diversity. Moreover, previous findings on natural image datasets have indicated a potential trade-off between data likelihood and classification accuracy. In this study, we explore the use of score-based generative models as classifiers for medical images, specifically mammographic images. Our findings suggest that our proposed generative classifier model not only achieves superior classification results on CBIS-DDSM, INbreast and Vin-Dr Mammo datasets, but also introduces a novel approach to image classification in a broader context. Our code is publicly available at https://github.com/sushmitasarker/sgc_for_medical_image_classification
format Preprint
id arxiv_https___arxiv_org_abs_2502_17727
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Can Score-Based Generative Modeling Effectively Handle Medical Image Classification?
Sarker, Sushmita
Sarker, Prithul
Bebis, George
Tavakkoli, Alireza
Computer Vision and Pattern Recognition
The remarkable success of deep learning in recent years has prompted applications in medical image classification and diagnosis tasks. While classification models have demonstrated robustness in classifying simpler datasets like MNIST or natural images such as ImageNet, this resilience is not consistently observed in complex medical image datasets where data is more scarce and lacks diversity. Moreover, previous findings on natural image datasets have indicated a potential trade-off between data likelihood and classification accuracy. In this study, we explore the use of score-based generative models as classifiers for medical images, specifically mammographic images. Our findings suggest that our proposed generative classifier model not only achieves superior classification results on CBIS-DDSM, INbreast and Vin-Dr Mammo datasets, but also introduces a novel approach to image classification in a broader context. Our code is publicly available at https://github.com/sushmitasarker/sgc_for_medical_image_classification
title Can Score-Based Generative Modeling Effectively Handle Medical Image Classification?
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.17727