arXiv reaDer
脳腫瘍セグメンテーションのためのクロスタスクガイド付き注意を伴うワンパスマルチタスクネットワーク
One-pass Multi-task Networks with Cross-task Guided Attention for Brain Tumor Segmentation
 クラスの不均衡は、医療画像セグメンテーションの主要な課題の1つとして浮上しています。モデルカスケード(MC)戦略は、粗から細へのセグメンテーションのために個々のディープモデルのセットを実行することにより、クラスの不均衡の問題を大幅に軽減します。ただし、その優れたパフォーマンスにもかかわらず、この方法は望ましくないシステムの複雑さを招き、モデル間の相関を無視します。これらの欠陥を処理するために、軽量の深いモデル、つまりワンパスマルチタスクネットワーク(OM-Net)を提案し、ワンパス計算のみを必要とする一方で、MCよりもクラスの不均衡を解決します。まず、OM-Netは、個別のセグメンテーションタスクを1つのディープモデルに統合します。このモデルは、共同機能を学習するための共有パラメーターと、識別機能を学習するためのタスク固有のパラメーターで構成されます。次に、OM-Netをより効果的に最適化するために、タスク間の相関関係を利用して、オンライントレーニングデータ転送戦略とカリキュラム学習ベースのトレーニング戦略の両方を設計します。第三に、予測結果をタスク間で共有することをさらに提案し、カテゴリ固有の統計に基づいてチャネルごとの特徴応答を適応的に再調整できるクロスタスク誘導アテンション(CGA)モジュールを設計します。最後に、セグメンテーション結果を改善するために、シンプルで効果的な後処理方法が導入されます。提案された手法の有効性を実証するために、広範な実験が実施されています。最も印象的なのは、BraTS 2015テストセットとBraTS 2017オンライン検証セットで最先端のパフォーマンスを達成していることです。これらの提案されたアプローチを使用して、64の参加チーム間でBraTS 2018チャレンジの共同3位を獲得しました。コードはhttps://github.com/chenhong-zhou/OM-Netで公開されています。
Class imbalance has emerged as one of the major challenges for medical image segmentation. The model cascade (MC) strategy significantly alleviates the class imbalance issue via running a set of individual deep models for coarse-to-fine segmentation. Despite its outstanding performance, however, this method leads to undesired system complexity and also ignores the correlation among the models. To handle these flaws, we propose a light-weight deep model, i.e., the One-pass Multi-task Network (OM-Net) to solve class imbalance better than MC does, while requiring only one-pass computation. First, OM-Net integrates the separate segmentation tasks into one deep model, which consists of shared parameters to learn joint features, as well as task-specific parameters to learn discriminative features. Second, to more effectively optimize OM-Net, we take advantage of the correlation among tasks to design both an online training data transfer strategy and a curriculum learning-based training strategy. Third, we further propose sharing prediction results between tasks and design a cross-task guided attention (CGA) module which can adaptively recalibrate channel-wise feature responses based on the category-specific statistics. Finally, a simple yet effective post-processing method is introduced to refine the segmentation results. Extensive experiments are conducted to demonstrate the effectiveness of the proposed techniques. Most impressively, we achieve state-of-the-art performance on the BraTS 2015 testing set and BraTS 2017 online validation set. Using these proposed approaches, we also won joint third place in the BraTS 2018 challenge among 64 participating teams. The code is publicly available at https://github.com/chenhong-zhou/OM-Net.
updated: Sat Feb 08 2020 03:34:20 GMT+0000 (UTC)
published: Wed Jun 05 2019 02:50:08 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アソシエイト