ガウス模型に基づく画像修復と 確率伝搬法 東北大 情報科学 田中和之,吉池紀子 山口大 工 庄野逸 理化学研究所 岡田真人 参考文献:田中和之, 統計力学を用いた確率的 画像処理アルゴリズムの基礎 ---確率伝搬法と 統計力学 ---, 計測と制御, Vol.42, No.8, pp.631636, 2003. 1 ポイント 可解確率場モデルを用いた確率的画 像処理に対して確率伝搬法はどの程 度の精度が出せるのだろうか? ハイパパラメータの推定も含めて検証 したい. 2 画像修復の確率モデル 雑音 通信路 原画像 劣化画像 P(劣化画像 | 原画像 ) P(原画像 ) P(原画像 | 劣化画像 ) P(劣化画像 ) 3 ベイズの公式と確率的画像処理 P f 事前確率 y 画素 劣化過程 f f x, y x, y 事後確率 P f g , , fˆx , y z g g 原画像 x Pg f , f x, y 劣化画像 g g x, y x, y Pg f , P f Pg , P z g , , dz 4 周辺尤度最大化によるハイパパラメータ推定 ˆ ,ˆ arg max Pg , , fˆx , y z Pg , Pg z , Pz dz y P f x 原画像 f f x, y x, y x, y P z g , ˆ , ˆ dz ZPOS g , , 2 Pg f , f g ZPR g 周辺化 Pg , 周辺尤度 g 劣化画像 g g x, y x, y 5 画像修復の劣化過程と事前確率 f x, y , g x, y , 劣化過程 P g f , ( x , y ) 1 1 2 exp 2 f x, y g x, y 2 2 事前確率 2 P f exp f x , y f x 1, y Z PR ( x , y ) 2 1 2 exp f x , y f x , y 1 2 ( x , y ) 6 ベイズの公式と画像修復の事後確率 P f g, , Pg f , P f P g , 1 f x, y g x, y f x, y , f x 1, y f x, y , f x. y 1 Z P OS g, , ( x, y ) f x, y g x, y 1 2 exp 2 f x, y g x, y 2 2 f x, y , f x ', y ' exp f x, y f x ', y ' 2 7 周辺確率の導入 Px , y ( f x , y ) f x , y z x , y Pz g, , dz x ', y ' x, y P ( f x, y , f x ', y ' ) f x, y z x, y f x ', y ' z x ', y ' Pz g, , dz fˆx , y z x , y Pz g, , dz Px , y d 8 確率伝搬法(ベーテ近似) Px, y f x, y P x 1, y x, y f x, y ( x, y 1) ( x, y 1) M xx,y1, y ( f x, y ) ( x 1, y) ( x, y ) M xx,,yy 1 ( f x, y ) M xx,y1, y ( f x, y ) f M x, y x , y 1 x, y f x, y Pxx, y1, y f x, y , f x 1, y 1 Z xx,y1, y M xx12,,yy ( f x1, y ) ( x 2, y ) M xx11,,yy1 ( f x1, y ) ( x, y 1) M ( x 1, y ) ( x, y ) ( x 1, y ) M xx,,yy 1 ( f x, y ) 1 f x , y g x , y M xx,y1, y f x , y M xx,y1, y f x , y Z x, y x , y 1 x, y M xx11,,yy1 ( f x1, y ) M xx,y1, y ( f x, y ) ( x, y 1) Px , y f x , y ( x 1, y 1) M xx,,yy 1 ( f x, y ) M xx,,yy1 ( f x, y ) ( x 1, y ) , z x 1, y dzx 1, y ( x 1, y 1) f x, y g x, y f x, y , f x1, y f x1, y g x1, y M xx,y1, y f x, y M xx,,yy 1 f x, y M xx,,yy 1 f x, y M xx12,,yy f x 1, y M xx11,,yy 1 f x 1, y M xx11,,yy 1 f x 1, y 9 Message Update Rule of Loopy Belief Propagation g , 'M M x, y x 1, y g x , y , M xx,y1, y M xx,,yy 1 M xx,,yy 1 d x, y ( x, y 1) M xx,,yy 1 ( f x, y ) M xx,y1, y ( f x, y ) ( x 1, y ) M xx,, yy 1 ( f x, y ) ( x, y 1) M xx,,yy 1 M xx,,yy 1 d d ' x ', y ' 1 x', y ' x, y x ', y ' x ', y ' 2 M x, y ( ) exp x, y x, y 2 2 M xx, 1y, y ( f x1, y ) ( x, y ) x 1, y x, y ( x 1, y ) Fixed-Point Equations M M Natural Iteration 10 画像修復 事前分布から生成された画像による数値実験 (ハイパパラメータα,σは周辺尤度最大化で決定) 原画像(α=0.001) 平均場近似 ˆ 0.000298 ˆ 29.1 劣化画像 (σ=40) 確率伝搬法 厳密解 ˆ 0.000784 ˆ 0.001090 ˆ 37.8 ˆ 39.4 11 ガウス模型を用いた画像修復 原画像 厳密解 MSE:306 劣化画像 平均場近似 確率伝搬法 MSE: 1409 MSE: 593 MSE: 324 ウィーナーフィルター メジアンフィルター 平滑化フィルター MSE: 268 MSE: 369 MSE: 259 12 ガウス模型を用いた画像修復 原画像 劣化画像 MSE: 1512 厳密解 平滑化フィルター MSE:315 MSE: 411 平均場近似 確率伝搬法 MSE: 591 MSE: 325 ウィーナーフィルター メジアンフィルター MSE: 545 MSE: 447 13 まとめ ベイズの公式と周辺尤度を用いた確率的画像処理におい て可解確率場モデルのひとつであるガウス模型を事前確 率に採用した場合の確率伝搬法(ベーテ近似)の精度と性 能の評価 ハイパパラメータ推定において平均場近似から確率伝搬 法へと,厳密解で得られる結果に系統的に近づいている. 既存のフィルターに比較して平均場近似,確率伝搬法のい ずれもMSEでみて同等以上の結果が得られるのみならず 見た目にも良好な結果が得られている. 14
© Copyright 2024 ExpyDoc