Deep Learning Based Disease Detection Using Domain Specific Transfer Learning

In this paper, we present our approach for the Medico Multimedia Task as part of the MediaEval 2018 Benchmark [13]. Our method is based on convolutional neural networks (CNNs), where we compare how fine-tuning, in the context of transfer learning, from different source domains (general versusmedical domain) affect classification performance. The preliminary results show that fine-tuning models trained on large and diverse datasets is favorable, even when the model’s source domain has little to no resemblance to the new target.

[1]  Michael Riegler,et al.  A Comparison of Deep Learning with Global Features for Gastrointestinal Disease Detection , 2017, MediaEval.

[2]  Michael Riegler,et al.  KVASIR: A Multi-Class Image Dataset for Computer Aided Gastrointestinal Disease Detection , 2017, MMSys.

[3]  Pushparaja Murugan,et al.  Regularization and Optimization strategies in Deep Convolutional Neural Network , 2017, ArXiv.

[4]  Edward Y. Chang,et al.  Transfer representation learning for medical image analysis , 2015, 2015 37th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[5]  Quoc V. Le,et al.  Do Better ImageNet Models Transfer Better? , 2018, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[6]  Klaus Schöffmann,et al.  Cataract-101: video dataset of 101 cataract surgeries , 2018, MMSys.

[7]  Michael Riegler,et al.  Nerthus: A Bowel Preparation Quality Video Dataset , 2017, MMSys.

[8]  Li Fei-Fei,et al.  ImageNet: A large-scale hierarchical image database , 2009, CVPR.

[9]  Klaus Schöffmann,et al.  Lapgyn4: a dataset for 4 automatic content analysis problems in the domain of laparoscopic gynecology , 2018, MMSys.

[10]  Michael Riegler,et al.  Medico Multimedia Task at MediaEval 2018 , 2018, MediaEval.

[11]  Atsuto Maki,et al.  A systematic study of the class imbalance problem in convolutional neural networks , 2017, Neural Networks.

[12]  Yong Man Ro,et al.  Modality-bridge transfer learning for medical image classification , 2017, 2017 10th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics (CISP-BMEI).

[13]  Anders Krogh,et al.  A Simple Weight Decay Can Improve Generalization , 1991, NIPS.

[14]  Geoff S. Nitschke,et al.  Improving Deep Learning with Generic Data Augmentation , 2018, 2018 IEEE Symposium Series on Computational Intelligence (SSCI).

[15]  Luca Antiga,et al.  Automatic differentiation in PyTorch , 2017 .