アーカイブ / Computer Vision / Generative AI
患者の形を保ち撮影条件を変えたMRI画像で、組織分割を検証する
Counterfactual MRI Data Augmentation
Counterfactual MRI Data Augmentationの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1: Comparison between the ground truth (True) and DL breast segmentation models trained without data augmentation (Pred.) and with data augmentation using cDDGM (Pred.Aug.), in out-of-distribution settings. (A) results of models trained in GE when applied to Siemens MRIs. (B) results of models trained in Siemens when applied to GE MRIs. Blue - Fat mask; Orange - FGT mask.
概要
- 撮影条件を指定した拡散モデルで、元の乳房MRIから条件違いの画像を生成する。
- メーカーをまたぐ二方向の評価で平均Diceは小幅に改善したが、同一メーカー内では低下した条件もある。
- 著者らは装置メーカーの多様性と正解マスク付き画像の少なさを限界に挙げる。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、Counterfactual MRI Data Augmentationの仕組み
- 01元の乳房MRI
- 02撮影条件を指定
- 03ノイズから画像を復元
- 04条件違いのMRI
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
この方法は、乳房の磁気共鳴画像（MRI）から、撮影条件の違いを模した別の画像を作る。MRIの見え方は装置メーカーや撮影設定で変わり、ある条件の画像で学習した組織分割モデルが別条件でも同じ精度を出すとは限らない。組織分割とは、画像の各場所を脂肪、線維腺組織、背景などに割り当てる処理である。
著者らは、患者の形を保ったまま見え方を変える画像を学習に加え、撮影条件をまたぐ評価で役立つか調べた。
手法
入力は元の乳房MRIと、変更先として指定する画像取得パラメータである。画像取得パラメータには装置メーカー、磁場強度、スライス厚などが含まれる。条件付き拡散モデルは、画像にノイズを加えてから少しずつ取り除く過程を学び、指定条件に沿う画像を作る生成モデルである。
著者らは元画像へのノイズ付加を途中で止め、指定した別条件を与えて復元する。例えばGEで撮った画像に別メーカーの条件を指定し、患者の組織配置を保ちながら濃淡を変えた画像を得る想定だ。生成画像を正解マスクと組にして分割モデルの学習へ加え、実際の別条件画像で性能を評価する。
新規性
著者らの提案は、元の乳房MRI画像を手がかりに、撮影条件だけが異なる反実仮想画像を生成し、組織分割の学習データに加えることだ。反実仮想とは「同じ患者を別の条件で撮ったら」と仮定した画像を指す。
拡散モデルへ撮影条件を指定し、元画像へのノイズ付加を途中で止めてから画像を復元する。患者の解剖学的な形を保ちながら、装置や撮影設定に伴う見え方を変えることを狙っている。
従来手法との違い
比較対象は、生成画像を追加せずに学習した乳房組織の分割モデルである。著者らは同じメーカーの画像で学習・評価する条件と、GEとSiemensの間で学習時と評価時のメーカーを変える条件を設けた。後者は学習時とは異なる撮影条件への対応を調べる。
表の平均Diceでは、メーカーをまたぐ二方向で生成画像を加えたモデルが上回った。一方、同一メーカー内の一条件では同値、もう一条件では低下しており、一律の改善とは読めない。
実験結果
Duke-Breast-Cancer-MRIの922人分を使用し、そのうち100人に乳房組織の正解マスクがあった。著者らは生成画像の評価にFID、SSIM、MMDなどを使い、分割性能は組織ごとの精度とDice係数で測った。Dice係数は予測領域と正解領域の重なりを示す指標である。
表では、Siemensで学習しGEで評価した平均Diceが0.730から0.739へ、GEで学習しSiemensで評価した値が0.742から0.750へ変化した。一方、Siemens内での評価は0.746から0.737へ低下した。著者らの結果は、条件によって小幅な改善と低下の双方を示す。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：乳房MRIの組織分割モデルを新しい撮影装置の画像へ適用する前に、装置差に対する弱さを調べる。入力は既存のMRI画像、撮影条件の記録、組織の正解マスクであり、出力は条件を変えた画像と分割精度の比較表である。
既存画像から別条件の反実仮想画像を作り、分割モデルの学習に加える。その後、実際に別装置で撮られた画像の正解マスクで性能を測る。生成画像の自然さだけを採用基準にせず、組織の境界が保たれるかも確認する。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：メーカーや撮影設定の異なる少数の画像を分け、元画像だけで学習した分割モデルを基準に置く。同じ学習用画像から反実仮想画像を作って追加したモデルと、別条件の実画像で比較する。評価用の患者を学習へ混ぜない。
脂肪と線維腺組織のDice係数を別々に記録し、平均値だけでなく悪化した症例を確認する。元画像と生成画像の組を見比べ、組織境界が保たれるかも点検する。導入判断には、対象装置での実画像評価を用いる。
限界と課題
著者らは、装置メーカーの種類とデータ数が限られると明記する。組織の正解マスクがある100人のうち、Siemensで撮影されたのは29人だった。表には生成画像の追加で平均Diceが下がる同一メーカー条件もあり、適用条件ごとの判断が必要である。
編集上の確認事項は、生成後も病変や細い組織境界が変わらないか、未検証の装置・撮影設定で同じ傾向が出るかである。画像を見た目だけで選別せず、元画像と正解マスクとの整合を専門家が確かめる。
出典
元のタイトル・要旨を確認する（英語）
Counterfactual MRI Data Augmentation using Conditional Denoising Diffusion Generative Models
Deep learning (DL) models in medical imaging face challenges in generalizability and robustness due to variations in image acquisition parameters (IAP). In this work, we introduce a novel method using conditional denoising diffusion generative models (cDDGMs) to generate counterfactual magnetic resonance (MR) images that simulate different IAP without altering patient anatomy. We demonstrate that using these counterfactual images for data augmentation can improve segmentation accuracy, particularly in out-of-distribution settings, enhancing the overall generalizability and robustness of DL models across diverse imaging conditions. Our approach shows promise in addressing domain and covariate shifts in medical imaging. The code is publicly available at https: //github.com/pedromorao/Counterfactual-MRI-Data-Augmentation
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-09-29