Homec4science
Diffusion tamaas (archive/legacy_mpi) Browse

tamaas/src/solvers/dfsane_solver.cpp
b0c44a420f99archive/legacy_mpi

Last Change
Raw File

dfsane_solver.cpp
View Options

/*
* SPDX-License-Indentifier: AGPL-3.0-or-later
*
* Copyright (©) 2016-2024 EPFL (École Polytechnique Fédérale de Lausanne),
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
* Copyright (©) 2020-2024 Lucas Frérot
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published
* by the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#include "dfsane_solver.hh"
/* -------------------------------------------------------------------------- */
namespace tamaas {
/* -------------------------------------------------------------------------- */
DFSANESolver::DFSANESolver(Residual& residual)
: EPSolver(residual),
search_direction(residual.getVector().dataSize(),
residual.getVector().getNbComponents()),
previous_residual(residual.getVector().dataSize(),
residual.getVector().getNbComponents()),
current_x(residual.getVector().dataSize(),
residual.getVector().getNbComponents()),
delta_x(residual.getVector().dataSize(),
residual.getVector().getNbComponents()),
delta_residual(residual.getVector().dataSize(),
residual.getVector().getNbComponents()) {}
void DFSANESolver::solve() {
EPSolver::beforeSolve();
auto& x_prev = *_x;
UInt k = 0, nmax = max_iterations;
Real F_norm = 0;
previous_merits.clear();
_residual.computeResidual(x_prev);
previous_residual = _residual.getVector();
const auto F_norm0 = previous_residual.l2norm();
Logger().get(LogLevel::debug)
<< "DF-SANE: initial residual norm: " << F_norm0 << "\n";
if (F_norm0 < getTolerance()) {
Logger().get(LogLevel::info)
<< "DF-SANE: sucessful convergence (0 iterations, " << getTolerance()
<< ")\n";
_residual.computeResidualDisplacement(*_x);
return;
}
eta = [F_norm0](UInt k) { return F_norm0 / ((1 + k) * (1 + k)); };
const std::pair<Real, Real> sigma_extrema{1e-10, 1e10};
Real sigma = 1;
do {
computeSearchDirection(sigma);
lineSearch(eta(k));
// No need for a solution update
// it was done automatically in line search
sigma = computeSpectralCoeff(sigma_extrema);
x_prev = current_x;
previous_residual = _residual.getVector();
F_norm = previous_residual.l2norm();
Logger().get(LogLevel::debug) << TAMAAS_MSG(k, ' ', F_norm);
} while (F_norm > getTolerance() and k++ < nmax);
const auto msg = [k, nmax]() {
if (k < nmax)
return "successful convergence";
else
return "did not converge";
}();
Logger().get(LogLevel::info)
<< "DF-SANE: " << msg << " (" << k << " iterations, " << getTolerance()
<< ", " << F_norm << ")\n";
_residual.computeResidualDisplacement(*_x);
}
Real DFSANESolver::computeSpectralCoeff(const std::pair<Real, Real>& bounds) {
delta_x = current_x;
delta_x -= *_x;
delta_residual = _residual.getVector();
delta_residual -= previous_residual;
auto trial = delta_x.dot(delta_x) / delta_x.dot(delta_residual);
if (std::abs(trial) >= bounds.first and std::abs(trial) <= bounds.second)
return trial;
auto F = previous_residual.l2norm();
// Rule from Cruz et al. (2006)
if (F > 1)
return 1;
if (F < 1e-5)
return 1e5;
return 1 / F;
}
void DFSANESolver::computeSearchDirection(Real sigma) {
_residual.computeResidual(*_x);
search_direction = _residual.getVector();
search_direction *= -sigma;
}
void DFSANESolver::lineSearch(Real eta_k) {
const UInt M = 10;
const Real gamma = 1e-4;
Real alphap = 1, alpham = 1;
const std::pair<Real, Real> tau_extrema{0.1, 0.5};
// Merit function (nexp = 2)
const auto f = [&](Real alpha, Real sign) {
current_x = search_direction;
current_x *= alpha * sign;
current_x += *_x;
_residual.computeResidual(current_x);
return _residual.getVector().dot(_residual.getVector());
};
// Manage previous merits
const auto fk = _residual.getVector().dot(_residual.getVector());
previous_merits.push_front(fk);
if (previous_merits.size() > M)
previous_merits.pop_back();
const auto fbar =
*std::max_element(previous_merits.begin(), previous_merits.end());
do {
Real fp = f(alphap, +1);
if (fp < fbar + eta_k - gamma * alphap * alphap * fk)
return;
Real fm = f(alpham, -1);
if (fm < fbar + eta_k - gamma * alpham * alpham * fk)
return;
alphap = computeAlpha(alphap, fp, fk, tau_extrema);
alpham = computeAlpha(alpham, fm, fk, tau_extrema);
if (std::isnan(alphap) or std::isnan(alpham))
throw nan_error{TAMAAS_MSG("NaN error in line search")};
} while (true);
}
Real DFSANESolver::computeAlpha(Real alpha, Real f, Real fk,
const std::pair<Real, Real>& bounds) {
const auto alphat = alpha * alpha * fk / (f + (2 * alpha - 1) * fk);
if (alphat < bounds.first * alpha)
return bounds.first * alpha;
if (alphat > bounds.second * alpha)
return bounds.second * alpha;
return alphat;
}
} // namespace tamaas
c4science · Help