Adversarial Debiasing techniques towards 'fair' skin lesion classification

Ramon L. Correa-Medero, Bhavik Patel, Imon Banerjee

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Early detection of skin cancer lesions impacts overall patient survival. However, as shown in the literature, people of color have worse prognoses and lower survival rates than people with lighter skin tones. Often, this is the result of delayed or incorrect diagnoses for people of color. Deep learning could provide an effective screening technology with readily available image-capturing techniques, even from a mobile phone. However, often skin complexion biases limit the accuracy of the deep learning models as the applications are mainly trained on data that is predominantly light skin. We propose an adversarial debiasing method with partial learning that produces fairer outcomes for both lighter and darker skin colors. The model unlearns the skin color bias by using an additional classifier to penalize the learning of features specific to skin color. In the partial learning, we added Testing with Concept Activation Vector(TCAV) to select the particular layer where the skin color features are most discernable. We evaluated the performance internally on Fizpatrick17k and externally on ISIC datasets. The debiased model performed equally well for identifying malignant cases on both light and darker skin color. In conclusion, our work finds the adversarial debiasing techniques able to increase the skin lesion model's performance for all the skin color variations without the need for a balanced training dataset and provide a generalization to the external datasets.

Original languageEnglish (US)
Title of host publication11th International IEEE/EMBS Conference on Neural Engineering, NER 2023 - Proceedings
PublisherIEEE Computer Society
ISBN (Electronic)9781665462921
DOIs
StatePublished - 2023
Event11th International IEEE/EMBS Conference on Neural Engineering, NER 2023 - Baltimore, United States
Duration: Apr 25 2023Apr 27 2023

Publication series

NameInternational IEEE/EMBS Conference on Neural Engineering, NER
Volume2023-April
ISSN (Print)1948-3546
ISSN (Electronic)1948-3554

Conference

Conference11th International IEEE/EMBS Conference on Neural Engineering, NER 2023
Country/TerritoryUnited States
CityBaltimore
Period4/25/234/27/23

Keywords

  • Adversarial Training
  • Deep Learning
  • Fairness
  • Melanoma

ASJC Scopus subject areas

  • Artificial Intelligence
  • Mechanical Engineering

Fingerprint

Dive into the research topics of 'Adversarial Debiasing techniques towards 'fair' skin lesion classification'. Together they form a unique fingerprint.

Cite this