内視鏡的メイヨースコアおよび潰瘍性大腸炎内視鏡的重症度指数は、潰瘍性大腸炎の内視鏡的重症度を評価するために一般的に使用されるスコアリングシステムです。それらは、病気の活動に関連するスコアの割り当てに基づいており、レベル間のランクが作成され、通常の回帰問題になります。一方、ほとんどの研究では、深層学習モデルをトレーニングするために、通常の回帰問題には最適ではないカテゴリ別のクロスエントロピー損失関数を使用しています。本研究では、クラスの順序を尊重し、コストの計算にクラスの距離を考慮に入れる、クラス距離加重クロスエントロピー(CDW-CE)と呼ばれる新しい損失関数を提案します。実験的評価は、CDW-CEが、通常の回帰問題のために設計された従来のカテゴリークロスエントロピーおよびCORNフレームワークよりも優れていることを示しています。さらに、CDW-CEは出力層での変更を必要とせず、クラスアクティベーションマップの視覚化手法と互換性があります。
Endoscopic Mayo score and Ulcerative Colitis Endoscopic Index of Severity are commonly used scoring systems for the assessment of endoscopic severity of ulcerative colitis. They are based on assigning a score in relation to the disease activity, which creates a rank among the levels, making it an ordinal regression problem. On the other hand, most studies use categorical cross-entropy loss function, which is not optimal for the ordinal regression problem, to train the deep learning models. In this study, we propose a novel loss function called class distance weighted cross-entropy (CDW-CE) that respects the order of the classes and takes the distance of the classes into account in calculation of cost. Experimental evaluations show that CDW-CE outperforms the conventional categorical cross-entropy and CORN framework, which is designed for the ordinal regression problems. In addition, CDW-CE does not require any modifications at the output layer and is compatible with the class activation map visualization techniques.