Kernekoncepter
本稿では、複雑な気道構造を正確にセグメント化するため、マルチスケールネスト型 Residual UNet(MNR-UNet)と重み付きBreakage-Aware Loss(wBAL)を組み合わせた新しい3段階セグメンテーション手法を提案する。
Resumé
肺CTにおける多段階気道セグメンテーション:マルチスケールネスト型 Residual UNetに基づく新しいアプローチ
本稿では、肺CTにおける正確かつ完全な気道セグメンテーションを実現するための新しい3段階手法を提案する。この手法は、特に小さな気道の検出と気道連続性の維持という課題を克服するために設計された、マルチスケールネスト型 Residual UNet(MNR-UNet)と重み付きBreakage-Aware Loss(wBAL)を中核としている。
慢性呼吸器疾患(CRD)の診断と治療において、胸部CT画像における気道の正確なセグメンテーションは不可欠である。近年、ディープラーニングは医用画像セグメンテーションに革命をもたらしたが、小さな気道構造の複雑さとクラスの不均衡により、完全で連続的な気道セグメンテーションの達成は依然として困難である。本研究は、これらの課題に対処し、気道のトポロジー的な完全性を維持しながら、セグメンテーションの精度を向上させることを目的とする。
1. MNR-UNetの設計
本稿では、様々なスケールの気道特徴の学習を強化するために、マルチスケール入力とResidual Multi-scale Modules(RMM)を組み込んだ、MNR-UNetと呼ばれる新しいU-Netアーキテクチャを提案する。このアーキテクチャは、マルチスケール入力とRMMを用いることで、階層的な残差ネスティングを実現する。具体的には、ネットワークに入力される3Dパッチは、3つの解像度にピラミッドプーリングされ、各段階で残差マルチスケールエンコーダからの出力特徴を用いて残差が計算される。MNR-UNetは、エンコード段階でマルチスケールの気道情報を効果的に捕捉・統合することで、複雑な気道構造のモデリング能力を高める。
2. 重み付きBreakage-Aware Loss(wBAL)
クラス内不均衡の問題に対処するため、重み付きBreakage-Aware Loss(wBAL)を提案する。この損失関数は、気道骨格内の検出困難なボクセルにペナルティを課すことで、気道の連続性を重視する。wBALは、異なる気道分岐のサイズに基づいて気道ボクセルの重みを調整する局所不均衡ベースの重みと、セグメント化困難な気道中心線付近の点により大きな注意を払い、破損の原因となるボクセルに追加の重みを割り当てる中心線距離ベースの重みを組み合わせることで、小さな気道の検出と分岐の連続性の維持を両立させる。
3. 3段階トレーニングパイプライン
トポロジー的な完全性と精度を両立させるため、主要気道の抽出、小さな気道のマイニング、破損の修復に最適化された3段階のトレーニングプロセスを採用する。第一段階では、ランダムクロップサンプリングとDice損失を用いて、ネットワークに主要気道を予測させる。第二段階では、小さな気道抽出の課題に対処するため、General Union Loss(GUL)とハードマイニングクロップサンプリングを導入する。第三段階では、予測における不連続性に対処するために、破損クロップサンプリングと、wBALとGULを統合した損失関数を組み合わせることで、気道連続性を損なう中心線ボクセルにペナルティを課し、気道ツリーの長さを拡張する。