プロンプトが鍵:基盤モデルでの肺結節セグメンテーション感度と合成プロンプト生成の検討
この論文は、CT(コンピュータ断層撮影)画像で肺結節を正確に切り出す(セグメンテーションする)際に、基盤モデルと呼ばれる大規模なセグメンテーションモデルが提示される「プロンプト」(点や境界ボックス)に非常に敏感であることを調べています。肺結節は小さくて境界があいまいなことが多いため、プロンプトの質や位置が最終的なマスクの良し悪しに大きく影響します。研究では、手作業のプロンプトに頼らず自動で粗いプロンプトを作る方法も提案しています。データはLUNA16(LIDC‑IDRI由来、888件の胸部CT、1,186個の注釈付き結節)を使っています。 研究チームは、Segment Anything(SAM)や医療向けのMedSAM、その3D対応版であるSAM2とMedSAM2といった複数の基盤モデルを比較しました。各モデルに対して、中心点のみ、境界ボックスのみ、点+ボックスの三種類の入力プロンプトで実験を行い、単枚・複数枚・ボリューム(3D)入力それぞれで結果を評価しました。さらに、プロンプトをわずかにずらす実験や、ボックスのサイズを拡大・縮小する実験で、モデルの頑健さ(ロバストネス)も調べています。 提案手法は、基盤モデルの画像エンコーダが作る特徴量を使って軽量なデコーダを学習するものです。このデコーダは、(1)粗い結節マスク、(2)点と境界ボックスの候補、を同時に出力します。生成した合成プロンプトは基盤モデルのプロンプトエンコーダに渡され、そこから得られるマスクの品質も学習時に評価して、最終的に有用なプロンプトが学べるようにしています。学習にはSørensen‑Dice(類似度指標)損失や交差エントロピー、ボックス回帰用のL1や一般化IoU(交差分割と合併の指標)などを使っています。 主要な発見は、境界ボックスのプロンプトが一般に点プロンプトよりも良い結果を出すこと、医療向けに特化したモデルが汎用モデルより性能が高いことです。点プロンプトのみでは多くの構成でDice係数(セグメンテーションの一致度)が0.1未満と低い場合が多かった一方で、MedSAM2は例外的に0.7を超える値を出しました。最高値ではSAM2が境界ボックス入力でDice 0.89を達成し、ほかの3Dモデルでも0.88前後の結果が得られています。提案した合成プロンプト生成を組み合わせた方法は論文の要約でDice 0.85を報告しており、手動プロンプトに頼らない実用的な戦略として有望だとしています。なお、点とボックスを同時に与えると改善しない場合や、むしろ性能が下がる場合があったことも報告されています。 重要な注意点として、基盤モデルはプロンプトに強く依存するため、プロンプトの小さな位置ずれやサイズの変化でマスクが大きく変わるという問題が残ります。また、汎用モデルは医療用CT特有の濃度分布や立体情報に対して事前学習が不十分で、ゼロショット(追加の訓練なし)性能は限られることが示されました。今回の評価はLUNA16データセットに基づくもので、臨床での運用や他のデータセットへの一般化にはさらなる検証が必要です。論文は合成プロンプトが有望だと結論づけていますが、臨床導入には追加の安全性・頑健性評価が求められます。