Collaborative Research: CIF: Medium: Robustness to Distribution Shifts in Computational Imaging - Inference, Sampling, and Adaptation

University Of California-Riverside, Riverside CA

Investigators

Abstract

Modern imaging technologies are central to progress in science, medicine, and engineering. Yet, many advanced imaging systems operate under physical or resource limitations that make it difficult to directly acquire high-quality images. Computational imaging addresses these challenges by using algorithms to reconstruct images from incomplete or indirect measurements. In recent years, deep learning has enabled new capabilities in computational imaging, but current methods assume that the training and test data share the same conditions. This assumption often does not hold in real-world settings. This project addresses this critical gap by developing new methods to ensure that deep-learning models for image reconstruction remain reliable and accurate even when the data conditions shift. The outcomes of this research will have broad use and transformative effects across a wide range of scientific, engineering, and biomedical applications, where robust image reconstruction is essential. Broader-impact activities include the organization of special sessions, workshops, and journal issues for the computational-imaging community, dissemination via open-source code, and curriculum development at both institutions. This project focuses on score-based models — a class of deep generative models that solve imaging problems by learning the score function of the image distribution. The central goal is to develop a unified mathematical framework for analyzing and improving the robustness of these models under distribution shifts between training and test data. The project introduces Robust Score-based Inversion (RoSI) as a foundation for (i) quantifying the extent of such shifts using the model's own score function; (ii) characterizing the effect of shifts on reconstruction and sampling performance; and (iii) enabling principled adaptation of models to new imaging settings. The research will be validated in real-world imaging systems, including lensless cameras, computational microscopes, and magnetic resonance imaging, providing both theoretical insights and practical tools for reliable computational imaging. The project will also promote education and engagement in the areas of computational imaging and machine learning. This award reflects NSF's statutory mission and has been deemed worthy of support through evaluation using the Foundation's intellectual merit and broader impacts review criteria.