Exploiting mathematical structures in spectral imaging to accelerate experiments and improve iterative reconstructions
Maike Meier, Hussam Al Daas, Boris Shustin, Lorenzo Lazzarino, and Paul Quinn
Abstract
Spectral imaging covers a range of techniques that aim to reconstruct the chemical state and structural properties of materials. Samples are simultaneously imaged in space, using methods such as microscopy, ptychography, or tomography, as well as resolved in the spectral dimension, giving information about chemical composition. Long acquisition times, sample degradation, and low signal-to-noise ratios plague spectral imaging; what is needed are ways to get more out of the data with fewer experiments.
Although a spectral imaging dataset is necessarily a three-dimensional tensor, its underlying dimension is small. In real space, the sample consists of a small number of spectrally distinguishable components present in each pixel in varying thicknesses. We develop computational routines to exploit this low-dimensionality; both in the experimental design phase, as well as in the reconstruction phase. By leveraging established NLA techniques such as random leverage-score sampling [2], CUR decompositions [5], iterative non-negative factorisation [3], and regularisation, we accelerate spectral imaging experiments through subsampling and improve reconstructions.
For example, consider spectro-microscopy, which combines spectroscopy and microscopy. The underlying structure of a (flattened) dataset is
\[D = \text {Pois}(\mu t), \quad \mu \in \mathbb {R}_{\geq 0}^{n_E\times S}, \quad t\in \mathbb {R}_{\geq 0}^{S\times n_Xn_Y},\]
where \(\mu \) are absorption spectra in \(n_E\) energies of \(S\) different components and \(t\) are thickness maps of the corresponding components in the \(n_Xn_Y\) pixels. We firstly develop a data-driven scheme to determine what entries of the dataset should be measured and what measurements can be skipped, building on [7]. The scheme is based on leverage score sampling important wavelengths and important pixels, determined on-the-fly. The result is a non-negative dataset with 80-95 % missing entries. We show that if the measured entries are structured in columns and rows of the flattened dataset, reconstructions are significantly more accurate than random missing entries.
Furthermore, we provide a novel algorithm to reconstruct a non-negative factorisation from a dataset with missing entries, which is especially efficient for a CUR decomposition of the data. This algorithm also allows regularisation, and leads to more physically-relevant reconstructions than existing methods [4].
We present similar schemes for spectro-ptychography [1], in which the low-rank structure is not readily present in the measured datasets. This leads to a combination of non-negative factorisation schemes combined with nonlinear conjugate gradient algorithms [6].
The work demonstrates that numerical linear algebra results from the last two decades have a wealth of applications beyond straightforward large-scale matrix manipualtions. In particular, the ideas of modern NLA can be applied before any data has been obtained, to aid in experiment design and accelerating acquisition times.
References
-
[1] Batey, D.J., Claus, D. and Rodenburg, J.M., 2014. Information multiplexing in ptychography. Ultramicroscopy, 138, pp.13-21.
-
[2] Drineas, P., Mahoney, M.W. and Muthukrishnan, S., 2006. Sampling algorithms for l2 regression and applications. Proc. Annu. ACM-SIAM Symp. on Discrete Algorithms, pp.1127-1136.
-
[3] Gillis, N., 2020. Nonnegative matrix factorization. SIAM.
-
[4] Lerotic, M., Jacobsen, C., Schäfer, T. and Vogt, S., 2004. Cluster analysis of soft X-ray spectromicroscopy data. Ultramicroscopy, 100(1-2), pp.35-57.
-
[5] Mahoney, M.W. and Drineas, P., 2009. CUR matrix decompositions for improved data analysis. PNAS, 106(3), pp.697-702.
-
[6] Thibault, P. and Guizar-Sicairos, M., 2012. Maximum-likelihood refinement for coherent diffractive imaging. New J. Phys., 14(6), p.063004.
-
[7] Quinn, P.D., Sabaté Landman, M., Davis, T., Freitag, M., Gazzola, S. and Dolgov, S., 2024. Optimal Sparse Energy Sampling for X-ray Spectro-Microscopy: Reducing the X-ray Dose and Experiment Time Using Model Order Reduction. Chem. Biomed. Imaging, 2(4), pp.283-292.