A Wavelet Guided Attention Module for Skin Cancer Classification with Gradient-based Feature Fusion

This is the official implementation of "A Wavelet Guided Attention Module for Skin Cancer Classification with Gradient-based Feature Fusion" (ISBI, 2024)

Overall workflow:

Symmetry-aware Feature Attention Module:

#

How to use

-Fork the repository.
-Download the ham10000 dataset and store each class in a separate folder named "class_name".
-Run the jupyter notebook to train the model and generate results. Make sure to change the paths according to your requirement.

Saved model: https://drive.google.com/drive/folders/1t7PEjDuakjjJIYgG5ziB4yDjE81Dac8r?usp=drive_link

Results

Heat maps

Confusion matrix and Subspace feature representation

Authors 🤓*

Ayush Roy
Sujan Sarkar, Ram Sarkar
Soham Ghosal
Dmitrii Kaplun, Asya Lyanova

Citation 🤔*

Please do cite our paper in case you find it useful for your research.
Citation-

-Link to our paper-