arXiv reaDer
順序ラベルを使用した医用画像分類のためのメタ順序回帰フォレスト
Meta Ordinal Regression Forest for Medical Image Classification with Ordinal Labels
医用画像分類のパフォーマンスは、通常、クロスエントロピー(CE)損失でトレーニングされる深い畳み込みニューラルネットワーク(CNN)によって強化されています。ただし、ラベルが本質的に固有の順序特性を示す場合、たとえば、良性腫瘍から悪性腫瘍への発生の場合、CE損失は、より一般化を可能にするためにそのような順序情報を考慮に入れることができません。順序情報によるモデルの一般化を改善するために、畳み込みニューラルネットワークとメタ学習フレームワークの差分フォレストの組み合わせを通じて順序関係を学習する、順序ラベルを使用した医療画像分類のための新しいメタ順序回帰フォレスト(MORF)メソッドを提案します。提案されたMORFのメリットは、ツリー単位の重み付けネット(TWW-Net)とグループ化された特徴選択(GFS)モジュールの2つのコンポーネントにあります。まず、TWW-Netは、フォレスト内の各ツリーに、対応するツリーの分類損失からマッピングされた特定の重みを割り当てます。したがって、すべてのツリーにはさまざまな重みがあり、ツリーごとの予測分散を緩和するのに役立ちます。次に、GFSモジュールは、以前に使用されていた固定フォレストではなく動的フォレストを有効にし、ランダムな機能の摂動を可能にします。トレーニング中は、ヘッセ行列を計算することで、メタ学習フレームワークのCNNバックボーンとTWW-Netのパラメーターを最適化します。順序ラベル付きの2つの医用画像分類データセット、つまりLIDC-IDRIと乳房超音波データセットの実験結果は、既存の最先端の方法よりも優れたMORF方法のパフォーマンスを示しています。
The performance of medical image classification has been enhanced by deep convolutional neural networks (CNNs), which are typically trained with cross-entropy (CE) loss. However, when the label presents an intrinsic ordinal property in nature, e.g., the development from benign to malignant tumor, CE loss cannot take into account such ordinal information to allow for better generalization. To improve model generalization with ordinal information, we propose a novel meta ordinal regression forest (MORF) method for medical image classification with ordinal labels, which learns the ordinal relationship through the combination of convolutional neural network and differential forest in a meta-learning framework. The merits of the proposed MORF come from the following two components: a tree-wise weighting net (TWW-Net) and a grouped feature selection (GFS) module. First, the TWW-Net assigns each tree in the forest with a specific weight that is mapped from the classification loss of the corresponding tree. Hence, all the trees possess varying weights, which is helpful for alleviating the tree-wise prediction variance. Second, the GFS module enables a dynamic forest rather than a fixed one that was previously used, allowing for random feature perturbation. During training, we alternatively optimize the parameters of the CNN backbone and TWW-Net in the meta-learning framework through calculating the Hessian matrix. Experimental results on two medical image classification datasets with ordinal labels, i.e., LIDC-IDRI and Breast Ultrasound Dataset, demonstrate the superior performances of our MORF method over existing state-of-the-art methods.
updated: Tue Mar 15 2022 08:43:57 GMT+0000 (UTC)
published: Tue Mar 15 2022 08:43:57 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アソシエイト