arXiv reaDer
深層学習に基づく頭頸部器官セグメンテーションのクラス不均衡問題への取り組み
Tackling the Class Imbalance Problem of Deep Learning Based Head and Neck Organ Segmentation
リスクのある臓器のセグメンテーション(OAR)は、画像誘導放射線療法によるがん治療に必要な前提条件です。したがって、セグメンテーションタスクの自動化は臨床的に非常に重要です。ディープラーニング(DL)ベースの医療画像セグメンテーションは現在最も成功しているアプローチですが、背景クラスの過剰な存在と、頭頸部(HAN)領域で最も深刻な解剖学的に与えられた臓器サイズの違いに悩まされています。 HAN領域固有のクラス不均衡の問題に取り組むために、まず、導入されたクラス不均衡測定に基づいて、現在最もパフォーマンスの高い汎用セグメンテーションフレームワークであるnnU-Netのパッチサイズを最適化し、次に、クラス適応ダイス損失をさらに導入します。非常に不均衡な設定を補正します。パッチサイズと損失関数はどちらもクラスの不均衡に直接影響するパラメーターであり、それらの最適化により、ベースラインと比較してDiceスコアが3%増加し、95%ハウスドルフ距離が22%減少し、最終的に0.8±に達します。単一の単純なニューラルネットワークを使用した7つのHAN器官のセグメンテーションの場合は0.15および3.17±1.7mm。パッチサイズの最適化とクラス適応型ダイス損失はどちらも、現在のDLベースのセグメンテーションアプローチに簡単に統合でき、クラスの不均衡なセグメンテーションタスクのパフォーマンスを向上させることができます。
The segmentation of organs at risk (OAR) is a required precondition for the cancer treatment with image guided radiation therapy. The automation of the segmentation task is therefore of high clinical relevance. Deep Learning (DL) based medical image segmentation is currently the most successful approach, but suffers from the over-presence of the background class and the anatomically given organ size difference, which is most severe in the head and neck (HAN) area. To tackle the HAN area specific class imbalance problem we first optimize the patch-size of the currently best performing general purpose segmentation framework, the nnU-Net, based on the introduced class imbalance measurement, and second, introduce the class adaptive Dice loss to further compensate for the highly imbalanced setting. Both the patch-size and the loss function are parameters with direct influence on the class imbalance and their optimization leads to a 3% increase of the Dice score and 22% reduction of the 95% Hausdorff distance compared to the baseline, finally reaching 0.8±0.15 and 3.17±1.7 mm for the segmentation of seven HAN organs using a single and simple neural network. The patch-size optimization and the class adaptive Dice loss are both simply integrable in current DL based segmentation approaches and allow to increase the performance for class imbalanced segmentation tasks.
updated: Wed Jan 05 2022 14:36:45 GMT+0000 (UTC)
published: Wed Jan 05 2022 14:36:45 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アソシエイト