Few-shot learning for skin lesion image classification

被引:11
作者
Liu, Xue-Jun [1 ]
Li, Kai-li [1 ]
Luan, Hai-ying [2 ]
Wang, Wen-hui [1 ]
Chen, Zhao-yu [1 ]
机构
[1] Beijing Inst Petrochem Technol, Sch Informat Engn, Beijing 102617, Peoples R China
[2] Beijing Res Inst Automat Machinery Ind Co Ltd, Fluid Power & Automot Equipment Ctr, Beijing 100120, Peoples R China
关键词
Image processing; Small sample learning; Relational network; Metric learning; Convolutional neural network;
D O I
10.1007/s11042-021-11472-0
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The mortality of skin pigmented malignant lesions is very high, especially melanoma. Due to the limitation of marking means, the large-scale annotation data of skin lesions are generally more difficult to obtain. When the deep learning model is trained on a small dataset, its generalization performance is limited. Using prior knowledge to expand small sample data is a general model method of learning classification, which is difficult to deal with complex skin problems. On the basis of a small amount of labeled skin lesion image data, this paper uses the improved Relational Network for measurement learning to realize the classification of skin disease. This method uses relative position network (RPN) and relative mapping network (RMN), in which RPN captures and extracts feature information by attention mechanism, and RMN obtains the similarity of image classification by weighted sum of attention mapping distance. The average accuracy of classification is 85% on the public ISIC melanoma dataset, and the results show the effectiveness and applicability of the method.
引用
收藏
页码:4979 / 4990
页数:12
相关论文
共 25 条
[1]  
[Anonymous], 2018, SIGNAL PROCESS, P145
[2]  
Bertinetto L., 2018, INT C LEARNING REPRE
[3]  
Bossuyt PM, 2015, BMJ-BRIT MED J, V351, DOI [10.1136/bmj.h5527, 10.1373/clinchem.2015.246280, 10.1148/radiol.2015151516]
[4]  
Codella NCF, 2018, I S BIOMED IMAGING, P168, DOI 10.1109/ISBI.2018.8363547
[5]   A Point-of-Care, Real-Time Artificial Intelligence System to Support Clinician of a Wide of Skin Diseases [J].
Dulmage, Brittany ;
Tegtmeyer, Kyle ;
Zhang, Michael Z. ;
Colavincenzo, Maria ;
Xu, Shuai .
JOURNAL OF INVESTIGATIVE DERMATOLOGY, 2020, 141 (05) :1230-1235
[6]   Dermatologist-level classification of skin cancer with deep neural networks [J].
Esteva, Andre ;
Kuprel, Brett ;
Novoa, Roberto A. ;
Ko, Justin ;
Swetter, Susan M. ;
Blau, Helen M. ;
Thrun, Sebastian .
NATURE, 2017, 542 (7639) :115-+
[7]   A better way to monitor haze through image based upon the adjusted LeNet-5 CNN model [J].
Fan, Yonglei ;
Rui, Xiaoping ;
Poslad, Stefan ;
Zhang, Guangyuan ;
Yu, Tian ;
Xu, Xijie ;
Song, Xianfeng .
SIGNAL IMAGE AND VIDEO PROCESSING, 2020, 14 (03) :455-463
[8]  
Jeddi Fatemeh Rangraz, 2016, Acta Inform Med, V24, P30, DOI 10.5455/aim.2016.24.30-33
[9]   DEEP FEATURES TO CLASSIFY SKIN LESIONS [J].
Kawahara, Jeremy ;
BenTaieb, Aicha ;
Hamarneh, Ghassan .
2016 IEEE 13TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2016, :1397-1400
[10]   Meta-Learning based prototype-relation network for few-shot classification [J].
Liu, Xiaoqian ;
Zhou, Fengyu ;
Liu, Jin ;
Jiang, Lianjie .
NEUROCOMPUTING, 2020, 383 :224-234