Single Snapshot Distillation for Phase Coded Mask Design in Phase Retrieval

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Fonseca, Karen, Suarez-Rodriguez, Leon, Jerez, Andres, Gutierrez-Barragan, Felipe, Arguello, Henry
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918128789749760
author Fonseca, Karen
Suarez-Rodriguez, Leon
Jerez, Andres
Gutierrez-Barragan, Felipe
Arguello, Henry
author_facet Fonseca, Karen
Suarez-Rodriguez, Leon
Jerez, Andres
Gutierrez-Barragan, Felipe
Arguello, Henry
contents Phase retrieval (PR) reconstructs phase information from magnitude measurements, known as coded diffraction patterns (CDPs), whose quality depends on the number of snapshots captured using coded phase masks. High-quality phase estimation requires multiple snapshots, which is not desired for efficient PR systems. End-to-end frameworks enable joint optimization of the optical system and the recovery neural network. However, their application is constrained by physical implementation limitations. Additionally, the framework is prone to gradient vanishing issues related to its global optimization process. This paper introduces a Knowledge Distillation (KD) optimization approach to address these limitations. KD transfers knowledge from a larger, lower-constrained network (teacher) to a smaller, more efficient, and implementable network (student). In this method, the teacher, a PR system trained with multiple snapshots, distills its knowledge into a single-snapshot PR system, the student. The loss functions compare the CPMs and the feature space of the recovery network. Simulations demonstrate that this approach improves reconstruction performance compared to a PR system trained without the teacher's guidance.
format Preprint
id arxiv_https___arxiv_org_abs_2505_18352
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Single Snapshot Distillation for Phase Coded Mask Design in Phase Retrieval
Fonseca, Karen
Suarez-Rodriguez, Leon
Jerez, Andres
Gutierrez-Barragan, Felipe
Arguello, Henry
Image and Video Processing
Phase retrieval (PR) reconstructs phase information from magnitude measurements, known as coded diffraction patterns (CDPs), whose quality depends on the number of snapshots captured using coded phase masks. High-quality phase estimation requires multiple snapshots, which is not desired for efficient PR systems. End-to-end frameworks enable joint optimization of the optical system and the recovery neural network. However, their application is constrained by physical implementation limitations. Additionally, the framework is prone to gradient vanishing issues related to its global optimization process. This paper introduces a Knowledge Distillation (KD) optimization approach to address these limitations. KD transfers knowledge from a larger, lower-constrained network (teacher) to a smaller, more efficient, and implementable network (student). In this method, the teacher, a PR system trained with multiple snapshots, distills its knowledge into a single-snapshot PR system, the student. The loss functions compare the CPMs and the feature space of the recovery network. Simulations demonstrate that this approach improves reconstruction performance compared to a PR system trained without the teacher's guidance.
title Single Snapshot Distillation for Phase Coded Mask Design in Phase Retrieval
topic Image and Video Processing
url https://arxiv.org/abs/2505.18352