Transfer learning for endoscopy disease detection and segmentation with mask-RCNN benchmark architecture
Conference paper
Rezvy, S., Zebin, T., Pang, W., Taylor, S. and Gao, X. 2020. Transfer learning for endoscopy disease detection and segmentation with mask-RCNN benchmark architecture. 2nd International Workshop and Challenge on Computer Vision in Endoscopy. Iowa City, United States 03 Apr 2020 pp. 68-72
Type | Conference paper |
---|---|
Title | Transfer learning for endoscopy disease detection and segmentation with mask-RCNN benchmark architecture |
Authors | Rezvy, S., Zebin, T., Pang, W., Taylor, S. and Gao, X. |
Abstract | We proposed and implemented a disease detection and semantic segmentation pipeline using a modified mask-RCNN infrastructure model on the EDD2020 dataset1. On the images provided for the phase-I test dataset, for ’BE’, we achieved an average precision of 51.14%, for ’HGD’ and ’polyp’ it is 50%. However, the detection score for ’suspicious’ and ’cancer’ were low. For phase-I, we achieved a dice coefficient of 0.4562 and an F2 score of 0.4508. We noticed the missed and mis-classification was due to the imbalance between classes. Hence, we applied a selective and balanced augmentation stage in our architecture to provide more accurate detection and segmentation. We observed an increase in detection score to 0.29 on phase-II images after balancing the dataset from our phase-I detection score of 0.24. We achieved an improved semantic segmentation score of 0.62 from our phase-I score of 0.52. |
Keywords | deep learning, computer vision, endoscopy, gastrointestinal |
Conference | 2nd International Workshop and Challenge on Computer Vision in Endoscopy |
Page range | 68-72 |
Proceedings Title | Proceedings of the 2nd International Workshop and Challenge on Computer Vision in Endoscopy |
ISSN | 1613-0073 |
Publication dates | |
25 Apr 2020 | |
Publication process dates | |
Deposited | 05 Oct 2020 |
Accepted | 30 Mar 2020 |
Output status | Published |
Publisher's version | License |
Copyright Statement | © 2020 for the individual papers by the papers' authors. |
Additional information | EndoCV2020 was held in conjunction with the 17th IEEE International Symposium on Biomedical Imaging (ISBI2020). |
Web address (URL) | http://ceur-ws.org/Vol-2595/ |
Scopus EID | 2-s2.0-85084482324 |
Language | English |
https://repository.mdx.ac.uk/item/891q7
Download files
64
total views9
total downloads1
views this month0
downloads this month