Effective Melanoma Recognition Using Deep Convolutional Neural Network with Covariance Discriminant Loss.

Journal: Sensors (Basel, Switzerland)
Published Date:

Abstract

Melanoma recognition is challenging due to data imbalance and high intra-class variations and large inter-class similarity. Aiming at the issues, we propose a melanoma recognition method using deep convolutional neural network with covariance discriminant loss in dermoscopy images. Deep convolutional neural network is trained under the joint supervision of cross entropy loss and covariance discriminant loss, rectifying the model outputs and the extracted features simultaneously. Specifically, we design an embedding loss, namely covariance discriminant loss, which takes the first and second distance into account simultaneously for providing more constraints. By constraining the distance between hard samples and minority class center, the deep features of melanoma and non-melanoma can be separated effectively. To mine the hard samples, we also design the corresponding algorithm. Further, we analyze the relationship between the proposed loss and other losses. On the International Symposium on Biomedical Imaging (ISBI) 2018 Skin Lesion Analysis dataset, the two schemes in the proposed method can yield a sensitivity of 0.942 and 0.917, respectively. The comprehensive results have demonstrated the efficacy of the designed embedding loss and the proposed methodology.

Authors

  • Lei Guo
    Shanghai Mental Health Center, Shanghai Jiao Tong University School of Medicine, Shanghai, China.
  • Gang Xie
    Shenzhen Institute of Molecular Crop Design Shenzhen, China.
  • Xinying Xu
    College of Information Engineering, Taiyuan University of Technology, Taiyuan 030024, Shanxi, China. congshen@tju.edu.cn.
  • Jinchang Ren
    Centre for Excellence in Signal and Image Processing, University of Strathclyde, Glasgow G1 1XW, UK.