animaBoundedLevenbergMarquardtOptimizer.h

Go to the documentation of this file.

#pragma once

#include <itkMultipleValuedNonLinearOptimizer.h>
#include <animaBLMLambdaCostFunction.h>
#include "AnimaOptimizersExport.h"

namespace anima
{

class ANIMAOPTIMIZERS_EXPORT BoundedLevenbergMarquardtOptimizer:
        public itk::MultipleValuedNonLinearOptimizer
{
public:
    typedef BoundedLevenbergMarquardtOptimizer Self;
    typedef itk::MultipleValuedNonLinearOptimizer Superclass;
    typedef itk::SmartPointer <Self> Pointer;
    typedef itk::SmartPointer <const Self> ConstPointer;

    itkNewMacro(Self)

    typedef Superclass::CostFunctionType CostFunctionType;
    typedef Superclass::MeasureType MeasureType;

    itkTypeMacro(BoundedLevenbergMarquardtOptimizer, MultipleValuedNonLinearOptimizer)

    void StartOptimization() ITK_OVERRIDE;

    itkSetMacro(NumberOfIterations, unsigned int)
    itkSetMacro(ValueTolerance, double)
    itkSetMacro(CostTolerance, double)

    itkGetMacro(CurrentValue, double)

    itkSetMacro(LowerBounds, ParametersType)
    itkSetMacro(UpperBounds, ParametersType)

protected:
    BoundedLevenbergMarquardtOptimizer()
    {
        m_NumberOfIterations = 2000;
        m_ValueTolerance = 1e-8;
        m_CostTolerance = 1e-5;
        m_LambdaParameter = 1.0e-8;
        m_DeltaParameter = 0.0;

        m_CurrentValue = 0.0;
        m_LambdaCostFunction = anima::BLMLambdaCostFunction::New();
    }

    virtual ~BoundedLevenbergMarquardtOptimizer() ITK_OVERRIDE {}

    double EvaluateCostFunctionAtParameters(ParametersType &parameters, MeasureType &residualValues);

    bool CheckSolutionIsInBounds(ParametersType &solutionVector, ParametersType &lowerBounds,
                                 ParametersType &upperBounds, unsigned int rank);

    void UpdateLambdaParameter(DerivativeType &derivative, ParametersType &dValues,
                               std::vector <unsigned int> &pivotVector,
                               ParametersType &qtResiduals, unsigned int rank);

    bool CheckConditions(unsigned int numIterations, ParametersType &newParams,
                         ParametersType &oldParams, ParametersType &dValues, double newCostValue);

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(BoundedLevenbergMarquardtOptimizer);

    unsigned int m_NumberOfIterations;
    double m_ValueTolerance;
    double m_CostTolerance;

    double m_LambdaParameter;
    double m_DeltaParameter;
    double m_CurrentValue;

    anima::BLMLambdaCostFunction::Pointer m_LambdaCostFunction;
    ParametersType m_LowerBounds, m_UpperBounds;
    ParametersType m_CurrentAddonVector;
    MeasureType m_ResidualValues;
};

} // end namespace anima