Deep Learning on Edge Device for Early Prescreening of Skin Cancers in Rural Communities

Abstract

Skin cancer is a possible curable disease when detected in the early stage, but skin cancer diagnosis is difficult for people in developing countries where residents lack access to proper healthcare. In this paper, we present a low-cost, easy-to-use, and internet-free prescreening solution to detect cancer earlier in rural areas where medical resources are scarce. We deliver a prototype of a device that can classify the skin anomaly into seven major categories and calculate the area segmentation. The prototype we designed includes a Raspberry Pi 3B+, Pi camera, magnifying camera attachment, a convolution neural network powering skin cancer recognition, another network for skin cancer boundary segmentation, and an interactive touchscreen user interface in a custom enclosure. We trained a MobileNetV2 for skin cancer recognition and a U-Net for skin cancer boundary segmentation on the Skin Cancer MNIST dataset collected by The International Skin Imaging Collaboration and used it as our skin cancer recognition model. We then deployed both models onto a Raspberry Pi, and made it into a handy device that takes a close-up picture and prescreen the patient's skin quickly. Through this research process, we found out that it is possible to run heavy computation such as deep learning on a Raspberry Pi and be packaged into a low cost handheld device for screening purposes for a very low cost.