arXiv reaDer
疎な小児データセットのための共有表現と対照的な正則化によるマルチタスク、マルチドメインディープセグメンテーション
Multi-Task, Multi-Domain Deep Segmentation with Shared Representations and Contrastive Regularization for Sparse Pediatric Datasets
磁気共鳴(MR)画像の自動セグメンテーションは、臨床診療における小児筋骨格系の形態学的評価に不可欠です。ただし、注釈付きの小児データの量が制限されているため、個々のセグメンテーションモデルの精度と一般化のパフォーマンスは制限されます。したがって、マルチタスクおよびマルチドメイン学習フレームワークで、解剖学的構造のさまざまな部分から生じる複数のデータセットでセグメンテーションモデルをトレーニングすることを提案します。このアプローチにより、より堅牢な共有表現の恩恵を受けながら、小児データの固有の不足を克服することができます。提案されたセグメンテーションネットワークは、共有畳み込みフィルター、それぞれのデータセット統計を計算するドメイン固有のバッチ正規化パラメーター、およびドメイン固有のセグメンテーションレイヤーで構成されます。さらに、教師あり対照正則化が統合され、ドメイン内の類似性を促進し、埋め込みスペースにドメイン間マージンを課すことにより、一般化機能をさらに改善します。骨のセグメンテーションのための足首と肩の関節の2つの小児画像データセットへの貢献を評価します。結果は、提案されたモデルが最先端のアプローチよりも優れていることを示しています。
Automatic segmentation of magnetic resonance (MR) images is crucial for morphological evaluation of the pediatric musculoskeletal system in clinical practice. However, the accuracy and generalization performance of individual segmentation models are limited due to the restricted amount of annotated pediatric data. Hence, we propose to train a segmentation model on multiple datasets, arising from different parts of the anatomy, in a multi-task and multi-domain learning framework. This approach allows to overcome the inherent scarcity of pediatric data while benefiting from a more robust shared representation. The proposed segmentation network comprises shared convolutional filters, domain-specific batch normalization parameters that compute the respective dataset statistics and a domain-specific segmentation layer. Furthermore, a supervised contrastive regularization is integrated to further improve generalization capabilities, by promoting intra-domain similarity and impose inter-domain margins in embedded space. We evaluate our contributions on two pediatric imaging datasets of the ankle and shoulder joints for bone segmentation. Results demonstrate that the proposed model outperforms state-of-the-art approaches.
updated: Fri May 21 2021 12:26:05 GMT+0000 (UTC)
published: Fri May 21 2021 12:26:05 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アソシエイト