deepinv.loss

This package provides a collection of supervised and self-supervised loss functions for training reconstruction networks. Refer to the user guide for more information.

Base class

User Guide: refer to Training Losses for more information.

deepinv.loss.Loss	Base class for all loss functions.
deepinv.loss.StackedPhysicsLoss	Loss function for stacked physics operators.

Supervised Learning

User Guide: refer to Supervised Learning for more information.

deepinv.loss.SupLoss

Standard supervised loss

Self-Supervised Learning

User Guide: refer to Self-Supervised Learning for more information.

deepinv.loss.MCLoss	Measurement consistency loss
deepinv.loss.EILoss	Equivariant imaging self-supervised loss.
deepinv.loss.MOILoss	Multi-operator imaging loss
deepinv.loss.MOEILoss	Multi-operator equivariant imaging.
deepinv.loss.Neighbor2Neighbor	Neighbor2Neighbor loss.
deepinv.loss.SplittingLoss	Measurement splitting loss.
deepinv.loss.SureGaussianLoss	SURE loss for Gaussian noise
deepinv.loss.SurePoissonLoss	SURE loss for Poisson noise
deepinv.loss.SurePGLoss	SURE loss for Poisson-Gaussian noise
deepinv.loss.TVLoss	Total variation loss (\(\ell_2\) norm).
deepinv.loss.R2RLoss	Generalized Recorrupted-to-Recorrupted (GR2R) Loss
deepinv.loss.ScoreLoss	Learns score of distribution in the context of Noise2Score.
deepinv.loss.AugmentConsistencyLoss	Data augmentation consistency (DAC) loss.

Specialized self-supervised losses for MRI

User Guide: refer to Specialized self-supervised losses for MRI for more information.

deepinv.loss.mri.WeightedSplittingLoss	K-Weighted Splitting Loss
deepinv.loss.mri.RobustSplittingLoss	Robust Weighted Splitting Loss
deepinv.loss.mri.Phase2PhaseLoss	Phase2Phase loss for dynamic data.
deepinv.loss.mri.Artifact2ArtifactLoss	Artifact2Artifact loss for dynamic data.

Adversarial Learning

User Guide: refer to Adversarial Learning for more information.

deepinv.loss.adversarial.DiscriminatorMetric	Generic GAN discriminator metric building block.
deepinv.loss.adversarial.GeneratorLoss	Base generator adversarial loss.
deepinv.loss.adversarial.DiscriminatorLoss	Base discriminator adversarial loss.
deepinv.loss.adversarial.SupAdversarialGeneratorLoss	Supervised adversarial consistency loss for generator.
deepinv.loss.adversarial.SupAdversarialDiscriminatorLoss	Supervised adversarial consistency loss for discriminator.
deepinv.loss.adversarial.UnsupAdversarialGeneratorLoss	Unsupervised adversarial consistency loss for generator.
deepinv.loss.adversarial.UnsupAdversarialDiscriminatorLoss	Unsupervised adversarial consistency loss for discriminator.
deepinv.loss.adversarial.UAIRGeneratorLoss	Reimplementation of UAIR generator's adversarial loss.

Network Regularization

User Guide: refer to Network Regularization for more information.

deepinv.loss.JacobianSpectralNorm	Computes the spectral norm of the Jacobian.
deepinv.loss.FNEJacobianSpectralNorm	Computes the Firm-Nonexpansiveness Jacobian spectral norm.

Loss schedulers

User Guide: refer to Loss schedulers for more information.

deepinv.loss.BaseLossScheduler	Base class for loss schedulers.
deepinv.loss.RandomLossScheduler	Schedule losses at random.
deepinv.loss.InterleavedLossScheduler	Schedule losses sequentially one-by-one.
deepinv.loss.InterleavedEpochLossScheduler	Schedule losses sequentially epoch-by-epoch.
deepinv.loss.StepLossScheduler	Activate losses at specified epoch.