Deconvolution and Restoration of Optical Endomicroscopy Images
Ahmed Karam Eldaly, Yoann Altmann, Antonios Perperidis, Nikola, Krstajic, Tushar Choudhary, Kevin Dhaliwal, and Stephen McLaughlin

TL;DR
This paper introduces a hierarchical Bayesian framework for deconvolving and restoring optical endomicroscopy images, improving image quality by addressing fiber core cross coupling and sparse sampling issues.
Contribution
It proposes a novel Bayesian model and compares three estimation algorithms, including MCMC, VB, and ADMM, for OEM image restoration.
Findings
Bayesian methods effectively restore OEM images
VB and ADMM algorithms reduce computational time
Restored images show improved visualization and analysis
Abstract
Optical endomicroscopy (OEM) is an emerging technology platform with preclinical and clinical imaging applications. Pulmonary OEM via fibre bundles has the potential to provide in vivo, in situ molecular signatures of disease such as infection and inflammation. However, enhancing the quality of data acquired by this technique for better visualization and subsequent analysis remains a challenging problem. Cross coupling between fiber cores and sparse sampling by imaging fiber bundles are the main reasons for image degradation, and poor detection performance (i.e., inflammation, bacteria, etc.). In this work, we address the problem of deconvolution and restoration of OEM data. We propose a hierarchical Bayesian model to solve this problem and compare three estimation algorithms to exploit the resulting joint posterior distribution. The first method is based on Markov chain Monte Carlo…
Click any figure to enlarge with its caption.
Figure 1
Figure 2
Figure 3
Figure 4
Figure 5
Figure 6
Figure 7
Figure 8
Figure 9
Figure 10
Figure 11
Figure 12
Figure 13
Figure 14
Figure 15
Figure 16
Figure 17
Figure 18
Figure 19
Figure 20
Figure 21
Figure 22
Figure 23
Figure 24
Figure 25
Figure 26
Figure 27
Figure 28| Method | MCMC | ADMM | VB | |
|---|---|---|---|---|
|
3100 | 35.51 | 5.12 |
| Dataset/Method | MCMC | ADMM | VB |
|---|---|---|---|
| USAF chart | 5.9 | ||
| Lung tissue | 16.05 |
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Deconvolution and Restoration of Optical Endomicroscopy Images
Ahmed Karam Eldaly, Yoann Altmann, Antonios Perperidis, Nikola Krstajić, Tushar R. Choudhary, Kevin Dhaliwal, and Stephen McLaughlin A. K. Eldaly, Y. Altmann, A. Perperidis and S. McLaughlin are with the Institute of Sensors, Signals and Systems, School of Engineering and Physical Sciences, Heriot-Watt University, Edinburgh, UK. (Emails: {AK577; Y.Altmann; A.Perperidis; S.Mclaughlin}@hw.ac.uk)T. R. Choudhary is with the Institute of Biological Chemistry, Biophysics and Bioengineering, Heriot-Watt University, Edinburgh, United Kingdom (Email: [email protected])N. Krstajić and K. Dhaliwal are with the EPSRC IRC Hub in Optical Molecular Sensing & Imaging, MRC Centre for Inflammation Research, Queen’s Medical Research Institute, University of Edinburgh, Edinburgh, UK (Emails: {N.Krstajic; Kev.Dhaliwal}@ed.ac.uk)This work was supported by the EPSRC via grant EP/K03197X/1 and the Royal Academy of Engineering through the research fellowship scheme.
Abstract
Optical endomicroscopy (OEM) is an emerging technology platform with preclinical and clinical imaging applications. Pulmonary OEM via fibre bundles has the potential to provide in vivo, in situ molecular signatures of disease such as infection and inflammation. However, enhancing the quality of data acquired by this technique for better visualization and subsequent analysis remains a challenging problem. Cross coupling between fiber cores and sparse sampling by imaging fiber bundles are the main reasons for image degradation, and poor detection performance (i.e., inflammation, bacteria, etc.). In this work, we address the problem of deconvolution and restoration of OEM data. We propose a hierarchical Bayesian model to solve this problem and compare three estimation algorithms to exploit the resulting joint posterior distribution. The first method is based on Markov chain Monte Carlo (MCMC) methods, however, it exhibits a relatively long computational time. The second and third algorithms deal with this issue and are based on a variational Bayes (VB) approach and an alternating direction method of multipliers (ADMM) algorithm respectively. Results on both synthetic and real datasets illustrate the effectiveness of the proposed methods for restoration of OEM images.
Index Terms:
Optical endomicroscopy, Deconvolution, Image restoration, Irregular sampling, Bayesian models.
I Introduction
Pneumonia is a major cause of morbidity and mortality in mechanically ventilated patients in intensive care [1]. However, the accurate diagnosis and monitoring of suspected pneumonia remain challenging [2]. Current methodologies consist of culturing bronchoalveolar lavage fluid (BALF) retrieved from bronchoscopy, but this often takes 48 hours to yield a result which still has low specificity and sensitivity [3]. Structural imaging with X-ray or computed tomography (CT) scans are also often non-diagnostic.
Optical endomicroscopy (OEM) is an emerging, optical fibre-based medical imaging modality with utility in a range of clinical indications and organ systems, including gastro-intestinal, urological and respiratory tracts. The technology employs a proximal light source, laser scanning or Light Emitting Diode (LED) illumination, linked to a flexible fibre bundle, performing microscopic fluorescent imaging at its distal end. The diameter of the packaged fibre can be 500\text{,}\mathrm{\SIUnitSymbolMicro m}$$ , enabling the real-time imaging of tissues that were previously inaccessible through conventional endoscopy. Probe-based confocal laser endomicroscopy, is currently the most widely used clinical OEM platform approved for clinical use. However, there have recently been a number of studies describing novel, flexible, versatile and low-cost OEM architectures [4, 5, 6], employing wide-field LED illumination sources, capable of imaging at multiple acquisition wavelengths [7]. Wide-field fiber optic imaging devices, such as the one being developed by our group provide sparse and usually irregularly-spaced intensity readings of the scene, due to the irregular packing of the fibre cores within the fibre bundle. Fibre bundles usually contain approximately 25,000 fibre cores that are transmitting and collecting the light simultaneously. Note that it is only the fibre cores which contain information while the cladding, (the space between the fibre cores), does not.
One of the main challenges of OEM images is enhancing the restoration of the signals at the receiver for better image visualization and/or subsequent analysis. Fiber core cross coupling is one of the main reasons for image degradation in this type of imaging [8, 9]. In confocal endomicroscopy, the detector pinhole can mask out light coupled to neighbouring cores before reaching the detector. Consequently, the effect of inter-core coupling in imaging capabilities is inherently of greater importance in wide-field endomicroscopy. Perperidis et al. [10] have quantified the average spread of inter-core coupled light, with approximately a third of the overall light coupling to neighbouring cores. Consequently, cross coupling causes severe blurring in the resulting images, whose restoration is formulated as an inverse problem. We will discuss in detail cross coupling effects in Section II. In this work, we consider a noisy observation vector , of an original intensity vector , that is modelled by the following linear forward model
[TABLE]
where is the matrix representing a linear operator which can model different degradation. Here, models fiber core cross coupling and/or spatial blur. We specify the dimensions of the variables later in the text. In (1), the vector stands for additive noise, modelling observation noise and model mismatch and is assumed to be a white Gaussian noise sequence. In wide-field OEM, the constant background fluorescence of the fiber bundle [11, 7], is significant (between 90% and 60% of the total signal) providing a significant offset to all fluorescence measurements from tissue. Hence, the total noise level does not depend on the tissue signal level. Also, we consider applications where the photon flux is high ( photoelectrons generated per pixel per typical exposure time 50 ms). Therefore, the Gaussian noise assumption holds [12, 13, 14].
The problem of estimating from is an ill-posed linear inverse problem (LIP); i.e., the matrix is singular or very ill-conditioned. Consequently, this problem requires additional regularization (or prior information, in Bayesian inference terms) in order to reduce uncertainties and improve estimation performance. State-of-the-art algorithms for solving such problems can be split into either convex optimization or Bayesian methods.
In [15, 16, 17, 18], the problem of estimating given is formulated as an unconstrained optimization problem as follows
[TABLE]
where is a regularization function, is the standard -norm, is a regularization parameter, and is the indicator function defined on the positive set of . For solving problems of the form (2), state-of-the-art algorithms potentially belonging to the iterative shrinkage/thresholding family [15, 16, 17, 18] can be used. In [19, 16], the unconstrained problem in Eq.(2) is solved by an algorithm called split augmented Lagrangian shrinkage algorithm (SALSA) which is based on variable splitting [20, 21].
Alternatively, many studies have considered hierarchical Bayesian models to solve the deconvolution and restoration problem [22, 23, 24, 25, 26, 27, 28, 29, 30, 31]. These models offer a flexible and consistent methodology to deal with uncertainty in inference when limited amount of data or information is available. Moreover, other unknown parameters can be jointly estimated within the algorithm such as noise variance(s) and regularization parameters. As such, they represent an attractive way to tackle ill-posed problems such as the one considered in this work. These methods rely on selecting an appropriate prior distribution for the unknown image and other unknown parameters. The full posterior distribution can then be derived from the Bayes’ rule, and then exploited by optimization or simulation-based (Markov chain Monte Carlo) methods.
The main contributions of this work are fourfold:
We address the problem of deconvolution and restoration in OEM. To the best of our knowledge, it is the first time this problem is addressed in a statistical framework by using a hierarchical Bayesian model. 2. 2.
We develop algorithms dedicated to irregularly sampled images which do not rely on strong assumptions about the spatial structure of the sampling patterns. The developed methods can thus be applied to a wide range of imaging systems, and fiber bundle designs. 3. 3.
We derive three estimation algorithms associated with the proposed hierarchical Bayesian model and compare them using extensive simulations conducted using controlled and real data. The first algorithm generates samples distributed according to the posterior distribution using Markov chain Monte Carlo (MCMC) methods [32]. This approach also allows the estimation of the hyperparameters associated with the priors. However, as mentioned previously, the resulting MCMC-based algorithm presents a high computational complexity. The second and third algorithms deal with this limitation and approximate the joint posterior distribution. The second algorithm uses the variational Bayes (VB) methodology [33, 34] to approximate the joint posterior distribution by minimizing the Kullback–Leibler (KL) divergence between the true posterior distribution and its approximation [35]. It can also estimate the hyperparameters associated with the prior distributions, and hence it is totally unsupervised, as is the MCMC-based method. The third algorithm is based on the alternating direction method of multipliers (ADMM). Although the low computation complexity of this algorithm, the hyperparameters associated with the priors need to be chosen carefully by the user, and hence it is considered as a semi-supervised method. 4. 4.
We use Gaussian Processes (GP) to interpolate the resulting samples to provide a meaningful image and quantify uncertainties at each interpolated sample.
The remaining sections of the paper are organized as follows. Section II discusses the cross coupling problem and formulates the problem of deconvolution and restoration of OEM data. The proposed hierarchical Bayesian model is then presented in Section III. Section IV introduces the three proposed estimation algorithms based on MCMC and optimization. Results of simulations conducted using synthetic and real datasets are discussed in Section VI and Section VII, respectively. Conclusions and future work are finally reported in Section VIII.
II Problem Formulation
Fig. 1 illustrates what happens in the fibre bundle when receiving fluorescent light from an object being imaged. The vectors , , and represent light intensities at the object being imaged (tissue in this case), at the distal end of the fibre bundle, and at the image plane respectively. The transform represents the cross coupling effect defined later in the text, represents the spatial blur acting between the proximal end of the fibre bundle and the image plane, whereas is that between the distal end of the fibre bundle and the tissue being imaged. The two spatial blurs and are spatially variant, can be characterized as the distance between the image plane and the proximal end of the fibre is known, whereas cannot be fully characterized as is unknown and the frames here are analyzed independently. Hence, to overcome this problem, we aim to recover the intensity vector rather than .
Fig. 2 provides and illustrative example of cross coupling between fiber cores. If an individual fiber core is illuminated in , the neighbouring cores in will be affected by a specific percentage of the incident light on the illuminated core. Experimental results in current fiber bundle (which might be different for other bundles) showed that around 61% of the light transmitted through a single core remains in that core, around 34% migrates to the immediate neighbouring cores, around 4% to the second order neighbours and less than 1% to the third, fourth, and fifth order neighbours [10].
Fig. 3 illustrates how we construct the forward observation model to mimic the same output as the endomicroscopy imaging system. The first image on the left-hand side of the figure represents the illumination of one fiber core. This results in cross coupling to the neighbouring cores (convolution with a first linear operator ), then the spatial blurring effect around each fiber core (convolution with a second linear operator ) and finally the fourth image of the figure shows the final system output after adding white Gaussian noise.
The linear model in (1) can now be written as
[TABLE]
where in (1) is replaced by in (3), the vector is the observed data matrix, and is the image to be restored.
From preliminary results, we propose to model cross-coupling by an isotropic zero mean 2D generalized Gaussian kernel applied to the fiber intensities [10] as follows
[TABLE]
where denotes the euclidean distance between the cores (or spatial locations) and , which corresponds to approximately 3.3 pixels between neighbouring cores. From (4), it can be seen that neighbouring fiber cores will be more closely coupled than distant ones. The values of and , which control the amount of cross-coupling (the higher, the more coupling) and which are system dependent, are adjusted from preliminary measurements (calibration). Note that other cross-coupling models could also be considered instead of (4) depending on the imaging system used.
The spatial blur affecting each fiber core can be modelled by a Gaussian spatial filter, as illustrated in Fig. 4, which shows a background image i.e., an image from a sample presenting constant intensity, using an endomicroscopy imaging system, and a zoomed-in region of this image, bright and dark areas represent fiber cores and their cladding, respectively. The intensity profile across one line in this image is a series of Gaussian kernels. However, the variation of the shape and width of the kernels is due to the variation in core sizes.
Due to the variation in core sizes, the blurring kernel varies accordingly, and hence the cores tend to overlap. So the complete model in (3) becomes more complex, and potentially computationally expensive for long image sequences (videos). Indeed, there is no structure in which allows us to compute rapidly. Hence we propose a simplification of this model and represent each core by a single intensity value. The mean intensities of fibre core pixels could be used, but the overlap between the cores makes its computation difficult. Since the variation of the width of this blur is not too significant, the maximum intensity of each core is considered instead ( in Fig. 1).
Following the above mentioned points, the model in (3) can be simplified to
[TABLE]
Assume that is the total number of pixels in the image, and representing number of fibre cores in the image, the input , where is the pseudo-inverse of , and the output are two vectors representing central core intensities, where, , and . The noise is assumed to be additive white noise which is independent and identically distributed (i.i.d) zero mean Gaussian noise with variance , denoted as , where means “is distributed according to” and is the identity matrix.
The problem investigated in this paper is to estimate the actual intensity values , and the noise variance from the observation vector . As mentioned previously, to solve this problem, we propose a hierarchical Bayesian model and a set of different estimation methods to estimate the unknown parameters.
III Hierarchical Bayesian Model
This section introduces a hierarchical Bayesian model proposed to estimate the unknown parameter vector and . This model is based on the likelihood function of the observations and on prior distributions assigned to the unknown parameters.
III-A Likelihood
Eq. (5) yields that . Consequently, the likelihood can be expressed as
[TABLE]
III-B Parameter Priors
III-B1 Prior for the underlying intensity field
A truncated multivariate Gaussian distribution (MVG) is assigned to the intensity field .
[TABLE]
where is the indicator function defined on the positive set of , controls the global correlation between intensities, and the covariance matrix which defines the spatial correlation between the cores is defined by
[TABLE]
where denotes the distance between the spatial locations , and . Equations (7) and (8) promote smooth intensity variations between neighbours while ensuring that the prior dependence between neighbouring cores decrease as increases. In this work is the standard euclidean distance. The parameters were learned from the irregular sampling pattern of the OEM system. Precisely, we used known images and selected by maximum likelihood estimation, which occurs when is at its greatest, which corresponds to maximizing . While is left unknown for each image, are fixed in the rest of the simulations as the average values obtained with the training images.
Considering such a prior is equivalent to assuming a Gaussian process on , this allows us to interpolate the resulting deconvolved intensities using Gaussian processes [36] as we will see in section V.
III-B2 Prior for the noise variance
A conjugate inverse-Gamma prior is assigned to the noise variance
[TABLE]
where is fixed arbitrarily, while the hyperparameter is estimated within the algorithm.
III-B3 Prior for the hyperparameter
The hyperparameter associated with the parameter prior defined above is assigned to a conjugate Gamma distribution:
[TABLE]
where and are fixed and user-defined parameters which might depend on the quality of the data to be recovered. In this work, we fixed arbitrarily.
III-B4 Prior for the hyperparameter
To reflect the lack of prior knowledge about the regularization parameter in (7), the following weakly informative conjugate inverse-Gamma prior is assigned to it.
[TABLE]
where are fixed to . Note that we did not observe significance change in the results when changing these hyperparameters.
The next section derives the joint posterior distribution of the unknown parameters associated with the proposed Bayesian model.
III-C Joint posterior distribution
Assuming the parameters and are a priori independent, the joint posterior distribution of the parameter vector and hyperparameters can be expressed as
[TABLE]
where
[TABLE]
The directed acyclic graph (DAG) summarizing the structure of proposed Bayesian model is depicted in Fig. 5. This posterior distribution will be used to evaluate Bayesian estimators of . For this purpose, we propose three algorithms: an MCMC-based approach and two optimization-based approaches, in which VB and ADMM are considered. The first approach uses an MCMC method to evaluate the minimum-mean-square-error (MMSE) estimator of by generating samples according to the joint posterior distribution. Moreover, it allows the estimation of the hyperparameter vector along with the noise variance . However, it exhibits a relatively long computational time. The second and third algorithms which deal with this issue and provide fast MMSE estimate for the VB approach and MAP estimate for the ADMM approach. The VB approach approximates the joint posterior distribution in (12) by minimizing the Kullback-Leibler (KL) divergence between the true posterior distribution and its approximation [35]. The ADMM approach is achieved by maximizing the posterior distribution (12) with respect to (w.r.t.) . Note however, that the hyperparameters as well as are fixed for this approach. The three estimation algorithms are described in the next section.
IV Bayesian Inference
IV-A MCMC algorithm
To overcome the challenging derivation of Bayesian estimators associated with , we propose to use an efficient MCMC method to generate samples asymptotically distributed according to the posterior presented in (12). More precisely, we consider a Gibbs sampler described next. The principle of the Gibbs sampler is to sample according to the conditional distributions of the posterior of interest [[32], Chap. 10]. In this work, we propose to sample sequentially the elements of using updates that are detailed below.
IV-A1 Sampling the intensity field
From (12), since the prior (7) is conjugate to the Gaussian distribution, the full conditional distribution of is given by
[TABLE]
where
[TABLE]
Sampling from (14) can be achieved efficiently by using the Hamiltonian method proposed in [37].
IV-A2 Sampling the noise variance
By cancelling out the terms that don’t depend on from the posterior distribution in (12), its conditional distribution can be written as
[TABLE]
which is easy to sample from.
IV-A3 Sampling the hyperparameters and
It can be easily shown that can be sampled from the following Gamma distribution
[TABLE]
In a similar fashion to the noise variance, can be sampled from the following inverse-Gamma distribution
[TABLE]
The algorithm for generating samples asymptotically distributed according to the posterior distribution using Gibbs sampler is shown in Algorithm 1.
The posterior distribution mean or minimum mean square error (MMSE) estimator of can be approximated by
[TABLE]
where the samples from the first iterations (corresponding to the transient regime or burn-in period, which is determined visually from preliminary runs) of the sampler are discarded.
IV-B Variational Bayes algorithm
For this approach, we consider an approximation of by a simpler tractable distribution following the variational methodology [34], moreover, here, we relax the positivity constraints about the intensity field vector . Note, however that the positivity constraints can be incorporated but the covariance matrix of the intensity field would become more complicated [38], chap. 5. As will be shown in Sections VI and VII, this constraint relaxation yields a fast estimation procedure providing estimation results which compete with the methods incorporating this constraint. The distribution will be found by minimizing the Kullback-Leibler (KL) divergence, between the actual posterior distribution and its approximation, given by [35] [39]
[TABLE]
which is always non-negative and equal to zero only when . In order to obtain a tractable approximation, the family of distributions are restricted utilizing the mean field approximation [40] so that , where .
The lower bound of the KL divergence is given by
[TABLE]
For , let us denote by , the subset of with removed; for instance, if , . Then utilizing the lower bound for the joint probability distribution in (20) we obtain an upper bound for the KL divergence as follows
[TABLE]
Therefore, we minimize this upper bound instead of minimizing the KL divergence in (20). Note that the form of the inequality in (22) suggests an alternating (cyclic) optimization strategy where the algorithm cycles through the unknown distributions and replaces each variable with a revised estimate given by the minimum of (22) with the other distributions held constant. Thus, given , the posterior distribution approximation can be computed by solving
[TABLE]
In order to solve this equation, we note that differentiating the integral on the right hand side in (22) w.r.t. results in (see [41], Eq. (2.28))
[TABLE]
where
[TABLE]
We obtain the following iterative procedure to find by applying this minimization to each unknown in an alternating way
Now we detail the solutions at each step of algorithm (2) explicitly.
IV-B1 Updating intensity field vector
From (24), it can be shown that is an -dimensional Gaussian distribution, rewritten as
[TABLE]
where the mean and covariance of this normal distribution can be calculated from step 3 in Algorithm 2 as
[TABLE]
IV-B2 Updating noise variance
It is easy to show from (24) that the noise variance follows an inverse-Gamma distribution given by
[TABLE]
whose mean is given by
[TABLE]
where
[TABLE]
where denotes the trace of the matrix.
IV-B3 Updating regularization parameter
In a similar fashion to noise variance, the regularization parameter follows an inverse-Gamma distribution given by
[TABLE]
whose mean is given by
[TABLE]
where
[TABLE]
IV-B4 Updating the hyperparameter
The hyperparameter follows a Gamma distribution given by
[TABLE]
whose mean is given by
[TABLE]
In Algorithm 2, no assumptions were imposed on the posterior approximation of . We can, however, assume as [28, 29, 30, 31, 42], that this distribution is degenerate, i.e., distribution which takes one value with probability one and the rest of the values with probability zero. We can obtain another algorithm under this assumption which is similar to algorithm 2.
The stopping criterion we use is , where [43].
It is clear that using degenerate distribution for in Algorithm 3 removes the uncertainty terms of the intensity field estimate. It has been shown that this helps to improve the restoration performance [28, 29, 30, 31, 42]. Moreover, it also reduces the computational complexity as there is no need to compute explicitly the covariance matrix at each iteration. Finally, a few remarks are needed to obtain a fast algorithm. The inverse of the covariance matrix needs to be computed only once before the loop in Algorithm 3. We also considered the MATLAB operation for the update of the intensity field vector , which is faster than computing the covariance matrix in (27b), then updating the mean in (27a). For very big images, diagonal approximation [29] or conjugate gradient [44] can be considered for the update of the intensity field vector .
IV-C ADMM algorithm
This section describes another alternative to the MCMC algorithm which is based on an optimization algorithm. The latter maximizes the joint posterior distribution (12) with respect to (w.r.t.) the parameters of interest, with fixing the hyperparameter vector , to approximate the MAP estimator of , or equivalently, by minimizing the negative log-posterior distribution given by . The resulting optimization problem is tackled using ADMM that sequentially updates the different parameters, which is widely used in the literature for solving imaging inverse problems [43, 45, 19]. We rewrite the model as an optimization problem as follows
[TABLE]
where the regularization function is proportional to the negative logarithm of the intensity field prior considered in (7) up to an additive constant, i.e. , and is the regularization parameter. Given this objective function, we write the constrained equivalent formulation as follows
[TABLE]
where and are the variables to minimize. In order to solve for and , we construct the augmented Lagrangian corresponding to (37) as follows
[TABLE]
where is a positive parameter. The ADMM algorithm for solving (38) is shown in Algorithm (4). During each step of the iterative algorithm, is optimized w.r.t. (step 3) and (step 4) and then the Lagrange multipliers are updated (step 6). The stopping criterion we use is , where [43].
V Non-Linear Interpolation Using Gaussian Process Regression
In order to visually view a meaningful image from the deconvolved intensities, we consider non-linear interpolation based on Gaussian processes (GP) [36], since it can provide confidence intervals for each interpolated pixel. A classic choice consists of considering a zero-mean GP with an arbitrary covariance matrix. Here, we choose this covariance matrix to be . Precisely, we interpolate using the prior distribution previously defined in (8). If is very small, then approaches its maximum . If is distant from , we have instead , i.e. the two points are considered to be a priori independent. So, for example, during interpolation at new location, distant cores will have negligible effect. The amount of spatial correlation depends on the parameters , and , which are estimated in the way we previously mentioned in section III-B1.
If we consider , contains all the positions of all the observed cores (whose estimated intensities are gathered into ), and a new spatial location for which we want to predict the intensity , the GP can be extended as follows
[TABLE]
where . Eq. (39) shows that the conditional distribution of each predicted intensity given the previously estimated intensities, follows a Gaussian distribution whose mean and variance are given by
[TABLE]
By setting , the mean in (40) is finally used to estimate each interpolated intensity, while the variance is used to provide additional information (measure of uncertainty) about the interpolated intensity values.
The Matlab implementations of this paper are provided at https://sites.google.com/site/akeldaly/publications.
VI Simulations Using Synthetic Data
VI-A Data creation
The performance of the proposed methods is investigated by reconstructing a standard test image. A subsampled version of this image is obtained by considering the sampling pattern of an actual endomicroscopy system, as illustrated in Fig. 6. This figure provides an example of a homogeneous region imaged through Alveoflex (Mauna Kea Technologies, France) fiber bundle [46][47]. Such image is used for calibration and to identify the number and positions of the fiber cores. The build-in MATLAB function “vision.BlobAnalysis” was used to detect central fibre core pixels.
Fig. 7 shows the original Lena image (left) and an example of system output (right) after applying the model in Eq. (3). This image is formed by creating a binary mask in which a value of 1 is assigned to pixels corresponding to the central pixels of each core in Fig. 6(b), and zero otherwise. This mask is then multiplied point by point by the Lena image in Fig. 7(a) in order to obtain the subsampled image. The model in Eq. (3) is then applied to obtain an image that simulates the system’s output which is shown in Fig. 7(b). This image is created using subsampled intensities corresponding to 1.29% of the original Lena image. For simulated data, we considered a Gaussian spatial blurring kernel with one size in all the simulations.
VI-B Performance analysis
The performance discriminator adopted in this work to measure the quality of the deconvolved fiber cores is the root mean square error (RMSE), which is computed using intensities at the core locations using
[TABLE]
where and are vectors of the subsampled reference Lena image and its deconvolved version respectively, and is the number of fibre cores.
For synthetic data, in order to check the performance of the algorithm with different cross coupling effects, different values of and in (4) can be considered. However, this can be simplified by considering a 2D Gaussian kernel defined by (42)
[TABLE]
since it involves only one variable to change, namely (representing a squared distance, in pixels). This is equivalent to setting . Note that this simplification is considered only for synthetic data in order to assess the influence of the kernel width. The generalized Gaussian cross coupling kernel defined in (4) will be considered for real data.
The three methods showed similar results in terms of RMSE and interpolated images. The following shows the VB method’s results. Fig. 8 shows examples of interpolated intensities after deconvolution using GP in the noise-free case () and noisy case () and different values of , with the corresponding confidence interval images. we can observe that the structure of the Lena image can be recovered in the two cases. Moreover, in the confidence interval images, we can observe that as we go away from central cores, the confidence interval of the interpolated intensities decreases.
In order to measure the performance of the algorithms, we consider different noise variances () as well as different cross coupling effects (). Fig. 9 shows the RMSE (in log-scale) before and after deconvolution versus at . We can observe that all of the methods are very effective since the RMSE after deconvolution is always lower than that before deconvolution. Moreover, the gain increases with cross coupling.
In order to analyze the effect of noise variance and cross coupling separately, we fix one of them and change the other as shown in Fig. 10. In this figure, we show plots of RMSEs after deconvolution for different at fixed and vice versa. In Fig. 10(a), we can observe that there is roughly a linear relationship between RMSE and at fixed . Moreover, the behaviour at is almost the same. In Fig. 10(b), we can observe that RMSE is fairly constant as increases at constant . Furthermore, it starts to increase as increases but still remains constant when changing .
For the MCMC method, in all of the simulations in this paper including the real datasets, , including , which were determined visually from preliminary runs, were used. For the ADMM method, different regularization parameter values are tested, we pick up the one corresponding to the lowest RMSE.
VI-C Comparison
In this section, we compare the three proposed methods for deconvolution and restoration of OEM images. The comparison is conducted in terms of RMSE before and after deconvolution, as well as in terms of computation time.
Fig. 11 compares RMSEs after deconvolution versus different as well as different . We can observe that for all of the methods, as increases at constant , RMSE increases. On the other hand, at fixed , RMSE seems to be roughly constant for , then, it starts to increase as increases. It is clear that all the methods behave similarly in terms of RMSE.
Table I shows the average computation time (in seconds) of the three proposed methods. The experiments were conducted on ACER core-i3-2.0 GHz processor laptop with 8 GB RAM. It is clear that the MCMC method is the most computationally expensive method. The ADMM method is second, and the VB the least. Despite the relatively high computation time of the MCMC method, it is a parameter free method compared to the ADMM-based method in which the regularization parameter should be chosen carefully. The VB approach is considered to be the best compared to MCMC and ADMM, it can provide similar RMSE but with lower computation complexity, moreover, it is fully automatic in the sense that it can estimate the hyperparameters associated with the parameters as mentioned previously in section IV-B.
Although the MCMC and ADMM algorithms can estimate the noise variance and model hyperparameters, in practice these parameters are very difficult to estimate accurately, (specifically and ) due to the similarity between and in (15b) and (27b). Therefore, we have to make an informed choice about one of these parameters, specifically the choice of the hyperparameters , and in (9) and (10). In Fig. 10(b), we observe that the RMSEs in practise are close to the true noise standard deviation, and hence the noise variance can be inferred.
VI-D Robustness
To test the robustness of the proposed methods, we create the data using a specific and we deconvolve using different values. Following this strategy, we create the data using and we deconvolve using . The three estimation approaches showed similar results.
Fig. 12 shows plots of RMSE after deconvolution versus at fixed and vice versa. In Fig. 12(a), we can observe that the noise variance has no effect on the deconvolution in the tested interval as RMSE is constant at fixed . In Fig. 12(b), there is an approximately linear relationship between RMSE and at constant . Furthermore, lower values of than the one we created the data with (i.e., ) yield lower RMSE than higher ones (i.e., ). In other words, it is slightly better to underestimate than to overestimate it.
We observe that deconvolution using the value we created the data with () yields the minimum RMSE. Moreover, RMSE after deconvolution is always lower than that before deconvolution except for at which it is higher.
VII Simulations Using Real Data
The performance of the proposed methods has been evaluated on two real datasets; the 1951 USAF resolution test chart and ex vivo human lung tissue. Both of them were collected using OEM system [7] with monochrome detection (Grasshopper3 camera GS3-U3-23S6M-C, Point Grey Research, Canada) and 470 nm LED illumination (M470L3, Thorlabs Ltd, UK) for lung autofluorescence excitation. Excised human lung tissue was placed in a well plate. Human tissue was used with regional ethics committee (REC: 13/ES/0126) approval and was retrieved from the periphery of specimens taken from lung cancer resections. In order to adjust the cross coupling kernel parameters , a study was performed to measure, analyze and quantify inter-core coupling within coherent fibre bundles [10]. This study showed how light is spread over the neighbouring cores, and gave statistical analysis on coupling percent in neighbouring cores. It showed that around 61% of transmitted light remains in the central core, around 34% in the first neighbouring cores, around 4% in the second neighbouring cores, and less than 1% in the third, fourth and fifth neighbouring cores. This leads to fixing (in pixels) and .
VII-A 1951 USAF resolution test chart
The 1951 USAF chart is a resolution test pattern set by US Air Force in 1951. It is widely accepted to test the resolution of optical imaging systems such as microscopes, cameras and image scanners [48]. Fig. 13 (a) shows the original USAF resolution test chart used in the project. The resulting image obtained by fiber bundle is shown in Fig. 13 (b) with image size and is composed of 7,776 fiber cores ( of the image).
A non-linear interpolation based on GP of central core intensities of the image in Fig. 13(b) is presented in Fig.14(a), with the corresponding confidence intervals image in Fig.14(c). We can observe the blurring which is caused by the cross coupling effect as well as the sparsity of the data.
The outputs of the MCMC, VB, and ADMM algorithms are very similar. Thus, we show the results of the VB method. Fig. 14(b) shows an example of one of the output images with the corresponding confidence intervals in Fig. 14(d). The set of ticker strips (top left corner of the image) is now better resolved and the overlap between them is reduced. The small set of strips which is at the bottom could not be resolved, which gives an indication about the resolving resolution of this endomicroscopy system. Regions of high uncertainty (which appear as blobs in dark red) are where there may be no cores or they are dead, this in addition to the irregular core sampling are the reasons for some strips appear a bit fragmented.
VII-B Ex vivo human lung tissues
Fig. 15(a) shows the output image of the OEM system. Image size is and is composed of 13,343 fiber cores ( of the image). Non-linear interpolation based on GP of central core intensities is presented in Fig.15(b). Similar to the USAF resolution test chart, we aim at reducing cross coupling effect as well as getting a more resolved image.
Similar to the USAF resolution test chart results, the outputs of the MCMC, VB, and ADMM algorithms are very similar. We only show the results of the VB method. Fig. 15(c) shows an example of interpolated deconvolved samples using GP. The lung structure is now better resolved and more sharper than before deconvolution. Moreover, confidence intervals are shown in Fig. 15(d). We can observe that as we move away from the central cores, the confidence of the interpolated intensities decreases and vice versa.
Table II provides the computation time of the 1951 USAF resolution test chart and the ex vivo lung tissue image. It is clear that the VB is still the fastest despite the change of the images size.
VIII Conclusion and Future Work
This paper introduced a hierarchical Bayesian model and three estimation algorithms for the deconvolution of optical endomicroscopy images. The deconvolution accounts and compensates for fibre core cross coupling which causes major image degradation in this type of imaging. The resulting joint posterior distribution was used to approximate the Bayesian estimators. First, a Markov chain Monte Carlo procedure based on a Gibbs sampler algorithm was used to sample the posterior distribution of interest and to approximate the MMSE estimators of the unknown parameters using the generated samples. Second, a variational Bayes approach to approximate the joint posterior distribution by minimizing the Kullback-Leibler divergence was used. Third, an approach based on an alternating direction method of multipliers was used to approximate the maximum a posteriori estimators. The three algorithms showed similar estimation performance while providing different characteristics, the MCMC and VB based approaches are fully automatic in the sense that they can jointly estimate the hyperparameters associated with the priors, however, the MCMC based approach showed high computational complexity which could be overcome by the VB and ADMM approaches. Although the ADMM approach has low computational complexity, it is semi-supervised in the sense that the hyperparameters associated with the priors need to be chosen carefully by the user. A non-linear interpolation approach based on Gaussian processes was considered to restore the full images from the samples to provide a meaningful image for interpretation. In the future, we will consider temporal information while deconvolving. Accounting for the different core sizes is also clearly an interesting route currently under investigation.
Acknowledgement
This work was supported in parts by the Engineering and Physical Sciences Research Council (EPSRC, United Kingdom) Interdisciplinary Research Collaboration grant EP/K03197X/1 and by the Royal Academy of Engineering under the Research Fellowship scheme (RF201617/16/31). We would like to thank the reviewers for their helpful comments that helped in improving the quality of the manuscript.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] J. Chastre and J.-Y. Fagon, “Ventilator-associated pneumonia,” American journal of respiratory and critical care medicine , vol. 165, no. 7, pp. 867–903, 2002.
- 2[2] P. Johnston, D. F. Mc Auley, and C. M. O’Kane, “Novel pulmonary biomarkers in the diagnosis of vap,” Thorax , vol. 65, no. 3, pp. 190–192, 2010.
- 3[3] V. S. Baselski and R. G. Wunderink, “Bronchoscopic diagnosis of pneumonia.” Clinical microbiology reviews , vol. 7, no. 4, pp. 533–558, 1994.
- 4[4] M. Pierce, D. Yu, and R. Richards-Kortum, “High-resolution fiber-optic microendoscopy for in situ cellular imaging,” Journal of visualized experiments: Jo VE , no. 47, 2011.
- 5[5] D. Shin, M. C. Pierce, A. M. Gillenwater, M. D. Williams, and R. R. Richards-Kortum, “A fiber-optic fluorescence microscope using a consumer-grade digital camera for in vivo cellular imaging,” P Lo S One , vol. 5, no. 6, p. e 11218, 2010.
- 6[6] X. Hong, V. K. Nagarajan, D. H. Mugler, and B. Yu, “Smartphone microendoscopy for high resolution fluorescence imaging,” Journal of Innovative Optical Health Sciences , vol. 9, no. 05, p. 1650046, 2016.
- 7[7] N. Krstajić, A. R. Akram, T. R. Choudhary, N. Mc Donald, M. G. Tanner, E. Pedretti, P. A. Dalgarno, E. Scholefield, J. M. Girkin, A. Moore et al. , “Two-color widefield fluorescence microendoscopy enables multiplexed molecular imaging in the alveolar space of human lung tissue,” Journal of Biomedical Optics , vol. 21, no. 4, pp. 046 009–046 009, 2016.
- 8[8] H. A. Wood, K. Harrington, J. M. Stone, T. A. Birks, and J. C. Knight, “Quantitative characterization of endoscopic imaging fibers,” Optics Express , vol. 25, no. 3, pp. 1985–1992, 2017.
