アーカイブ / Computer Vision
少数の医用画像に合わせる学習で、役立つ補助データを順に選ぶASAP
ASAP
ASAPの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1 : Illustration of our active and sequential domain adaptation (ASAP) framework. The agent defines the policy π \pi that determines which arm to pull. The environment includes the auxiliary data pool 𝒟 𝒜 \mathcal{D}_{\mathcal{A}} , the target dataset 𝒟 𝒯 \mathcal{D}_{\mathcal{T}} , and the model f θ f_{\theta} . At each turn t t , ASAP executes the four shown steps.
概要
- ASAPは学習中に補助データ群を順に選び、少数の正解付き医用画像へ分割モデルを適応させる。
- MRI二課題とCT二課題を、目標例3件・5分割交差検証の条件で評価した。
- WMHのMambaUNetではDice値71.06％で、Direct FTの62.35％を上回った。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、ASAPの仕組み
- 01少数の正解付き画像
- 02補助データを選択
- 03共同でモデル更新
- 04領域の予測
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
ASAPは、少数の正解付き医用画像しかない分割課題に、学習済みモデルを合わせる方法である。分割とは、画像の画素ごとに病変や臓器の領域を割り当てる処理を指す。正解領域を専門家が描くには手間がかかる。
別の画像データを補助に使えても、目標と合わないデータを混ぜると性能を損ね得る。どの補助データをいつ使うかを、学習中の情報から決めることが実務上の焦点になる。
手法
入力は事前学習済みの分割モデル、少数の正解付き目標画像、複数の補助データ群である。ASAPは補助データ群を選択肢として扱い、各回に一群を選んで目標画像と一緒に学習する。例えば白質高信号域のMRIを目標にした場合、候補群からその回に使う画像群を選び、モデルの重みを更新する。
選択には、各群での損失と、目標課題と補助課題の勾配の向きを使う。損失は予測の誤り、勾配は誤りを減らすため重みを動かす方向を表す。多腕バンディットと呼ぶ方式で、これまで有効だった群の利用と、まだ十分試していない群の探索を両立させる。最終出力は目標画像の領域を予測する調整済みモデルである。
新規性
ASAPの特徴は、学習前に補助データを一つに決めず、学習の各段階で使うデータ群を選び直す点にある。損失と勾配という学習中に得られる情報を報酬に使い、追加の評価モデルに頼らず選択する。
著者らは、元の事前学習データにアクセスしない条件で、目標画像と補助データを一回の継続学習の中で利用した。複数段階の再学習を前提とする方法との差は、選択を学習過程に組み込んだところにある。
従来手法との違い
比較されたのは、目標データだけで継続学習するDirect FT、学習前に補助データを選ぶGMS、全補助データを同時に使うMTL、学習中に補助データを選ぶDALである。ASAPはDALと同じく選択を更新するが、目標との勾配の向きに加え、補助データ上の損失も判断に用いる。
著者らの表では、MRIとCTの各目標課題でASAPのDice値が比較手法を上回った。ただし、表の一部では別の評価指標でASAPが常に最高というわけではなく、用途に応じて指標を分けて読む必要がある。
実験結果
著者らはMRIでPVLとWMH、CTで血管と肝臓の分割を評価した。各課題は正解付き目標例を3件使い、5分割交差検証を行った。UNet、SwinUNet、MambaUNetで、領域の重なりを測るDice値と平均IoUを報告している。
例えばWMHのMambaUNetではASAPのDice値は71.06％、Direct FTは62.35％、DALは63.62％だった。肝臓のMambaUNetではASAPが92.91％、DALが87.71％。著者らはMambaUNetの肝臓適応時間をASAPで15.28時間、MTLで75.03時間と報告した。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：病院内で少数の正解マスクがあるMRIの白質高信号域の輪郭抽出を試す。入力は既存の学習済み分割モデル、専門家が輪郭を描いた少数の目標画像、利用可能な補助画像群である。
ASAPで補助データを順に選びながらモデルを調整し、新しいMRI画像から領域の輪郭を出す。出力は専門家の確認に回し、修正後の輪郭を保存する。研究の結果をそのまま診断性能とみなさず、まず輪郭作成の支援として扱う。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：一つの目標領域を選び、専門家が輪郭を確認した少数例と、別に評価用画像を用意する。同じ事前学習済みモデルから、目標例だけで調整するDirect FTとASAPを始め、評価画像のDice値と平均IoUを比べる。
平均値に加え、症例ごとの低い値、輪郭の修正量、調整に要した時間を記録する。補助データ群を変えた場合にも同じ測定を行い、改善が特定の群の選び方に依存していないか確認する。
限界と課題
著者らは、目標データが少なく、撮像法や対象領域が変わると適応が難しいという課題を出発点にしている。一方、本文で示された検証はMRIの二課題、CTの二課題と、各課題の少数例を使う条件に限られる。
編集上の検証課題は、施設や撮像条件が変わった画像でも輪郭が保たれるか、補助データ群の構成を変えても選択が安定するかである。小さな病変の見逃しや境界のずれは、平均スコアとは別に専門家が確認したい。
出典
元のタイトル・要旨を確認する（英語）
Adapting Foundation Models for Few-Shot Medical Image Segmentation: Actively and Sequentially
Recent advances in foundation models have brought promising results in computer vision, including medical image segmentation. Fine-tuning foundation models on specific low-resource medical tasks has become a standard practice. However, ensuring reliable and robust model adaptation when the target task has a large domain gap and few annotated samples remains a challenge. Previous few-shot domain adaptation (FSDA) methods seek to bridge the distribution gap between source and target domains by utilizing auxiliary data. The selection and scheduling of auxiliaries are often based on heuristics, which can easily cause negative transfer. In this work, we propose an Active and Sequential domain AdaPtation (ASAP) framework for dynamic auxiliary dataset selection in FSDA. We formulate FSDA as a multi-armed bandit problem and derive an efficient reward function to prioritize training on auxiliary datasets that align closely with the target task, through a single-round fine-tuning. Empirical validation on diverse medical segmentation datasets demonstrates that our method achieves favorable segmentation performance, significantly outperforming the state-of-the-art FSDA methods, achieving an average gain of 27.75% on MRI and 7.52% on CT datasets in Dice score. Code is available at the git repository: https://github.com/techicoco/ASAP.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-09-29