#include <FIRE.h>

Inheritance diagram for FIRE:

Public Member Functions
	FIRE (std::shared_ptr< ObjectiveFunction > a_objf, const Parameters &a_params)
virtual	~FIRE ()=default
int	step (double a_maxMove) override
int	run (size_t a_maxIterations, double a_maxMove) override
Public Member Functions inherited from eonc::Optimizer
	Optimizer (std::shared_ptr< ObjectiveFunction > a_objf, const OptimizerConfig &a_config)
	Optimizer (std::shared_ptr< ObjectiveFunction > a_objf, OptType a_optype, const OptimizerConfig &a_config)
	Optimizer (std::shared_ptr< ObjectiveFunction > a_objf, const Parameters &a_params)
	Optimizer (std::shared_ptr< ObjectiveFunction > a_objf, OptType a_optype, const Parameters &a_params)
virtual	~Optimizer ()

Private Attributes
double	m_dt
double	m_dt_max
double	m_max_move
size_t	m_N_min
size_t	m_N
Eigen::VectorXd	m_vel
double	m_alpha_start
double	m_alpha
double	m_f_inc
double	m_f_dec
double	m_f_a
size_t	m_iteration
eonc::log::FileScoped	m_log {"fire", "_fire.log"}

Additional Inherited Members
Protected Attributes inherited from eonc::Optimizer
const OptimizerConfig	m_optConfig
std::shared_ptr< ObjectiveFunction >	m_objf

Detailed Description

Definition at line 19 of file FIRE.h.

Constructor & Destructor Documentation

◆ FIRE()

eonc::FIRE::FIRE	(	std::shared_ptr< ObjectiveFunction >	a_objf,
		const Parameters &	a_params )

inline

Definition at line 22 of file FIRE.h.

      : Optimizer(a_objf, OptType::FIRE, OptimizerConfig::fromParams(a_params)),
        m_dt{a_params.optimizer_options.time_step},
        m_dt_max{a_params.optimizer_options.max_time_step},
        m_max_move{a_params.optimizer_options.max_move},
        m_N_min{5},
        m_N{0},
        m_vel{Eigen::VectorXd::Zero(a_objf->degreesOfFreedom())},
        m_alpha_start{0.1},
        m_alpha{m_alpha_start},
        m_f_inc{1.1},
        m_f_dec{0.5},
        m_f_a{0.99},
        m_iteration{0} {}

◆ ~FIRE()

virtual eonc::FIRE::~FIRE ( )

virtualdefault

Member Function Documentation

◆ run()

int FIRE::run	(	size_t	a_maxIterations,
		double	a_maxMove )

overridevirtual

Implements eonc::Optimizer.

Definition at line 70 of file FIRE.cpp.

                                                      {
  while (!m_objf->isConverged() && m_iteration < a_maxIterations) {
    step(a_maxMove);
  }
  return m_objf->isConverged() ? 1 : 0;
}

◆ step()

int FIRE::step ( double a_maxMove )

overridevirtual

Implements eonc::Optimizer.

Definition at line 19 of file FIRE.cpp.

                               {
  double P = 0;
  // Check convergence.
  if (m_objf->isConverged()) {
    return 1;
  }
 
  // Velocity Verlet
  Eigen::VectorXd f = -m_objf->getGradient();
  Eigen::VectorXd x = m_objf->getPositions();
 
  m_vel += f * m_dt;
  Eigen::VectorXd dx = m_vel * m_dt;
 
  dx = eonc::helpers::maxAtomMotionAppliedV(dx, m_max_move);
  m_objf->setPositions(x + dx);
 
  f = -m_objf->getGradient();
  Eigen::VectorXd f_unit = f / f.norm();
 
  // FIRE
  P = f.dot(m_vel);
  m_vel = (1 - m_alpha) * m_vel + m_alpha * f_unit * m_vel.norm();
  QUILL_LOG_DEBUG(m_log,
                  "P: {:.4f}, v: {:.4f}, m_dt: {:.4f}, m_alpha: {:.4f}, N: {}",
                  P, m_vel.norm(), m_dt, m_alpha, m_N);
  if (P >= 0) {
    m_N++;
    if (m_N > m_N_min) {
      m_dt = std::min(m_dt * m_f_inc, m_dt_max);
      m_alpha = m_alpha * m_f_a;
    }
  } else {
    m_dt = m_dt * m_f_dec;
    m_vel.setZero();
    m_alpha = m_alpha_start;
    m_N = 0;
  }
 
  // add a sanity check on m_dt
  if (m_dt < 1e-6) {
    QUILL_LOG_CRITICAL(m_log, "[FIRE] [critical] m_dt is too small: {:.4f}",
                       m_dt);
    throw std::runtime_error(
        std::format("[FIRE] m_dt is too small: {:.4f}", m_dt));
  }
 
  m_iteration++;
  return m_objf->isConverged() ? 1 : 0;
}