CB-HRNet: A Class-Balanced High-Resolution Network for the evaluation of endoscopic activity in patients with ulcerative colitis

Clin Transl Sci. 2023 Aug;16(8):1421-1430. doi: 10.1111/cts.13542. Epub 2023 May 15.

Abstract

Endoscopic evaluation is the key to the management of ulcerative colitis (UC). However, there is interobserver variability in interpreting endoscopic images among gastroenterologists. Furthermore, it is time-consuming. Convolutional neural networks (CNNs) can help overcome these obstacles and has yielded preliminary positive results. We aimed to develop a new CNN-based algorithm to improve the performance for evaluation tasks of endoscopic images in patients with UC. A total of 12,163 endoscopic images from 308 patients with UC were collected from January 2014 to December 2021. The training set and test set images were randomly divided into 37,515 and 3191 after excluding possible interference and data augmentation. Mayo Endoscopic Subscores (MES) were predicted by different CNN-based models with different loss functions. Their performances were evaluated by several metrics. After comparing the results of different CNN-based models with different loss functions, High-Resolution Network with Class-Balanced Loss achieved the best performances in all MES classification subtasks. It was especially great at determining endoscopic remission in UC, which achieved a high accuracy of 95.07% and good performances in other evaluation metrics with sensitivity 92.87%, specificity 95.41%, kappa coefficient 0.8836, positive predictive value 93.44%, negative predictive value 95.00% and area value under the receiver operating characteristic curve 0.9834, respectively. In conclusion, we proposed a new CNN-based algorithm, Class-Balanced High-Resolution Network (CB-HRNet), to evaluate endoscopic activity of UC with excellent performance. Besides, we made an open-source dataset and it can be a new benchmark in the task of MES classification.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Colitis, Ulcerative* / diagnostic imaging
  • Colonoscopy / methods
  • Humans
  • ROC Curve
  • Severity of Illness Index