Open Access
8 March 2023 Unrolled-DOT: an interpretable deep network for diffuse optical tomography
Yongyi Zhao, Ankit Raghuram, Fay Wang, Stephen Hyunkeol Kim, Andreas H. Hielscher, Jacob T. Robinson, Ashok Veeraraghavan
Author Affiliations +
Abstract

Significance

Imaging through scattering media is critical in many biomedical imaging applications, such as breast tumor detection and functional neuroimaging. Time-of-flight diffuse optical tomography (ToF-DOT) is one of the most promising methods for high-resolution imaging through scattering media. ToF-DOT and many traditional DOT methods require an image reconstruction algorithm. Unfortunately, this algorithm often requires long computational runtimes and may produce lower quality reconstructions in the presence of model mismatch or improper hyperparameter tuning.

Aim

We used a data-driven unrolled network as our ToF-DOT inverse solver. The unrolled network is faster than traditional inverse solvers and achieves higher reconstruction quality by accounting for model mismatch.

Approach

Our model “Unrolled-DOT” uses the learned iterative shrinkage thresholding algorithm. In addition, we incorporate a refinement U-Net and Visual Geometry Group (VGG) perceptual loss to further increase the reconstruction quality. We trained and tested our model on simulated and real-world data and benchmarked against physics-based and learning-based inverse solvers.

Results

In experiments on real-world data, Unrolled-DOT outperformed learning-based algorithms and achieved over 10× reduction in runtime and mean-squared error, compared to traditional physics-based solvers.

Conclusion

We demonstrated a learning-based ToF-DOT inverse solver that achieves state-of-the-art performance in speed and reconstruction quality, which can aid in future applications for noninvasive biomedical imaging.

CC BY: © The Authors. Published by SPIE under a Creative Commons Attribution 4.0 International License. Distribution or reproduction of this work in whole or in part requires full attribution of the original publication, including its DOI.
Yongyi Zhao, Ankit Raghuram, Fay Wang, Stephen Hyunkeol Kim, Andreas H. Hielscher, Jacob T. Robinson, and Ashok Veeraraghavan "Unrolled-DOT: an interpretable deep network for diffuse optical tomography," Journal of Biomedical Optics 28(3), 036002 (8 March 2023). https://doi.org/10.1117/1.JBO.28.3.036002
Received: 19 September 2022; Accepted: 9 February 2023; Published: 8 March 2023
Advertisement
Advertisement
KEYWORDS
Image restoration

Education and training

Reconstruction algorithms

Machine learning

Data modeling

Image quality

Diffuse optical tomography

Back to Top