Monte-Carlo based efficient image reconstruction in coherent imaging with speckle noise

Preview

There are 4 python files in this repo. The test data is under the /data/ folder.

• train_DIP_deblur_ite_MC.py: training DIP based PGD algorithm.

• function_grad.py: i) Forward operator in coherent imaging, and ii) implementation of Monte-Carlo sampling and conjugate gradient methods to avoid matrix inverse in MLE based optimization.

• decoder.py: basic network structures of the Deep Image Prior/Deep Decoder we use for projection.

• utils.py: all the other helper functions.

Running the code

Run the MLE-based PGD algorithm (efficient Monte-Carlo and conjugate gradient methods) for recovering images:

python train_DIP_deblur_ite_MC.py

Specify the hyperparameters and experiment setting:

E.g., recover images from measurements with number of looks L=1, aperture percentage=0.8, additive noise level=50, Monte-Carlo samples=10:

python train_DIP_deblur_ite_MC.py --dataset 'Set11' --mask_rate 0.8 --add_std 0.2 --num_look 1 --lr_NN 1e-3 --lr_GD 0.01 --outer_ite 100 --MC True --num_ite_MC 10

Relevant works on image reconstruction in coherent imaging with speckle

[1] Chen, Xi, Soham Jana, Christopher Metzler, Arian Maleki, and Shirin Jalali. "Multilook Coherent Imaging: Theoretical Guarantees and Algorithms." arXiv preprint arXiv:2505.23594 (2025) paper

[2] Chen, Xi, Christopher Metzler, Arian Maleki, and Shirin Jalali. "Chen, Xi, et al. "Efficient multilook coherent imaging with temporally dependent speckle noise." Unconventional Imaging, Sensing, and Adaptive Optics 2025. Vol. 13619. SPIE, 2025. paper

[3] Chen, Xi, Christopher Metzler, Arian Maleki, and Shirin Jalali. "Monte-Carlo Based Efficient Image Reconstruction in Coherent Imaging With Speckle Noise." 2025 IEEE 22nd International Symposium on Biomedical Imaging (ISBI). IEEE, 2025. paper

[4] Chen, Xi, Christopher Metzler, Arian Maleki, and Shirin Jalali. "Novel approach to coherent imaging in the presence of speckle noise." Unconventional Imaging, Sensing, and Adaptive Optics 2024. Vol. 13149. SPIE, 2024. paper

[5] Chen, Xi, Zhewen Hou, Christopher Metzler, Arian Maleki, and Shirin Jalali. "Bagged Deep Image Prior for Recovering Images in the Presence of Speckle Noise." Forty-first International Conference on Machine Learning (ICML 2024). paper

[6] Chen, Xi, Zhewen Hou, Christopher Metzler, Arian Maleki, and Shirin Jalali. "Multilook compressive sensing in the presence of speckle noise." In NeurIPS 2023 Workshop on Deep Learning and Inverse Problems. 2023. paper

[7] Zhou, Wenda, Shirin Jalali, and Arian Maleki. "Compressed sensing in the presence of speckle noise." IEEE Transactions on Information Theory 68.10 (2022): 6964-6980. paper