Interpretation of the Outputs of Deep Learning Model Trained with Skin Cancer Dataset
2018.06.02 17:48
Our manuscript, "Interpretation of the Outputs of Deep Learning Model Trained with Skin Cancer Dataset" was published as a letter article in the Journal of Investigative Dermatology today (https://www.jidonline.org/article/S0022-202X(18)31992-4/fulltext).
When we train a CNN model, we somtimes get a disappointing Top-1 accuracy. I also suffered this problem and I did not understand exactly what was wrong at that time. When my early version of the 12DX paper was reviewed in JAMA dermatology 2 years ago, the biggest reason for rejection was the low Top-1 accuracy.
However, unlike general object recognition studies, it is very difficult to determine medical research results with Top-1 accuracy, and it is important that the AUC can be high even with a low Top-1 accuracy. If you look carefully, most of medical AI researches have used AUC rather than Top-(n) accuracy.
Because of small and imbalanced training data in medical researches, the analysis of each class as Top-(n) accuracy is inadequate (but the mean Top-(n) of all classes is meaningful). Top-(n) accuracy of each classes vary whenever we repeat the training of CNN with imbalanced dataset. Therefore, we should see the corrected value while using thresholds of each classes, that is ROC curve.
With the AUC results, we published "Classification of the Clinical Images for Benign and Malignant Cutaneous Tumors Using a Deep Learning Algorithm" (https://www.jidonline.org/article/S0022-202X(18)30111-8/fulltext)
There was a debate that my 12DX algorithm is not sensitive (low top-1 accuracy) with the ISIC dataset (Automated Dermatological Diagnosis: Hype or Reality?; https://www.jidonline.org/article/S0022-202X(18)31991-2/fulltext).
There was an additional problem as well as the Top accuracy problem.
When we analyze a clinical image, "the problem of judging whether it is melanoma or not" is easier than "the problem of matching the type of cancer".
Analyzing the output of the AI (CNN) model is equivalent to "the problem of matching the type of cancer", and analyzing the ratio of output is proper if we want to analyze the problem of judging "whether cancer or not".
We interpreted the ratio of melanoma output and nevus output rather than using melanoma output alone.
RATIO (Melanoma Index) = melanoma output / (melanoma output + nevus output).
The clinical image of skin cancer consists of a nodular lesion and a background. If you want to concentrate on only the lesion, we need to analyze it with RATIO as above to get more accurate results.
In the attached photograph, (b) is "matching what cancer is" and (a) is judging "whether it is cancer or not".
We made web-DEMO (http://dx.medicalphoto.org), and we have made it possible to show what conclusions are coming up depending on the Top-5 output and how it is interpreted.
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
1549 | ambient-light-sensor | WHRIA | 2018.09.29 | 317 |
1548 | register | WHRIA | 2018.09.21 | 1996 |
1547 | Deep learning 기반 DEMO | WHRIA | 2018.09.16 | 7134 |
1546 | slide note 삭제 슬라이드 노트 삭제 | WHRIA | 2018.09.12 | 4481 |
1545 | faster rcnn resnet 101 | WHRIA | 2018.09.09 | 193 |
1544 | 나이 40에... | WHRIA | 2018.09.01 | 156 |
1543 | letencrypt - win | WHRIA | 2018.08.12 | 67 |
1542 | 달러 환율 | WHRIA | 2018.06.16 | 1341 |
» | Interpretation of the Outputs of Deep Learning Model Trained with Skin Cancer Dataset [1] | WHRIA | 2018.06.02 | 7802 |
1540 | skin cancer [1] | WHRIA | 2018.05.23 | 1488 |
1539 | deep learning framework | WHRIA | 2018.05.17 | 1803 |
1538 | mxnet [2] | WHRIA | 2018.05.17 | 3234 |
1537 | intel 내장 그래픽 / CUDA | WHRIA | 2018.05.12 | 3246 |
1536 | 당좌대월이자율 | WHRIA | 2018.03.06 | 2616 |
1535 | security 0 | WHRIA | 2018.03.02 | 2522 |
https://i.imgur.com/jnZUavi.png