org.deeplearning4j.nn.layers

Class BasePretrainNetwork<LayerConfT extends BasePretrainNetwork>

java.lang.Object

org.deeplearning4j.nn.layers.AbstractLayer<LayerConfT>

org.deeplearning4j.nn.layers.BaseLayer<LayerConfT>

org.deeplearning4j.nn.layers.BasePretrainNetwork<LayerConfT>

All Implemented Interfaces:

Serializable, Cloneable, Layer, Model, Trainable

Direct Known Subclasses:

AutoEncoder

public abstract class BasePretrainNetwork<LayerConfT extends BasePretrainNetwork>
extends BaseLayer<LayerConfT>

Baseline class for any Neural Network used as a layer in a deep network *

Author:

Adam Gibson

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from interface org.deeplearning4j.nn.api.Layer

Layer.TrainingMode, Layer.Type

Field Summary

Fields inherited from class org.deeplearning4j.nn.layers.BaseLayer

gradient, gradientsFlattened, gradientViews, optimizer, params, paramsFlattened, score, solver, weightNoiseParams

Fields inherited from class org.deeplearning4j.nn.layers.AbstractLayer

cacheMode, conf, dropoutApplied, epochCount, index, input, inputModificationAllowed, iterationCount, maskArray, maskState, preOutput, trainingListeners

Constructor Summary

Constructors

Constructor and Description
BasePretrainNetwork(NeuralNetConfiguration conf)
BasePretrainNetwork(NeuralNetConfiguration conf, INDArray input)

Method Summary

Modifier and Type	Method and Description
Pair<Gradient,INDArray>	backpropGradient(INDArray epsilon, LayerWorkspaceMgr workspaceMgr) Calculate the gradient relative to the error in the next layer
double	calcL1(boolean backpropParamsOnly) Calculate the l1 regularization term 0.0 if regularization is not used.
double	calcL2(boolean backpropParamsOnly) Calculate the l2 regularization term 0.0 if regularization is not used.
protected Gradient	createGradient(INDArray wGradient, INDArray vBiasGradient, INDArray hBiasGradient)
INDArray	getCorruptedInput(INDArray x, double corruptionLevel) Corrupts the given input by doing a binomial sampling given the corruption level
long	numParams() The number of parameters for the model, for backprop (i.e., excluding visible bias)
long	numParams(boolean backwards) the number of parameters for the model
INDArray	params() Returns the parameters of the neural network as a flattened row vector
Map<String,INDArray>	paramTable(boolean backpropParamsOnly) Table of parameters by key, for backprop For many models (dense layers, etc) - all parameters are backprop parameters
abstract Pair<INDArray,INDArray>	sampleHiddenGivenVisible(INDArray v) Sample the hidden distribution given the visible
abstract Pair<INDArray,INDArray>	sampleVisibleGivenHidden(INDArray h) Sample the visible distribution given the hidden
void	setParams(INDArray params) Set the parameters for this model.
protected void	setScoreWithZ(INDArray z)

Methods inherited from class org.deeplearning4j.nn.layers.BaseLayer

activate, clear, clearNoiseWeightParams, clone, computeGradientAndScore, fit, fit, getGradientsViewArray, getOptimizer, getParam, getParamWithNoise, gradient, hasBias, layerConf, paramTable, preOutput, score, setBackpropGradientsViewArray, setParam, setParams, setParamsViewArray, setParamTable, toString, update, update

Methods inherited from class org.deeplearning4j.nn.layers.AbstractLayer

activate, addListeners, allowInputModification, applyConstraints, applyDropOutIfNecessary, applyMask, assertInputSet, backpropDropOutIfPresent, batchSize, conf, feedForwardMaskArray, getConfig, getEpochCount, getHelper, getIndex, getInput, getInputMiniBatchSize, getListeners, getMaskArray, gradientAndScore, init, input, layerId, setCacheMode, setConf, setEpochCount, setIndex, setInput, setInputMiniBatchSize, setListeners, setListeners, setMaskArray, type, updaterDivideByMinibatch

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.deeplearning4j.nn.api.Layer

getIterationCount, isPretrainLayer, setIterationCount

Constructor Detail

BasePretrainNetwork

public BasePretrainNetwork(NeuralNetConfiguration conf)

BasePretrainNetwork

public BasePretrainNetwork(NeuralNetConfiguration conf,
                           INDArray input)

Method Detail

getCorruptedInput

public INDArray getCorruptedInput(INDArray x,
                                  double corruptionLevel)

Corrupts the given input by doing a binomial sampling given the corruption level

Parameters:

x - the input to corrupt

corruptionLevel - the corruption value

Returns:

the binomial sampled corrupted input

createGradient

protected Gradient createGradient(INDArray wGradient,
                                  INDArray vBiasGradient,
                                  INDArray hBiasGradient)

numParams

public long numParams(boolean backwards)

Description copied from interface: Model

the number of parameters for the model

Specified by:

numParams in interface Model

Overrides:

numParams in class AbstractLayer<LayerConfT extends BasePretrainNetwork>

Returns:

the number of parameters for the model

sampleHiddenGivenVisible

public abstract Pair<INDArray,INDArray> sampleHiddenGivenVisible(INDArray v)

Sample the hidden distribution given the visible

Parameters:

v - the visible to sample from

Returns:

the hidden mean and sample

sampleVisibleGivenHidden

public abstract Pair<INDArray,INDArray> sampleVisibleGivenHidden(INDArray h)

Sample the visible distribution given the hidden

Parameters:

h - the hidden to sample from

Returns:

the mean and sample

setScoreWithZ

protected void setScoreWithZ(INDArray z)

Overrides:

setScoreWithZ in class BaseLayer<LayerConfT extends BasePretrainNetwork>

paramTable

public Map<String,INDArray> paramTable(boolean backpropParamsOnly)

Description copied from interface: Model

Table of parameters by key, for backprop For many models (dense layers, etc) - all parameters are backprop parameters

Specified by:

paramTable in interface Model

Specified by:

paramTable in interface Trainable

Overrides:

paramTable in class BaseLayer<LayerConfT extends BasePretrainNetwork>

Parameters:

backpropParamsOnly - If true, return backprop params only. If false: return all params (equivalent to paramsTable())

Returns:

Parameter table

params

public INDArray params()

Description copied from class: BaseLayer

Returns the parameters of the neural network as a flattened row vector

Specified by:

params in interface Model

Specified by:

params in interface Trainable

Overrides:

params in class BaseLayer<LayerConfT extends BasePretrainNetwork>

Returns:

the parameters of the neural network

numParams

public long numParams()

The number of parameters for the model, for backprop (i.e., excluding visible bias)

Specified by:

numParams in interface Model

Specified by:

numParams in interface Trainable

Overrides:

numParams in class BaseLayer<LayerConfT extends BasePretrainNetwork>

Returns:

the number of parameters for the model (ex. visible bias)

setParams

public void setParams(INDArray params)

Description copied from interface: Model

Set the parameters for this model. This expects a linear ndarray which then be unpacked internally relative to the expected ordering of the model

Specified by:

setParams in interface Model

Overrides:

setParams in class BaseLayer<LayerConfT extends BasePretrainNetwork>

Parameters:

params - the parameters for the model

backpropGradient

public Pair<Gradient,INDArray> backpropGradient(INDArray epsilon,
                                                LayerWorkspaceMgr workspaceMgr)

Description copied from interface: Layer

Calculate the gradient relative to the error in the next layer

Specified by:

backpropGradient in interface Layer

Overrides:

backpropGradient in class BaseLayer<LayerConfT extends BasePretrainNetwork>

Parameters:

epsilon - w^(L+1)*delta^(L+1). Or, equiv: dC/da, i.e., (dC/dz)*(dz/da) = dC/da, where C is cost function a=sigma(z) is activation.

workspaceMgr - Workspace manager

Returns:

Pair where Gradient is gradient for this layer, INDArray is epsilon (activation gradient) needed by next layer, but before element-wise multiply by sigmaPrime(z). So for standard feed-forward layer, if this layer is L, then return.getSecond() == dL/dIn = (w^(L)*(delta^(L))^T)^T. Note that the returned array should be placed in the ArrayType.ACTIVATION_GRAD workspace via the workspace manager

calcL2

public double calcL2(boolean backpropParamsOnly)

Description copied from interface: Layer

Calculate the l2 regularization term
0.0 if regularization is not used. Or 0.5 * l2Coeff * l2Magnitude otherwise.
Note that this does not divide by mini-batch size

Specified by:

calcL2 in interface Layer

Overrides:

calcL2 in class BaseLayer<LayerConfT extends BasePretrainNetwork>

Parameters:

backpropParamsOnly - If true: calculate L2 based on backprop params only. If false: calculate based on all params (including pretrain params, if any)

Returns:

the l2 regularization term for this layer.

calcL1

public double calcL1(boolean backpropParamsOnly)

Description copied from interface: Layer

Calculate the l1 regularization term
0.0 if regularization is not used. Or l1Coeff * l1Magnitude otherwise.
Note that this does not divide by mini-batch size

Specified by:

calcL1 in interface Layer

Overrides:

calcL1 in class BaseLayer<LayerConfT extends BasePretrainNetwork>

Parameters:

backpropParamsOnly - If true: calculate L1 based on backprop params only. If false: calculate based on all params (including pretrain params, if any)

Returns:

the l1 regularization term for this layer.