A reinforcement learning algorithm for automated detection of skin lesions

Usmani, U.A. and Watada, J. and Jaafar, J. and Aziz, I.A. and Roy, A. (2021) A reinforcement learning algorithm for automated detection of skin lesions. Applied Sciences (Switzerland), 11 (20). ISSN 20763417

Full text not available from this repository.
Official URL: https://www.scopus.com/inward/record.uri?eid=2-s2....

Abstract

Skin cancers are increasing at an alarming rate, and detection in the early stages is essential for advanced treatment. The current segmentation methods have limited labeling ability to the ground truth images due to the numerous noisy expert annotations present in the datasets. The precise boundary segmentation is essential to correctly locate and diagnose the various skin lesions. In this work, the lesion segmentation method is proposed as a Markov decision process. It is solved by training an agent to segment the region using a deep reinforcement-learning algorithm. Our method is similar to the delineation of a region of interest by the physicians. The agent follows a set of serial actions for the region delineation, and the action space is defined as a set of continuous action parameters. The segmentation model learns in continuous action space using the deep deterministic policy gradient algorithm. The proposed method enables continuous improvement in performance as we proceed from coarse segmentation results to finer results. Finally, our proposed model is evaluated on the International Skin Imaging Collaboration (ISIC) 2017 image dataset, Human against Machine (HAM10000), and PH2 dataset. On the ISIC 2017 dataset, the algorithm achieves an accuracy of 96.33 for the naevus cases, 95.39 for the melanoma cases, and 94.27 for the seborrheic keratosis cases. The other metrics are evaluated on these datasets and rank higher when compared with the current state-of-the-art lesion segmentation algorithms. © 2021 by the authors. Licensee MDPI, Basel, Switzerland.

Item Type: Article
Additional Information: cited By 10
Depositing User: Mr Ahmad Suhairi UTP
Date Deposited: 10 Nov 2023 03:28
Last Modified: 10 Nov 2023 03:28
URI: https://khub.utp.edu.my/scholars/id/eprint/14362

Actions (login required)

View Item
View Item