arXiv reaDer
潰瘍性大腸炎の重症度推定のためのクラス距離加重クロスエントロピー損失
Class Distance Weighted Cross-Entropy Loss for Ulcerative Colitis Severity Estimation
メイヨー内視鏡スコアや潰瘍性大腸炎内視鏡インデックスの重症度など、潰瘍性大腸炎の内視鏡活動を測定するために使用されるスコアリングシステムでは、レベルは疾患活動の重症度とともに増加します。スコア間のそのような相対的なランク付けは、それを通常の回帰問題にします。一方、ほとんどの研究では、カテゴリ別のクロスエントロピー損失関数を使用して深層学習モデルをトレーニングしますが、これは通常の回帰問題には最適ではありません。この研究では、クラスの順序を尊重し、コストの計算でクラスの距離を考慮に入れる、新しい損失関数であるクラス距離加重クロスエントロピー(CDW-CE)を提案します。実験的評価は、CDW-CEでトレーニングされたモデルが、従来のカテゴリクロスエントロピーおよび通常の回帰問題用に設計された他の一般的に使用される損失関数でトレーニングされたモデルよりも優れていることを示しています。さらに、CDW-CE損失でトレーニングされたモデルのクラスアクティベーションマップは、クラス識別性が高く、ドメインの専門家によってより合理的であることがわかります。
In scoring systems used to measure the endoscopic activity of ulcerative colitis, such as Mayo endoscopic score or Ulcerative Colitis Endoscopic Index Severity, levels increase with severity of the disease activity. Such relative ranking among the scores makes it an ordinal regression problem. On the other hand, most studies use categorical cross-entropy loss function to train deep learning models, which is not optimal for the ordinal regression problem. In this study, we propose a novel loss function, 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 the cost. Experimental evaluations show that models trained with CDW-CE outperform the models trained with conventional categorical cross-entropy and other commonly used loss functions which are designed for the ordinal regression problems. In addition, the class activation maps of models trained with CDW-CE loss are more class-discriminative and they are found to be more reasonable by the domain experts.
updated: Sun Jun 12 2022 09:51:33 GMT+0000 (UTC)
published: Wed Feb 09 2022 18:47:50 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト