表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

医用画像の予測の不一致は何を示すか――学習データの与え方を変えた5モデルの比較

Deep Ensembles and Cross-Validation Ensembles

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理

概要

  • 5分割CVのモデル間不一致には、各モデルが見た学習画像の違いによる影響も含まれる。
  • 著者らの3データセット比較では、DEの分割精度はCVと同程度で、較正誤差ACEは各データセットで低かった。
  • 注釈者間の違いとの対応は一部の指標でCVが良く、用途に応じて方式を選ぶ必要がある。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、Deep Ensembles and Cross-Validation Ensemblesの仕組み

左から右へ読む
  1. 01評価用医用画像
  2. 02各モデルの確率予測
  3. 03確率平均と領域決定
  4. 04領域図と症例別不一致

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

医用画像の領域分割では、複数モデルの予測が食い違う度合いを、予測の不確かさの手掛かりに使う。ただし、モデルごとに学習した画像が違えば、一部の画像を見ていない影響も食い違いに含まれる。

この値で人が確認する症例を選ぶなら、何に由来する不一致かが重要になる。著者らは、交差検証で作ったモデル群と、同じ画像で独立に学習したモデル群を区別して調べた。

手法

まず、領域ラベル付き画像から5モデルを学習する。CVは画像を5組に分け、各モデルが1組を除いて学習する。DEは5モデルが同じ画像で学習し、初期値や学習中のデータの並びなどを変える。例えばCTの各位置について、各モデルは対象領域に属する確率を出す。

推論時は位置ごとに5モデルの確率を平均し、最も確率の高い領域を最終予測にする。症例の不一致度は、モデル2つずつの領域図が重ならない度合いから求める。こうして得た値を症例の確認順などに使い、予測の確率が実際の正しさに見合うかも評価する。

新規性

著者らは、5モデルの予測をまとめる場合でも、各モデルに与える学習画像の違いが不一致の解釈を変えると示した。同じ画像で独立に学習するDEと、異なる部分集合で学習するCVを、ネットワークなどの条件をそろえて比較している。

加えて、CVをDEと呼ぶ先行研究の用語と実装の食い違いを調査した。nnU-Netの標準的な流れでDEを学習できる小規模な変更も提示している。

従来手法との違い

比較基準は、nnU-Netで標準的に使われる5分割交差検証のアンサンブル(CV)である。学習画像を5組に分け、各モデルは1組を除く80%で学習するため、モデルごとに見た画像が異なる。

ディープアンサンブル(DE)では、5モデルすべてが同じ学習画像を使い、初期値や学習中の偶然性を変える。著者らはネットワーク構造、学習条件、推論条件をそろえ、学習画像の与え方による差を調べた。

実験結果

著者らは複数人が注釈したGoldAtlasのMRI 19例、CURVASのCT 89例、RIGAの眼底画像749例を用いた。CURVASでは注釈の位置情報が合わない1例を除外した。画像単位で学習・検証を分け、領域分割の精度、確率の較正、注釈者間の違いとの対応、失敗症例の検出、分布変化下の性能を調べた。

表の値は100倍表示で、著者らが報告した較正誤差ACEはCV対DEで順にGoldAtlasが19.3対16.7、CURVASが19.6対18.3、RIGAが19.9対17.9だった。分割精度DSCは84.6対85.2、93.5対93.6、93.1対93.2。注釈者間の違いとの対応はCURVASとRIGAの一部指標でCVが良く、症例単位の失敗検出指標AURCは3データセットともDEの値が低かった。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:施設内のCT画像で領域分割を行い、5モデルの予測の不一致が大きい症例から担当者に確認してもらう。入力はCT画像、出力は領域図と症例ごとの不一致度を付けた確認一覧とし、担当者が原画像と予測を照合する。

対象領域のラベルが付いたCT画像を学習に使い、撮影装置や施設の違いを記録する。対象施設の画像に合わせてモデルを学習し、確認順が誤りの発見に役立つかを別の症例で確かめる。改善は検証前の仮説である。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:対象施設のCT画像と対象領域のラベルを集め、画像単位で学習用と評価用に分ける。同じ画像に複数人のラベルがあれば一緒に管理し、両側にまたがらないようにする。撮影装置や施設の違いも記録する。

5分割CVを比較基準とし、同じ学習画像群で5モデルのDEも学習する。対象領域に合わせて設定したうえで、評価用画像のDSC、ACE、不一致度で並べた際の失敗症例の取りこぼしを測る。別装置や別施設の画像があれば分けて確認する。

限界と課題

著者らは、通常の交差検証で得るモデルに加えてDEを学習する費用が生じると述べる。また、設定した分布変化の下では両方式の差は概して小さく、調べた全データセットで一方が常に優位とは報告していない。

編集上の検証事項は、対象施設の症例構成や撮影条件でも、不一致度による確認順が実際の誤り発見に役立つかである。注釈の曖昧さとの対応を調べるには、同一画像に対する複数人の領域ラベルが必要になる。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

Lost in the Folds: When Cross-Validation Is Not a Deep Ensemble for Uncertainty Estimation

Ensemble disagreement is widely used as a proxy for epistemic uncertainty in medical image segmentation. In practice, many studies form ensembles via K-fold cross-validation (CV), yet refer to them as ``deep ensembles'' (DE). Because CV members are trained on different data subsets, their disagreement mixes seed-driven variability with data-exposure effects, which can change how uncertainty should be interpreted. We audit recent segmentation uncertainty studies and find that terminology--implementation mismatches are common. We then compare a standard 5-fold CV ensemble to a 5-member DE (fixed training set, different random seeds) under otherwise identical configurations on three multi-rater segmentation datasets spanning three modalities. We evaluate uncertainty for calibration, failure detection, ambiguity modeling, and robustness under distribution shift. DE match segmentation accuracy while improving calibration and failure detection, whereas CV ensembles sometimes correlate more strongly with inter-rater variability on the studied datasets. Thus, ensemble construction should be chosen to match the research question: DE for reliability-oriented use (e.g., selective referral/failure detection) and CV ensembles as a proxy for ambiguity. We provide a lightweight nnU-Net modification enabling DE training within the default pipeline.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01