[論文レビュー] Accelerated Wirtinger Flow: A fast algorithm for ptychography

Rui Xu, Mahdi Soltanolkotabi+5 more|arXiv (Cornell University)|Jun 13, 2018

Advanced X-ray Imaging Techniques参考文献 54被引用数 24

ひとこと要約

本稿では、ワイアーティングル勾配降下法にネステロフのモーメンタムを組み合わせることで、高速かつ安定した位相回復アルゴリズムである加速ワイアーティングフロー（AWF）を提案する。リプシッツ的定数に基づく固定ステップサイズを導出することで、ラインサーチや手動によるパrameterチューニングの必要性を排除し、古典的手法（e-PIE や WF）よりも高速な収束と高い画像品質を達成している。これは、ノイズやプローブのずれが生じる状況下でも同様に有効である。

ABSTRACT

This paper presents a new algorithm, Accelerated Wirtinger Flow (AWF), for ptychographic image reconstruction from phaseless diffraction pattern measurements. AWF is based on combining Nesterov's acceleration approach with Wirtinger gradient descent. Theoretical results enable prespecification of all AWF algorithm parameters, with no need for computationally-expensive line searches and no need for manual parameter tuning. AWF is evaluated in the context of simulated X-ray ptychography, where we demonstrate fast convergence and low per-iteration computational complexity. We also show examples where AWF reaches higher image quality with less computation than classical algorithms. AWF is also shown to have robustness to noise and probe misalignment.

研究の動機と目的

• 大規模かつ高解像度の画像再構成において、古典的手法の収束が遅い問題に対処すること。
• X線pTYCHOGRAPHYにおける位相なし画像再構成のための高速でロバストかつパrameterフリーなアルゴリズムの開発。
• 勾配ベースの位相回復において、計算コストの高いラインサーチや手動によるパrameterチューニングを排除すること。
• 収束速度と再構成品質を向上させつつ、ノイズおよびプローブずれに対してロバスト性を維持すること。
• 非凸位相回復問題における固定ステップサイズの収束を理論的に裏付けること。

提案手法

• AWFは、ネステロフの加速勾配法とワイアーティングフロー（WF）を組み合わせ、位相回復における収束速度を向上させる。
• リプシッツ的定数に基づく理論的導出による固定ステップサイズを用いることで、ラインサーチの必要性を回避する。
• パrameterの手動チューニングを必要とせず、非凸コスト関数を最適化する。
• 未知のプローブを扱うために、e-PIEに類似した手続きを用いてプローブを更新する。
• 任意の位相なし測定に対して収束を分析し、確率的センシング集合を超える理論的保証を拡張する。
• 実際のノイズとずれを含むシミュレーテッドX線pTYCHOGRAPHYデータを用いて評価する。

実験結果

リサーチクエスチョン

• RQ1 ネステロフの加速法をpTYCHOGRAPHYにおけるワイアーティングフローに効果的に適用することで、収束速度の向上が達成できるか？
• RQ2 リプシッツ的定数に基づく固定ステップサイズが、非凸位相回復問題において収束性と安定性を保証するか？
• RQ3 e-PIE や WF と比較して、AWF の再構成品質と計算コストはどのように異なるか？
• RQ4 実用的な画像取得状況において、ノイズやプローブずれに対してもAWFはロバスト性を維持できるか？
• RQ5 ラインサーチや手動によるパrameterチューニングなしで、高品質な画像再構成が可能か？

主な発見

• AWFは、標準的なワイアーティングフローおよび e-PIE よりも顕著に高速な収束を達成し、低再構成誤差に到達するための反復回数を削減する。
• シミュレーションでは、1000反復後においても、計算コストを考慮した上で、e-PIE や WF よりも低い相対的再構成誤差を達成する。
• AWFはポissonノイズおよびプローブずれに対してロバストであり、高速な収束と高い画像品質を維持する。
• 特に1反復あたりの効率に優れ、古典的手法よりも少ない計算量で高い画像品質を達成する。
• 理論的リプシッツ的定数に基づく固定ステップサイズの使用により、ラインサーチやチューニングなしで安定した収束が可能である。
• CGMですらラインサーチを用いているにもかかわらず、1反復あたりの計算量が低いため、AWFは誤差および計算コストの両面でCGMを上回る。