A Dynamic Multi‐Output Convolutional Neural Network for Skin Lesion Classification

Author:

Zhou Yingyue12,Guo Junfei12,Yao Hanmin12ORCID,Zhao Jiaqi12,Li Xiaoxia12,Qin Jiamin3,Liu Shuangli12

Affiliation:

1. School of Information Engineering Southwest University of Science and Technology Mianyang Sichuan China

2. Robot Technology Used for Special Environment Key Laboratory of Sichuan Province Mianyang Sichuan China

3. Sichuan Mianyang 404 Hospital Mianyang Sichuan China

Abstract

ABSTRACTSkin cancer is a pressing global health issue, with high incidence and mortality rates. Convolutional neural network (CNN) models have been proven to be effective in improving performance in skin lesion image classification and reducing the medical burden. However, the inherent class imbalance in training data caused by the difficulty of collecting dermoscopy images leads to categorical overfitting, which still limits the effectiveness of these data‐driven models in recognizing few‐shot categories. To address this challenge, we propose a dynamic multi‐output convolutional neural network (DMO‐CNN) model that incorporates exit nodes into the standard CNN structure and includes feature refinement layers (FRLs) and an adaptive output scheduling (AOS) module. This model improves feature representation ability through multi‐scale sub‐feature maps and reduces the inter‐layer dependencies during backpropagation. The FRLs ensure efficient and low‐loss down‐sampling, while the AOS module utilizes a trainable layer selection mechanism to refocus the model's attention on few‐shot lesion categories. Additionally, a novel correction factor loss is introduced to supervise and promote AOS convergence. Our evaluation of the DMO‐CNN model on the HAM10000 dataset demonstrates its effectiveness in multi‐class skin lesion classification and its superior performance in recognizing few‐shot categories. Despite utilizing a very simple VGG structure as the sole backbone structure, DMO‐CNN achieved impressive performance of 0.885 in BACC and 0.983 in weighted AUC. These results are comparable to those of the ensemble model that won the ISIC 2018 challenge, highlighting the strong potential of DMO‐CNN in dealing with few‐shot skin lesion data.

Publisher

Wiley

同舟云学术

1.学者识别学者识别

2.学术分析学术分析

3.人才评估人才评估

"同舟云学术"是以全球学者为主线,采集、加工和组织学术论文而形成的新型学术文献查询和分析系统,可以对全球学者进行文献检索和人才价值评估。用户可以通过关注某些学科领域的顶尖人物而持续追踪该领域的学科进展和研究前沿。经过近期的数据扩容,当前同舟云学术共收录了国内外主流学术期刊6万余种,收集的期刊论文及会议论文总量共计约1.5亿篇,并以每天添加12000余篇中外论文的速度递增。我们也可以为用户提供个性化、定制化的学者数据。欢迎来电咨询!咨询电话:010-8811{复制后删除}0370

www.globalauthorid.com

TOP

Copyright © 2019-2024 北京同舟云网络信息技术有限公司
京公网安备11010802033243号  京ICP备18003416号-3