Predictive risk estimation for the Expectation Maximization algorithm with Poisson data

11/11/2020 ∙ by Paolo Massa, et al. ∙ 0

In this work, we introduce a novel estimator of the predictive risk with Poisson data, when the loss function is the Kullback-Leibler divergence, in order to define a regularization parameter's choice rule for the Expectation Maximization (EM) algorithm. To this aim, we prove a Poisson counterpart of the Stein's Lemma for Gaussian variables, and from this result we derive the proposed estimator showing its analogies with the well-known Stein's Unbiased Risk Estimator valid for a quadratic loss. We prove that the proposed estimator is asymptotically unbiased with increasing number of measured counts, under certain mild conditions on the regularization method. We show that these conditions are satisfied by the EM algorithm and then we apply this estimator to select its optimal reconstruction. We present some numerical tests in the case of image deconvolution, comparing the performances of the proposed estimator with other methods available in the literature, both in the inverse crime and non-inverse crime setting.



There are no comments yet.


page 11

page 12

page 15

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.