arXiv reaDer
UniSeg: プロンプト駆動のユニバーサル セグメンテーション モデルと強力な表現学習器
UniSeg: A Prompt-driven Universal Segmentation Model as well as A Strong Representation Learner
ユニバーサル モデルは、医用画像セグメンテーションの有望なトレンドとして浮上しており、医用画像大規模モデル (MILM) を構築する道を切り開いています。ユニバーサル モデルを構築するための一般的な戦略の 1 つは、各タスクをワンホット ベクトルとしてエンコードし、デコーダーの最後に動的畳み込みレイヤーを生成して、関心のあるターゲットを抽出することです。成功しましたが、タスク間の相関関係を無視しており、進行中のタスクをモデルに認識させるには遅すぎます。両方の問題に対処するために、多様なモダリティとドメインを使用したマルチタスク医療画像セグメンテーションのためのプロンプト駆動型ユニバーサル セグメンテーション モデル (UniSeg) を提案します。最初に、すべてのタスク間の相関関係を説明する学習可能なユニバーサル プロンプトを考案し、次に、このプロンプトと画像の特徴をタスク固有のプロンプトに変換します。これは、入力の一部としてデコーダに供給されます。したがって、進行中のタスクを早期にモデルに「認識」させ、デコーダ全体のタスク固有のトレーニングを強化します。私たちの結果は、提案された UniSeg が 11 のアップストリーム タスクで他のユニバーサル モデルおよび単一タスク モデルよりも優れていることを示しています。さらに、UniSeg は 2 つのダウンストリーム データセットで他の事前トレーニング済みモデルよりも優れており、コミュニティに 3D 医療画像セグメンテーション用の高品質の事前トレーニング済みモデルを提供します。コードとモデルは https://github.com/yeerwen/UniSeg で入手できます。
The universal model emerges as a promising trend for medical image segmentation, paving up the way to build medical imaging large model (MILM). One popular strategy to build universal models is to encode each task as a one-hot vector and generate dynamic convolutional layers at the end of the decoder to extract the interested target. Although successful, it ignores the correlations among tasks and meanwhile is too late to make the model 'aware' of the ongoing task. To address both issues, we propose a prompt-driven Universal Segmentation model (UniSeg) for multi-task medical image segmentation using diverse modalities and domains. We first devise a learnable universal prompt to describe the correlations among all tasks and then convert this prompt and image features into a task-specific prompt, which is fed to the decoder as a part of its input. Thus, we make the model 'aware' of the ongoing task early and boost the task-specific training of the whole decoder. Our results indicate that the proposed UniSeg outperforms other universal models and single-task models on 11 upstream tasks. Moreover, UniSeg also beats other pre-trained models on two downstream datasets, providing the community with a high-quality pre-trained model for 3D medical image segmentation. Code and model are available at https://github.com/yeerwen/UniSeg.
updated: Fri Apr 07 2023 06:28:51 GMT+0000 (UTC)
published: Fri Apr 07 2023 06:28:51 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アソシエイト