metadata

tags:
  - skin-cancer-detection
  - medical
  - image-classification
  - tensorflow
  - keras
license: mit
datasets:
  - HAM10000
metrics:
  - accuracy
model-index:
  - name: Skin Cancer Detection Model
    results:
      - task:
          type: image-classification
          name: Image Classification
        dataset:
          name: HAM10000
          type: HAM10000
        metrics:
          - name: Accuracy
            type: accuracy
            value: 0.73

Skin Cancer Detection Model

This model is built to detect different types of skin cancer from dermatoscopic images. It was trained using the HAM10000 dataset and is designed to classify seven types of skin lesions. The goal of this model is to assist in the early detection of skin cancer, particularly in regions where temperatures are rising, making skin cancer detection increasingly important.

Model Overview

The model is a Convolutional Neural Network (CNN) developed using TensorFlow and Keras. It takes input images of skin lesions and predicts one of the seven skin cancer classes. The model achieved an accuracy of 73%, which can be further improved with fine-tuning and larger datasets.

Classes Detected

The model predicts the following seven types of skin cancer:

akiec: Actinic Keratoses and Intraepithelial Carcinoma (pre-cancerous)
bcc: Basal Cell Carcinoma (a common type of skin cancer)
bkl: Benign Keratosis (non-cancerous lesion)
df: Dermatofibroma (benign skin lesion)
nv: Melanocytic Nevus (a common mole)
vasc: Vascular Lesions (non-cancerous lesions of blood vessels)
mel: Melanoma (most dangerous type of skin cancer)

Explanation of Each Class:

Actinic Keratoses (akiec): A rough, scaly patch on the skin that develops from years of sun exposure. It is considered pre-cancerous and can lead to squamous cell carcinoma.
Basal Cell Carcinoma (bcc): A type of skin cancer that starts in the basal cells and is often caused by UV radiation. It rarely spreads to other parts of the body.
Benign Keratosis (bkl): Non-cancerous skin growths that resemble moles but do not pose any risk.
Dermatofibroma (df): A common, benign fibrous skin lesion, usually found on the legs.
Melanocytic Nevus (nv): Also known as a mole, these are benign proliferations of melanocytes but can sometimes develop into melanoma.
Vascular Lesions (vasc): Benign growths formed by abnormal blood vessels, generally harmless.
Melanoma (mel): The most serious type of skin cancer, arising from melanocytes. Early detection is critical for effective treatment.

Dataset

The HAM10000 dataset, which contains 10,015 dermatoscopic images of different skin lesions, was used to train this model. The dataset is publicly available and widely used for research on skin lesion classification.

Model Architecture

The model is based on a Convolutional Neural Network (CNN) architecture with multiple layers of convolution, max-pooling, and fully connected layers to classify images. The input images were resized to 224x224 pixels and normalized to ensure uniformity across the dataset.

Key architecture components include:

Convolution layers with ReLU activation
Max Pooling layers for down-sampling
Fully connected dense layers
Softmax activation for the final classification

Training

The model was trained on 80% of the HAM10000 dataset, with 20% used for validation. Data augmentation was applied to the training set to improve generalization, including techniques like rotation, flipping, and scaling.

The training configuration:

Optimizer: Adam
Loss function: Categorical Crossentropy
Batch size: 32
Epochs: 10

Metrics

Accuracy: The primary evaluation metric was accuracy. The model achieved 73% accuracy on the validation set.

Further improvements can be achieved with additional training or fine-tuning using transfer learning methods.

How to Use

You can use this model to classify skin lesion images. Below is an example code snippet to load the model and make predictions:

from tensorflow.keras.models import load_model
from tensorflow.keras.preprocessing import image
import numpy as np

# Load the model from Hugging Face
model = load_model('skin_cancer_model.h5')

# Preprocess the image and make a prediction
img = image.load_img('test_image.jpg', target_size=(224, 224))
img_array = image.img_to_array(img)
img_array = np.expand_dims(img_array, axis=0) / 255.0

predictions = model.predict(img_array)
predicted_class = np.argmax(predictions, axis=1)[0]

# Mapping class index to class name
class_names = ['akiec', 'bcc', 'bkl', 'df', 'nv', 'vasc', 'mel']
print(f"Predicted class: {class_names[predicted_class]}")