tick.solver.
SAGA
(step: float = None, epoch_size: int = None, rand_type: str = 'unif', tol: float = 0.0, max_iter: int = 100, verbose: bool = True, print_every: int = 10, record_every: int = 1, seed: int = -1, n_threads: int = 1)[source]¶Stochastic Average Gradient solver, for the minimization of objectives of the form
where the functions \(f_i\) have smooth gradients and \(g\) is
prox-capable. Note that SAGA
works only
with linear models, see tick.linear_model and tick.robust, where all
linear models are listed.
Function \(f = \frac 1n \sum_{i=1}^n f_i\) corresponds
to the model.loss
method of the model (passed with set_model
to the
solver) and \(g\) corresponds to the prox.value
method of the
prox (passed with the set_prox
method).
One iteration of SAGA
corresponds to the
following iteration applied epoch_size
times:
where \(i\) is sampled at random (strategy depends on rand_type
) at
each iteration, and where \(\bar w\) and \(\nabla f(\bar w)\)
are updated at the beginning of each epoch. The step-size \(\eta\) can be
tuned with step
, the seed of the random number generator for generation
of samples \(i\) can be seeded with seed
. The iterations stop
whenever tolerance tol
is achieved, or after max_iter
epochs
(namely max_iter
\(\times\) epoch_size
iterates).
The obtained solution \(w\) is returned by the solve
method, and is
also stored in the solution
attribute of the solver.
Internally, SAGA
has dedicated code when
the model is a generalized linear model with sparse features, and a
separable proximal operator: in this case, each iteration works only in the
set of non-zero features, leading to much faster iterates.
step : float
Step-size parameter, the most important parameter of the solver. Whenever possible, this can be automatically tuned as
step = 1 / model.get_lip_max()
. Otherwise, use a try-an-improve approach
tol : float
, default=1e-10
The tolerance of the solver (iterations stop when the stopping criterion is below it)
max_iter : int
, default=10
Maximum number of iterations of the solver, namely maximum number of epochs (by default full pass over the data, unless
epoch_size
has been modified from default)
verbose : bool
, default=True
If
True
, solver verboses history, otherwise nothing is displayed, but history is recorded anyway
seed : int
, default=-1
The seed of the random sampling. If it is negative then a random seed (different at each run) will be chosen.
epoch_size : int
, default given by model
Epoch size, by default, this is automatically tuned using information from the model object passed through
set_model
.
rand_type : {‘unif’, ‘perm’}, default=’unif’
How samples are randomly selected from the data
if
'unif'
samples are uniformly drawn among all possibilitiesif
'perm'
a random permutation of all possibilities is generated and samples are sequentially taken from it. Once all of them have been taken, a new random permutation is generated
print_every : int
, default=1
Print history information every time the iteration number is a multiple of
print_every
. Used only isverbose
is True
record_every : int
, default=1
Save history information every time the iteration number is a multiple of
record_every
n_threads : int
, default=1
Number of threads to use for parallel optimization. The strategy used for this is asynchronous updates of the iterates.
model : Model
The model used by the solver, passed with the
set_model
method
prox : Prox
Proximal operator used by the solver, passed with the
set_prox
method
solution : numpy.array
, shape=(n_coeffs,)
Minimizer found by the solver
history : dict
-like
A dict-type of object that contains history of the solver along iterations. It should be accessed using the
get_history
method
time_start : str
Start date of the call to
solve()
time_elapsed : float
Duration of the call to
solve()
, in seconds
time_end : str
End date of the call to
solve()
dtype : {'float64', 'float32'}
, default=’float64’
Type of the arrays used. This value is set from model and prox dtypes.
References
A. Defazio, F. Bach, S. Lacoste-Julien, SAGA: A fast incremental gradient method with support for non-strongly convex composite objectives, NIPS 2014
R. Leblond, F. Pedregosa, and S. Lacoste-Julien: Asaga: Asynchronous Parallel Saga, (AISTATS) 2017
__init__
(step: float = None, epoch_size: int = None, rand_type: str = 'unif', tol: float = 0.0, max_iter: int = 100, verbose: bool = True, print_every: int = 10, record_every: int = 1, seed: int = -1, n_threads: int = 1)[source]¶Initialize self. See help(type(self)) for accurate signature.
get_history
(key=None)¶Returns history of the solver
key : str
, default=None
If
None
all history is returned as adict
If
str
, name of the history element to retrieve
output : list
or dict
If
key
is None orkey
is not in history then output is a dict containing history of all keysIf
key
is the name of an element in the history, output is alist
containing the history of this element
objective
(coeffs, loss: float = None)¶Compute the objective function
coeffs : np.array
, shape=(n_coeffs,)
Point where the objective is computed
loss : float
, default=`None`
Gives the value of the loss if already known (allows to avoid its computation in some cases)
output : float
Value of the objective at given
coeffs
set_model
(model: tick.base_model.model_generalized_linear.ModelGeneralizedLinear)[source]¶Set model in the solver
model : ModelGeneralizedLinear
Sets the model in the solver. The model gives the first order information about the model (loss, gradient, among other things). SAGA only accepts childs of
ModelGeneralizedLinear
output : Solver
The
Solver
with given model
set_prox
(prox: tick.prox.base.prox.Prox)¶Set proximal operator in the solver
prox : Prox
The proximal operator of the penalization function
output : Solver
The solver with given prox
Notes
In some solvers, set_model
must be called before
set_prox
, otherwise and error might be raised
solve
(x0=None, step=None)¶Launch the solver
x0 : np.array
, shape=(n_coeffs,), default=`None`
Starting point of the solver
step : float
, default=`None`
Step-size or learning rate for the solver. This can be tuned also using the
step
attribute
output : np.array
, shape=(n_coeffs,)
Obtained minimizer for the problem, same as
solution
attribute
tick.solver.SAGA
¶