[論文レビュー] Software Implementation of the Krylov Methods Based Reconstruction for the 3D Cone Beam CT Operator

Vojtěch Kulvait, Georg Rose|arXiv (Cornell University)|Oct 26, 2021

Medical Imaging Techniques and Applications参考文献 16被引用数 4

ひとこと要約

本稿では、効率的なレイトレーシング投影演算子を用いて、3次元コーンビームCT再構成のためのKrylov部分空間法—特にCGLSおよびLSQR—のオープンソースC++およびOpenCL実装を提示する。CGLSは、最先端のARTベースの手法PSIRTと比較して著しく高速に収束することを示しており、現代のGPUハードウェア上で大規模な3次元CT再構成に非常に実用的で効率的な代替手法であることが明らかになった。

ABSTRACT

Krylov subspace methods are considered a standard tool to solve large systems of linear algebraic equations in many scientific disciplines such as image restoration or solving partial differential equations in mechanics of continuum. In the context of computer tomography however, the mostly used algebraic reconstruction techniques are based on classical iterative schemes. In this work we present software package that implements fully 3D cone beam projection operator and uses Krylov subspace methods, namely CGLS and LSQR to solve related tomographic reconstruction problems. It also implements basic preconditioning strategies. On the example of the cone beam CT reconstruction of 3D Shepp-Logan phantom we show that the speed of convergence of the CGLS clearly outperforms PSIRT algorithm. Therefore Krylov subspace methods present an interesting option for the reconstruction of large 3D cone beam CT problems.

研究の動機と目的

• Krylov部分空間法を用いた、高パフォーマンスでオープンソースの3次元コーンビームCT再構成ソフトウェアパッケージの開発。
• 3次元CBCTにおけるCGLSおよびLSQRの収束速度と精度を、古典的な反復法（例：PSIRT）と比較して評価すること。
• 行列の直接保存を回避する演算子ベースの計算により、効率的で大規模な3次元再構成を可能にすること。
• Krylov法が臨床的CTにおけるアナリティカル再構成の実用的代替手段としての可能性を検討すること。
• 3次元CBCT問題における基本的な前処理戦略が収束性および安定性に与える影響を評価すること。

提案手法

• Siddonのレイトレーシング、フットプリント法、および新しいカットボクセルプロジェクタを用いて、完全な3次元コーンビーム投影演算子を実装。レイトレーシングとボクセルの交差計算を高精度に実現。
• CGLSおよびLSQRアルゴリズムに遅延された残差計算を適用し、1反復あたり1回の投影およびバックプロジェクションを削減することで、効率性を向上。
• 行列Aの非正方で大きな行列AᵀAを直接保存せずに、正規方程式AᵀAx = Aᵀbを共役勾配法で解くことで、システム行列Aの保存を回避。
• 基本的なヤコビ前処理を、AᵀAの対角成分を近似することで適用し、収束性を向上。ただし、コーンの縁付近の小さな対角成分により制限を受ける。
• OpenCLを介したGPUアクセラレーションにより、投影およびバックプロジェクション演算を高速化し、高速な反復的再構成を実現。
• デフォルト設定では、残差ベクトルの再正規直交化を回避している。これは、10反復間で0.0001%未満の数値誤差蓄積が観察されたため。

実験結果

リサーチクエスチョン

• RQ1 CGLSおよびLSQRといったKrylov部分空間法は、3次元コーンビームCT再構成において、PSIRTのような古典的なARTベースの手法と比較して収束速度で優位性を示せるか？
• RQ2 一般的な3次元CTデータにおいて、LSQRとCGLSの数値的安定性および再構成精度の観点での性能差は何か？
• RQ3 大規模な3次元CBCT問題において、基本的なヤコビ前処理が収束性にどれほど向上効果をもたらすか？
• RQ4 行列ベースの代数的再構成におけるメモリおよび計算のボトルネックを、演算子ベースのKrylov法が回避できるか？
• RQ5 Krylov法に順序付きサブセット戦略を拡張する可能性はあるが、収束性や安定性を損なわずに行えるか？

主な発見

• CGLSは3次元シェップ・ロジスティック・ファンタムにおいてPSIRTと比較して著しく高速に収束し、同じ再構成精度に到達するための反復回数が大幅に少ない。
• CGLSの収束速度はPSIRTよりもはるかに速く、大規模な3次元CBCT問題においてKrylov法がより効率的な代替手段であることが示された。
• LSQRとCGLSはほぼ同一の再構成結果を生成しており、LSQRの数値的安定性の優位性が一般的なCTデータでは重要ではないことが示唆された。
• デフォルトのCGLS設定では、残差の再正規直交化が不要である。誤差蓄積は非常に小さく（10反復間で0.0001%未満）、無視できる。
• 特にカットボクセルプロジェクタが優れた性能を発揮し、中程度のサイズの問題において数分で高速再構成が可能となった。
• 順序付きサブセット戦略のKrylov法への拡張によりさらなる高速化が期待できるが、正規直交性および安定性を維持するためのさらなる検討が必要である。