Skip to content

← deep_learning package

NoisySGDStep

The NoisySGD optimizer step.

Based on Neelakantan et al, 2014, noisy SGD is a variant of stochastic gradient descent (SGD) that adds additional Gaussian-distributed noise. This can help generalization in particularly deep networks. Like all step nodes, this node only processes gradients, and the resulting updates must be applied manually to the weights (this can be accomplished using the Add node). However, you can also pass it to the StepSolver node which implements the full optimization loop. The learning rate can instead be given as a schedule, by wiring one of the Schedule nodes into the learning_rate_schedule port. More Info... Version 0.2.0

Ports/Properties

gradients

Gradients to be transformed.

  • verbose name: Gradients
  • default value: None
  • port type: DataPort
  • value type: object (can be None)
  • data direction: INOUT

weights

Optional current weights.

  • verbose name: Weights
  • default value: None
  • port type: DataPort
  • value type: object (can be None)
  • data direction: IN

state

Explicit state of the node.

  • verbose name: State
  • default value: None
  • port type: DataPort
  • value type: object (can be None)
  • data direction: INOUT

learning_rate_schedule

Optional learning rate schedule.

  • verbose name: Learning Rate Schedule
  • default value: None
  • port type: DataPort
  • value type: BaseNode (can be None)
  • data direction: IN

eta

Initial variance for the Gaussian noise added to gradients.

  • verbose name: Eta
  • default value: 0.01
  • port type: FloatPort
  • value type: float (can be None)

gamma

A parameter controlling the annealing of noise over time, the variance decays according to (1+t)^-gamma.

  • verbose name: Gamma
  • default value: 0.55
  • port type: FloatPort
  • value type: float (can be None)

seed

A seed for the pseudo-random number generation.

  • verbose name: Seed
  • default value: 12345
  • port type: IntPort
  • value type: int (can be None)

set_breakpoint

Set a breakpoint on this node. If this is enabled, your debugger (if one is attached) will trigger a breakpoint.

  • verbose name: Set Breakpoint (Debug Only)
  • default value: False
  • port type: BoolPort
  • value type: bool (can be None)

metadata

User-definable meta-data associated with the node. Usually reserved for technical purposes.

  • verbose name: Metadata
  • default value: {}
  • port type: DictPort
  • value type: dict (can be None)