核心概念
グラフラプラシアン正則化子を用いた最大事後確率問題の解を展開することで、既知の良好なディノイザーから初期化された解釈可能な深層ディノイザーを構築する。
摘要
本論文では、グラフラプラシアン正則化子(GLR)を用いた最大事後確率(MAP)問題の解を展開することで、解釈可能な深層ディノイザーを構築する手法を提案している。
まず、既知の良好なディノイザーΨを選択し、Theorem 1に基づいてΨに対応するグラフラプラシアンLを導出する。Lは切断Taylor級数展開を用いて近似される。次に、線形システム(I + μL)x* = yの解xを共役勾配法を用いて展開し、パラメータ調整可能な深層ネットワークを構築する。
提案手法のGDDは以下の特徴を持つ:
- Ψから初期化されるため、最小限のパフォーマンス保証がある
- 各層が共役勾配法の反復に対応し、完全に解釈可能
- 少数のパラメータしか必要とせず、小規模データセットでも効果的に学習可能
- 共変量シフトに対してロバスト
実験結果では、GDDが既存手法と同等以上の性能を示しつつ、パラメータ数が大幅に少ないことが確認された。
統計資料
提案手法GDDは、ノイズ標準偏差σ = 10の場合、既知の良好なディノイザーBFよりも8.53dBの性能向上を示した。
提案手法GDDは、Total Variation法よりも最大1.28dBの性能向上を示した。
小規模データセットと共変量シフトの条件下で、提案手法GDDはDnCNNよりも約1.25dBの性能向上を示した。
引述
"グラフラプラシアン正則化子(GLR)は、画像ディノイジング[16]や JPEG逆量子化[17]、コントラスト強調[18]、点群ディノイジング[19]など、さまざまな画像復元問題に成功裏に適用されてきた。"
"Theorem 1 [24]は、任意の(擬)線形ディノイザーΨが、一定の条件の下で、GLRを正則化子とするMAP問題の解フィルタであることを示している。"
"提案手法GDDは、パラメータ数が大幅に少なく、小規模データセットでも効果的に学習可能であり、共変量シフトに対してもロバストである。"