|
Packit |
67cb25 |
/* multifit_nlinear/common.c
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* Copyright (C) 2014, 2015, 2016 Patrick Alken
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This program is free software; you can redistribute it and/or modify
|
|
Packit |
67cb25 |
* it under the terms of the GNU General Public License as published by
|
|
Packit |
67cb25 |
* the Free Software Foundation; either version 3 of the License, or (at
|
|
Packit |
67cb25 |
* your option) any later version.
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This program is distributed in the hope that it will be useful, but
|
|
Packit |
67cb25 |
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
67cb25 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit |
67cb25 |
* General Public License for more details.
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* You should have received a copy of the GNU General Public License
|
|
Packit |
67cb25 |
* along with this program; if not, write to the Free Software
|
|
Packit |
67cb25 |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
Packit |
67cb25 |
*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static double scaled_enorm (const gsl_vector * d, const gsl_vector * f);
|
|
Packit |
67cb25 |
static void scaled_addition (const double alpha, const gsl_vector * x,
|
|
Packit |
67cb25 |
const double beta, const gsl_vector * y,
|
|
Packit |
67cb25 |
gsl_vector * z);
|
|
Packit |
67cb25 |
static double quadratic_preduction(const gsl_vector *f, const gsl_matrix * J,
|
|
Packit |
67cb25 |
const gsl_vector * dx, gsl_vector * work);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* compute || diag(d) f || */
|
|
Packit |
67cb25 |
static double
|
|
Packit |
67cb25 |
scaled_enorm (const gsl_vector * d, const gsl_vector * f)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
double e2 = 0;
|
|
Packit |
67cb25 |
size_t i, n = f->size;
|
|
Packit |
67cb25 |
for (i = 0; i < n; i++)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
double fi = gsl_vector_get (f, i);
|
|
Packit |
67cb25 |
double di = gsl_vector_get (d, i);
|
|
Packit |
67cb25 |
double u = di * fi;
|
|
Packit |
67cb25 |
e2 += u * u;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
return sqrt (e2);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* compute z = alpha*x + beta*y */
|
|
Packit |
67cb25 |
static void
|
|
Packit |
67cb25 |
scaled_addition (const double alpha, const gsl_vector * x,
|
|
Packit |
67cb25 |
const double beta, const gsl_vector * y, gsl_vector * z)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
const size_t N = z->size;
|
|
Packit |
67cb25 |
size_t i;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (i = 0; i < N; i++)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
double xi = gsl_vector_get (x, i);
|
|
Packit |
67cb25 |
double yi = gsl_vector_get (y, i);
|
|
Packit |
67cb25 |
gsl_vector_set (z, i, alpha * xi + beta * yi);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/*
|
|
Packit |
67cb25 |
quadratic_preduction()
|
|
Packit |
67cb25 |
Calculate predicted reduction based on standard
|
|
Packit |
67cb25 |
quadratic model:
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
m_k(dx) = Phi(x_k) + dx' g + 1/2 dx' B_k dx
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
predicted_reduction = m_k(0) - m_k(dx)
|
|
Packit |
67cb25 |
= -2 g^T dx / ||f||^2 - ( ||J*dx|| / ||f|| )^2
|
|
Packit |
67cb25 |
= -2 fhat . beta - ||beta||^2
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
where: beta = J*dx / ||f||
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
Inputs: f - f(x), size n
|
|
Packit |
67cb25 |
J - Jacobian J(x), n-by-p
|
|
Packit |
67cb25 |
dx - proposed step, size p
|
|
Packit |
67cb25 |
work - workspace, size n
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
Return: predicted reduction
|
|
Packit |
67cb25 |
*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static double
|
|
Packit |
67cb25 |
quadratic_preduction(const gsl_vector * f, const gsl_matrix * J,
|
|
Packit |
67cb25 |
const gsl_vector * dx, gsl_vector * work)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
const size_t n = f->size;
|
|
Packit |
67cb25 |
const double normf = gsl_blas_dnrm2(f);
|
|
Packit |
67cb25 |
double pred_reduction;
|
|
Packit |
67cb25 |
double norm_beta; /* ||J*dx|| / ||f|| */
|
|
Packit |
67cb25 |
size_t i;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* compute beta = J*dx / ||f|| */
|
|
Packit |
67cb25 |
gsl_blas_dgemv(CblasNoTrans, 1.0 / normf, J, dx, 0.0, work);
|
|
Packit |
67cb25 |
norm_beta = gsl_blas_dnrm2(work);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* initialize to ( ||J*dx|| / ||f|| )^2 */
|
|
Packit |
67cb25 |
pred_reduction = -norm_beta * norm_beta;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* subtract 2*fhat.beta */
|
|
Packit |
67cb25 |
for (i = 0; i < n; ++i)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
double fi = gsl_vector_get(f, i);
|
|
Packit |
67cb25 |
double betai = gsl_vector_get(work, i);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
pred_reduction -= 2.0 * (fi / normf) * betai;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
return pred_reduction;
|
|
Packit |
67cb25 |
}
|