ITK
6.0.0
Insight Toolkit
|
#include <itkGradientDescentOptimizer.h>
Implement a gradient descent optimizer.
GradientDescentOptimizer implements a simple gradient descent optimizer. At each iteration the current position is updated according to
\[ p_{n+1} = p_n + \mbox{learningRate} \, \frac{\partial f(p_n) }{\partial p_n} \]
The learning rate is a fixed scalar defined via SetLearningRate(). The optimizer steps through a user defined number of iterations; no convergence checking is done.
Additionally, user can scale each component, \( \partial f / \partial p \), by setting a scaling vector using method SetScale().
Definition at line 72 of file itkGradientDescentOptimizer.h.
Public Member Functions | |
virtual void | AdvanceOneStep () |
virtual SizeValueType | GetCurrentIteration () const |
virtual const DerivativeType & | GetGradient () const |
virtual const double & | GetLearningRate () const |
const char * | GetNameOfClass () const override |
virtual const SizeValueType & | GetNumberOfIterations () const |
virtual const double & | GetValue () const |
void | ResumeOptimization () |
virtual void | SetLearningRate (double _arg) |
virtual void | SetNumberOfIterations (SizeValueType _arg) |
void | StartOptimization () override |
void | StopOptimization () |
virtual const bool & | GetMaximize () const |
virtual void | SetMaximize (bool _arg) |
virtual void | MaximizeOn () |
bool | GetMinimize () const |
void | SetMinimize (bool v) |
void | MinimizeOn () |
void | MinimizeOff () |
virtual const StopConditionGradientDescentOptimizerEnum & | GetStopCondition () const |
std::string | GetStopConditionDescription () const override |
Public Member Functions inherited from itk::SingleValuedNonLinearOptimizer | |
virtual CostFunctionType * | GetModifiableCostFunction () |
const char * | GetNameOfClass () const override |
MeasureType | GetValue (const ParametersType ¶meters) const |
virtual void | SetCostFunction (CostFunctionType *costFunction) |
const char * | GetNameOfClass () const override |
Public Member Functions inherited from itk::Optimizer | |
virtual const ParametersType & | GetCurrentPosition () const |
virtual const ParametersType & | GetInitialPosition () const |
const char * | GetNameOfClass () const override |
virtual std::string | GetStopConditionDescription () const |
virtual void | SetInitialPosition (const ParametersType ¶m) |
void | SetScales (const ScalesType &scales) |
virtual void | StartOptimization () |
virtual const ScalesType & | GetScales () const |
virtual const ScalesType & | GetInverseScales () const |
Public Member Functions inherited from itk::Object | |
unsigned long | AddObserver (const EventObject &event, Command *cmd) const |
unsigned long | AddObserver (const EventObject &event, std::function< void(const EventObject &)> function) const |
LightObject::Pointer | CreateAnother () const override |
virtual void | DebugOff () const |
virtual void | DebugOn () const |
Command * | GetCommand (unsigned long tag) |
bool | GetDebug () const |
MetaDataDictionary & | GetMetaDataDictionary () |
const MetaDataDictionary & | GetMetaDataDictionary () const |
virtual ModifiedTimeType | GetMTime () const |
const char * | GetNameOfClass () const override |
virtual const TimeStamp & | GetTimeStamp () const |
bool | HasObserver (const EventObject &event) const |
void | InvokeEvent (const EventObject &) |
void | InvokeEvent (const EventObject &) const |
virtual void | Modified () const |
void | Register () const override |
void | RemoveAllObservers () |
void | RemoveObserver (unsigned long tag) const |
void | SetDebug (bool debugFlag) const |
void | SetReferenceCount (int) override |
void | UnRegister () const noexcept override |
void | SetMetaDataDictionary (const MetaDataDictionary &rhs) |
void | SetMetaDataDictionary (MetaDataDictionary &&rrhs) |
virtual void | SetObjectName (std::string _arg) |
virtual const std::string & | GetObjectName () const |
Public Member Functions inherited from itk::LightObject | |
Pointer | Clone () const |
virtual Pointer | CreateAnother () const |
virtual void | Delete () |
virtual const char * | GetNameOfClass () const |
virtual int | GetReferenceCount () const |
void | Print (std::ostream &os, Indent indent=0) const |
virtual void | Register () const |
virtual void | SetReferenceCount (int) |
virtual void | UnRegister () const noexcept |
Static Public Member Functions | |
static Pointer | New () |
Static Public Member Functions inherited from itk::SingleValuedNonLinearOptimizer | |
static Pointer | New () |
Static Public Member Functions inherited from itk::NonLinearOptimizer | |
static Pointer | New () |
Static Public Member Functions inherited from itk::Optimizer | |
static Pointer | New () |
Static Public Member Functions inherited from itk::Object | |
static bool | GetGlobalWarningDisplay () |
static void | GlobalWarningDisplayOff () |
static void | GlobalWarningDisplayOn () |
static Pointer | New () |
static void | SetGlobalWarningDisplay (bool val) |
Static Public Member Functions inherited from itk::LightObject | |
static void | BreakOnError () |
static Pointer | New () |
Protected Member Functions | |
GradientDescentOptimizer () | |
void | PrintSelf (std::ostream &os, Indent indent) const override |
~GradientDescentOptimizer () override=default | |
Protected Member Functions inherited from itk::SingleValuedNonLinearOptimizer | |
void | PrintSelf (std::ostream &os, Indent indent) const override |
SingleValuedNonLinearOptimizer () | |
~SingleValuedNonLinearOptimizer () override=default | |
Protected Member Functions inherited from itk::NonLinearOptimizer | |
NonLinearOptimizer ()=default | |
~NonLinearOptimizer () override | |
Protected Member Functions inherited from itk::Optimizer | |
Optimizer () | |
void | PrintSelf (std::ostream &os, Indent indent) const override |
virtual void | SetCurrentPosition (const ParametersType ¶m) |
~Optimizer () override=default | |
Protected Member Functions inherited from itk::Object | |
Object () | |
bool | PrintObservers (std::ostream &os, Indent indent) const |
void | PrintSelf (std::ostream &os, Indent indent) const override |
virtual void | SetTimeStamp (const TimeStamp &timeStamp) |
~Object () override | |
Protected Member Functions inherited from itk::LightObject | |
virtual LightObject::Pointer | InternalClone () const |
LightObject () | |
virtual void | PrintHeader (std::ostream &os, Indent indent) const |
virtual void | PrintSelf (std::ostream &os, Indent indent) const |
virtual void | PrintTrailer (std::ostream &os, Indent indent) const |
virtual | ~LightObject () |
Protected Attributes | |
DerivativeType | m_Gradient {} |
double | m_LearningRate { 1.0 } |
bool | m_Maximize { false } |
Protected Attributes inherited from itk::SingleValuedNonLinearOptimizer | |
CostFunctionPointer | m_CostFunction {} |
Protected Attributes inherited from itk::Optimizer | |
ParametersType | m_CurrentPosition {} |
bool | m_ScalesInitialized { false } |
Protected Attributes inherited from itk::LightObject | |
std::atomic< int > | m_ReferenceCount {} |
Private Attributes | |
SizeValueType | m_CurrentIteration { 0 } |
SizeValueType | m_NumberOfIterations { 100 } |
bool | m_Stop { false } |
StopConditionGradientDescentOptimizerEnum | m_StopCondition |
std::ostringstream | m_StopConditionDescription {} |
double | m_Value { 0.0 } |
using itk::GradientDescentOptimizer::ConstPointer = SmartPointer<const Self> |
Definition at line 81 of file itkGradientDescentOptimizer.h.
Definition at line 80 of file itkGradientDescentOptimizer.h.
Standard class type aliases.
Definition at line 78 of file itkGradientDescentOptimizer.h.
using itk::GradientDescentOptimizer::StopConditionGradientDescentOptimizerEnum = GradientDescentOptimizerEnums::StopConditionGradientDescentOptimizer |
Definition at line 89 of file itkGradientDescentOptimizer.h.
Definition at line 79 of file itkGradientDescentOptimizer.h.
|
protected |
|
overrideprotecteddefault |
|
virtual |
Advance one step following the gradient direction.
Reimplemented in itk::QuaternionRigidTransformGradientDescentOptimizer.
|
virtual |
Get the current iteration number.
|
virtual |
Get Gradient condition.
|
virtual |
Get the learning rate.
|
virtual |
Methods to configure the cost function.
|
inline |
Methods to configure the cost function.
Definition at line 105 of file itkGradientDescentOptimizer.h.
|
overridevirtual |
Reimplemented from itk::Object.
Reimplemented in itk::QuaternionRigidTransformGradientDescentOptimizer.
|
virtual |
Get the number of iterations.
|
virtual |
Get Stop condition.
|
overridevirtual |
Get Stop condition.
Reimplemented from itk::Optimizer.
|
virtual |
Get the current value.
|
virtual |
Methods to configure the cost function.
|
inline |
Methods to configure the cost function.
Definition at line 120 of file itkGradientDescentOptimizer.h.
|
inline |
Methods to configure the cost function.
Definition at line 115 of file itkGradientDescentOptimizer.h.
|
static |
Method for creation through the object factory.
|
overrideprotectedvirtual |
Methods invoked by Print() to print information about the object including superclasses. Typically not called by the user (use Print() instead) but used in the hierarchical print process to combine the output of several classes.
Reimplemented from itk::Object.
void itk::GradientDescentOptimizer::ResumeOptimization | ( | ) |
Resume previously stopped optimization with current parameters
|
virtual |
Set the learning rate.
|
virtual |
Methods to configure the cost function.
|
inline |
Methods to configure the cost function.
Definition at line 110 of file itkGradientDescentOptimizer.h.
|
virtual |
Set the number of iterations.
|
overridevirtual |
Start optimization.
Reimplemented from itk::Optimizer.
void itk::GradientDescentOptimizer::StopOptimization | ( | ) |
Stop optimization.
|
private |
Definition at line 191 of file itkGradientDescentOptimizer.h.
|
protected |
Definition at line 178 of file itkGradientDescentOptimizer.h.
|
protected |
Definition at line 182 of file itkGradientDescentOptimizer.h.
|
protected |
Definition at line 180 of file itkGradientDescentOptimizer.h.
|
private |
Definition at line 190 of file itkGradientDescentOptimizer.h.
|
private |
Definition at line 185 of file itkGradientDescentOptimizer.h.
|
private |
Definition at line 187 of file itkGradientDescentOptimizer.h.
|
private |
Definition at line 192 of file itkGradientDescentOptimizer.h.
|
private |
Definition at line 186 of file itkGradientDescentOptimizer.h.