arXiv reaDer
ドメイン一般化された医療画像セグメンテーションのための解きほぐしを伴う半教師ありメタ学習
Semi-supervised Meta-learning with Disentanglement for Domain-generalised Medical Image Segmentation
ディープモデルを新しいセンター(ここではドメインと呼ばれる)からの新しいデータに一般化することは、依然として課題です。これは主に、ソースドメインと非表示ドメイン間のデータ統計のシフト(ドメインシフト)に起因します。最近、トレーニングデータをメタトレーニングセットとメタテストセットに分割してトレーニング中のドメインシフトをシミュレートおよび処理する勾配ベースのメタ学習アプローチにより、一般化のパフォーマンスが向上することが示されています。ただし、現在の完全に監視されたメタ学習アプローチは、ピクセル単位の注釈を作成するために多大な労力が必要な医療画像セグメンテーションに対してスケーラブルではありません。一方、低データ体制では、シミュレートされたドメインシフトは、ソースドメインと非表示ドメイン間で実際のドメインシフトを適切に近似しない場合があります。この問題に対処するために、我々は解きほぐしを伴う新しい半教師ありメタ学習フレームワークを提案します。ドメインシフトに関連する表現を明示的にモデル化します。表現を解きほぐし、それらを組み合わせて入力画像を再構築することで、ラベルのないデータを使用して、メタ学習の真のドメインシフトをより適切に近似できます。したがって、特にラベル付けされたデータの量が限られている場合、モデルはより優れた一般化パフォーマンスを実現できます。実験は、提案された方法が異なるセグメンテーションタスクで堅牢であり、2つの公開ベンチマークで最先端の一般化パフォーマンスを達成することを示しています。
Generalising deep models to new data from new centres (termed here domains) remains a challenge. This is largely attributed to shifts in data statistics (domain shifts) between source and unseen domains. Recently, gradient-based meta-learning approaches where the training data are split into meta-train and meta-test sets to simulate and handle the domain shifts during training have shown improved generalisation performance. However, the current fully supervised meta-learning approaches are not scalable for medical image segmentation, where large effort is required to create pixel-wise annotations. Meanwhile, in a low data regime, the simulated domain shifts may not approximate the true domain shifts well across source and unseen domains. To address this problem, we propose a novel semi-supervised meta-learning framework with disentanglement. We explicitly model the representations related to domain shifts. Disentangling the representations and combining them to reconstruct the input image allows unlabeled data to be used to better approximate the true domain shifts for meta-learning. Hence, the model can achieve better generalisation performance, especially when there is a limited amount of labeled data. Experiments show that the proposed method is robust on different segmentation tasks and achieves state-of-the-art generalisation performance on two public benchmarks.
updated: Fri Oct 01 2021 19:30:50 GMT+0000 (UTC)
published: Thu Jun 24 2021 19:50:07 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アソシエイト