LevenbergMarquardtFitter.h

Go to the documentation of this file.

/********************************************************************
**                Image Component Library (ICL)                    **
**                                                                 **
** Copyright (C) 2006-2013 CITEC, University of Bielefeld          **
**                         Neuroinformatics Group                  **
** Website: www.iclcv.org and                                      **
**          http://opensource.cit-ec.de/projects/icl               **
**                                                                 **
** File   : ICLMath/src/ICLMath/LevenbergMarquardtFitter.h         **
** Module : ICLMath                                                **
** Authors: Christof Elbrechter                                    **
**                                                                 **
**                                                                 **
** GNU LESSER GENERAL PUBLIC LICENSE                               **
** This file may be used under the terms of the GNU Lesser General **
** Public License version 3.0 as published by the                  **
**                                                                 **
** Free Software Foundation and appearing in the file LICENSE.GPL  **
** included in the packaging of this file.  Please review the      **
** following information to ensure the license requirements will   **
** be met: http://www.gnu.org/licenses/lgpl-3.0.txt                **
**                                                                 **
** The development of this software was supported by the           **
** Excellence Cluster EXC 277 Cognitive Interaction Technology.    **
** The Excellence Cluster EXC 277 is a grant of the Deutsche       **
** Forschungsgemeinschaft (DFG) in the context of the German       **
** Excellence Initiative.                                          **
**                                                                 **
********************************************************************/

#pragma once

#include <ICLMath/DynVector.h>            
#include <ICLUtils/Function.h>

namespace icl{
  namespace math{
    
    

    template<class Scalar>
    class LevenbergMarquardtFitter{
      public:
      
      typedef DynColVector<Scalar> Vector; 
      typedef DynColVector<Scalar> Params; 
      typedef DynMatrix<Scalar> Matrix;    
    
      struct Result{ 
        int iteration;                
        Scalar error;                 
        std::vector<Scalar> lambdas;  
        Params params;                
      
        friend inline std::ostream &operator<<(std::ostream &str, const Result &d){
          return str << "iteration: " << d.iteration << "  error:" << d.error 
                     << "  lambda[0]:" << d.lambdas[0] << "  params:" << d.params.transp();
        }
      };

      struct Data{
        Matrix x;  
        Matrix y;  
      };

      typedef icl::utils::Function<Vector,const Params&,const Vector&> Function;
    

      typedef icl::utils::Function<void,const Params&, const Vector&, Vector &> Jacobian;
    
      typedef icl::utils::Function<void,const Result&> DebugCallback;
    
    

      protected:
      Function f;  

      bool useMultiThreading;   
      Scalar initialLambda;     
      int maxIterations;        
      Scalar minError;          
      Scalar lambdaMultiplier;  
      std::string linSolver;    

      Vector dst;        
      Matrix J;          
      Matrix H;          

      std::vector<Jacobian> js;
      
      DebugCallback dbg;  
      Params params_new;  
      Matrix y_est;       
      //Matrix y_est_tmp;   //!< current estimated outputs (with temporary update)
      Vector dy;          


      Scalar error(const Matrix &ys, const Matrix &y_est) const;

      public:
    
      LevenbergMarquardtFitter();
    

      LevenbergMarquardtFitter(Function f, int outputDim,
          const std::vector<Jacobian> &js=std::vector<Jacobian>(),
          Scalar initialLambda=1.e-8, int maxIterations=200,
          Scalar minError = 1.e-6, Scalar lambdaMultiplier=10,
          const std::string &linSolver="svd");
    

      void setUseMultiThreading(bool enable);
      

      void init(Function f, int outputDim,
                const std::vector<Jacobian> &js=std::vector<Jacobian>(),
                Scalar initialLambda=1.e-8, int maxIterations=1000,
                Scalar minError = 1.e-6, Scalar lambdaMultiplier=10,
                const std::string &linSolver="svd");
    

      Result fit(const Matrix &xs, const Matrix &ys, Params initParams);
    
      
    

      static Jacobian create_numerical_jacobian(int o, Function f, float delta=1.E-5);
      
      static std::vector<Jacobian> create_numerical_jacobians(int n, Function f, float delta=1.e-5);


      static Data create_data(const Params &p, Function f, int xDim, int yDim,
                              int num=1000, Scalar minX=-5, Scalar maxX=5);

      static void default_debug_callback(const Result &r);
      
      void setDebugCallback(DebugCallback dbg=default_debug_callback);
    };
  } // namespace math
} // namespace icl