One-Stage Five-Class Polyp Detection and Classification.

Yu Tian,Leonardo Z. C. T. Pu,Rajvinder Singh,Alastair D. Burt,Gustavo Carneiro
DOI: https://doi.org/10.1109/isbi.2019.8759521
2019-01-01
Abstract:The detection and classification of anatomies from medical images has traditionally been developed in a two-stage process, where the first stage detects the regions of interest (ROIs), while the second stage classifies the detected ROIs. Recent developments from the computer vision community allowed the unification of these two stages into a single detection and classification model that is trained in an end to end fashion. This allows for a simpler and faster training and inference procedures because only one model (instead of the two models needed for the two-stage approach) is required. In this paper, we adapt a recently proposed one stage detection and classification approach for the new 5 class polyp classification problem. We show that this one stage approach is not only competitive in terms of detection and classification accuracy with respect to the two-stage approach, but it is also substantially faster for training and testing. We also show that the one-stage approach produces competitive detection results compared to the state of the art results on the MICCAI 2015 polyp detection challenge.
What problem does this paper attempt to address?