Densely Connected Neural Network with Unbalanced Discriminant and Category Sensitive Constraints for Polyp Recognition

Research output: Contribution to journalJournal articleResearchpeer-review

  • Yixuan Yuan
  • Wenjian Qin
  • Ibragimov, Bulat
  • Guanglei Zhang
  • Bin Han
  • Max Q.H. Meng
  • Lei Xing

Automatic polyp recognition in endoscopic images is challenging because of the low contrast between polyps and the surrounding area, the fuzzy and irregular polyp borders, and varying imaging light conditions. In this article, we propose a novel densely connected convolutional network with 'unbalanced discriminant (UD)' loss and 'category sensitive (CS)' loss (DenseNet-UDCS) for the task. We first utilize densely connected convolutional network (DenseNet) as the basic framework to conduct end-to-end polyp recognition task. Then, the proposed dual constraints, UD loss and CS loss, are simultaneously incorporated into the DenseNet model to calculate discriminative and suitable image features. The UD loss in our network effectively captures classification errors from both majority and minority categories to deal with the strong data imbalance of polyp images and normal ones. The CS loss imposes the ratio of intraclass and interclass variations in the deep feature learning process to enable features with large interclass variation and small intraclass compactness. With the joint supervision of UD loss and CS loss, a robust DenseNet-UDCS model is trained to recognize polyps from endoscopic images. The experimental results achieved polyp recognition accuracy of 93.19%, showing that the proposed DenseNet-UDCS can accurately characterize the endoscopic images and recognize polyps from the images. In addition, our DenseNet-UDCS model is superior in detection accuracy in comparison with state-of-the-art polyp recognition methods. Note to Practitioners - Wireless capsule endoscopy (WCE) is a crucial diagnostic tool for polyp detection and therapeutic monitoring, thanks to its noninvasive, user-friendly, and nonpainful properties. A challenge in harnessing the enormous potential of the WCE to benefit the gastrointestinal (GI) patients is that it requires clinicians to analyze a huge number of images (about 50 000 images for each patient). We propose a novel automatic polyp recognition scheme, namely, DenseNet-UDCS model, by addressing practical image unbalanced problem and small interclass variances and large intraclass differences in the data set. The comprehensive experimental results demonstrate superior reliability and robustness of the proposed model compared to the other polyp recognition approaches. Our DenseNet-UDCS model can be further applied in the clinical practice to provide valuable diagnosis information for GI disease recognition and precision medicine.

Original languageEnglish
Article number8842597
JournalIEEE Transactions on Automation Science and Engineering
Volume17
Issue number2
Pages (from-to)574-583
Number of pages10
ISSN1545-5955
DOIs
Publication statusPublished - 2020

    Research areas

  • Category sensitive (CS) loss, densely connected convolutional network (DenseNet), polyp image classification, unbalanced discriminant (UD) loss

ID: 243526540