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

Skin cancer is a highly dangerous type of cancer that requires an accurate diagnosis from experienced physicians. To help physicians diagnose skin cancer more efficiently, a computer-aided diagnosis (CAD) system can be very helpful. In this paper, we propose a novel model, which uses a novel attenti...

Full description

Saved in:
Bibliographic Details
Main Authors: Roy, Ayush, Sarkar, Sujan, Ghosal, Sohom, Kaplun, Dmitrii, Lyanova, Asya, Sarkar, Ram
Format: Conference Proceeding
Language:English
Subjects:
Biological system modeling
Feature extraction
Feature fusion
Image color analysis
Medical imaging
Medical services
Melanoma
Skin
Skin cancer
Soft Attention
Solid modeling
Wavelet transform
Online Access:Request full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Skin cancer is a highly dangerous type of cancer that requires an accurate diagnosis from experienced physicians. To help physicians diagnose skin cancer more efficiently, a computer-aided diagnosis (CAD) system can be very helpful. In this paper, we propose a novel model, which uses a novel attention mechanism to pinpoint the differences in features across the spatial dimensions and symmetry of the lesion, thereby focusing on the dissimilarities of various classes based on symmetry, uniformity in texture and color, etc. Additionally, to take into account the variations in the boundaries of the lesions for different classes, we employ a gradient-based fusion of wavelet and soft attention-aided features to extract boundary information of skin lesions. We have tested our model on the multi-class and highly class-imbalanced dataset, called HAM10000, and achieved promising results, with a 91.17% F1-score and 90.75% accuracy. The code is made available at: https://github.com/AyushRoy2001/WAGF-Fusion.
ISSN:1945-8452
DOI:10.1109/ISBI56570.2024.10635704