Symmetry-Aware Gradient Coordination for Physics-Guided Non-Line-of-Sight Imaging

Physics-guided computational imaging typically aggregates data fidelity, geometric reconstruction, and sensor consistency into a single scalar loss. In low signal-to-noise ratio (low-SNR) non-line-of-sight imaging, this centralized approach creates asymmetric gradient conflicts where the dominant constraints suppress physically meaningful updates. We propose treating multi-constraint training as a gradient coordination problem rather than scalar balancing. Our framework coordinates heterogeneous objectives through branch-wise gradient routing: soft conflict projection (PCGrad), hard physical constraint enforcement (PhysGuard), learnable sensor calibration, and a staged training protocol that decouples representation learning from nuisance parameter estimation. On held-out test scenes, the fully staged model improved the peak signal-to-noise ratio (PSNR) from 19.09 dB to 20.49 dB and the structural similarity index (SSIM) from 0.67 to 0.71 over the baseline, with consistent gains across the 48, 28, and 25 dB SNR levels. Qualitative evaluation on seven real-world scenes indicates sharper structure recovery and fewer artifacts. In this NLOS setting, gradient-level coordination is more reliable than scalar aggregation under heterogeneous constraints.