ott.geometry.regularizers.STVS

Contents

ott.geometry.regularizers.STVS#

class ott.geometry.regularizers.STVS(gamma=1.0)[source]#

Soft thresholding operator with vanishing shrinkage regularizer [Schreck et al., 2016].

The operator is defined as:

\[\gamma^2 \mathbf{1}_d^T \left(\sigma(x) - \frac{1}{2} \exp\left(-2\sigma(x)\right) + \frac{1}{2}\right)\]

where \(\sigma(x) := \text{asinh}\left(\frac{x}{2\gamma}\right)\).

Parameters:

gamma (float) – Strength of the regularization.

Methods

moreau_envelope(x[, tau])

Moreau Envelope.

prox(v[, tau])

Proximal operator.

prox_dual(v[, tau])

Proximal operator of the convex conjugate.