Deep Learning in Diabetic Foot Ulcers Detection: A Comprehensive Evaluation
arXiv:2010.03341 · doi:10.1016/j.compbiomed.2021.104596
Abstract
There has been a substantial amount of research involving computer methods and technology for the detection and recognition of diabetic foot ulcers (DFUs), but there is a lack of systematic comparisons of state-of-the-art deep learning object detection frameworks applied to this problem. DFUC2020 provided participants with a comprehensive dataset consisting of 2,000 images for training and 2,000 images for testing. This paper summarises the results of DFUC2020 by comparing the deep learning-based algorithms proposed by the winning teams: Faster R-CNN, three variants of Faster R-CNN and an ensemble method; YOLOv3; YOLOv5; EfficientDet; and a new Cascade Attention Network. For each deep learning method, we provide a detailed description of model architecture, parameter settings for training and additional stages including pre-processing, data augmentation and post-processing. We provide a comprehensive evaluation for each method. All the methods required a data augmentation stage to increase the number of images available for training and a post-processing stage to remove false positives. The best performance was obtained from Deformable Convolution, a variant of Faster R-CNN, with a mean average precision (mAP) of 0.6940 and an F1-Score of 0.7434. Finally, we demonstrate that the ensemble method based on different deep learning methods can enhanced the F1-Score but not the mAP.
arXiv comment19 pages, 18 figures, 10 tables
References in corpus (15)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- YOLOv4: Optimal Speed and Accuracy of Object Detection
- MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications
- YOLOv3: An Incremental Improvement
- mixup: Beyond Empirical Risk Minimization
- Gaussian Error Linear Units (GELUs)
- Improved Regularization of Convolutional Neural Networks with Cutout
- Weighted boxes fusion: Ensembling boxes from different object detection models
- DropBlock: A regularization method for convolutional networks
- Cascade R-CNN: Delving into High Quality Object Detection
- DetNet: A Backbone network for Object Detection
- Recognition of Ischaemia and Infection in Diabetic Foot Ulcers: Dataset and Techniques
- Bag of Freebies for Training Object Detection Neural Networks
- DFUC2020: Analysis Towards Diabetic Foot Ulcer Detection
- A Refined Deep Learning Architecture for Diabetic Foot Ulcers Detection