Reset filters

Search publications


Search by keyword
List by department / centre / faculty

No publications found.

 

CosSIF: Cosine similarity-based image filtering to overcome low inter-class variation in synthetic medical image datasets

Authors: Islam MZunair HMohammed N


Affiliations

1 Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka, Bangladesh. Electronic address: mominul.islam05@northsouth.edu.
2 Concordia Institute for Information Systems Engineering, Concordia University, Montreal, QC, Canada. Electronic address: md.zunair@mail.concordia.ca.
3 Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka, Bangladesh. Electronic address: nabeel.mohammed@northsouth.edu.

Description

Crafting effective deep learning models for medical image analysis is a complex task, particularly in cases where the medical image dataset lacks significant inter-class variation. This challenge is further aggravated when employing such datasets to generate synthetic images using generative adversarial networks (GANs), as the output of GANs heavily relies on the input data. In this research, we propose a novel filtering algorithm called Cosine Similarity-based Image Filtering (CosSIF). We leverage CosSIF to develop two distinct filtering methods: Filtering Before GAN Training (FBGT) and Filtering After GAN Training (FAGT). FBGT involves the removal of real images that exhibit similarities to images of other classes before utilizing them as the training dataset for a GAN. On the other hand, FAGT focuses on eliminating synthetic images with less discriminative features compared to real images used for training the GAN. The experimental results reveal that the utilization of either the FAGT or FBGT method reduces low inter-class variation in clinical image classification datasets and enables GANs to generate synthetic images with greater discriminative features. Moreover, modern transformer and convolutional-based models, trained with datasets that utilize these filtering methods, lead to less bias toward the majority class, more accurate predictions of samples in the minority class, and overall better generalization capabilities. Code and implementation details are available at: https://github.com/mominul-ssv/cossif.


Keywords: ConvNeXtCosine similarityGenerative adversarial networksMedical image datasetsSkin lesion classificationSwin transformerVision transformer


Links

PubMed: https://pubmed.ncbi.nlm.nih.gov/38492455/

DOI: 10.1016/j.compbiomed.2024.108317