155119615STodd Munson #include <../src/tao/unconstrained/impls/ntl/ntlimpl.h> 2a7e14dcfSSatish Balay 3aaa7dc30SBarry Smith #include <petscksp.h> 4a7e14dcfSSatish Balay 5a7e14dcfSSatish Balay #define NTL_PC_NONE 0 6a7e14dcfSSatish Balay #define NTL_PC_AHESS 1 7a7e14dcfSSatish Balay #define NTL_PC_BFGS 2 8a7e14dcfSSatish Balay #define NTL_PC_PETSC 3 9a7e14dcfSSatish Balay #define NTL_PC_TYPES 4 10a7e14dcfSSatish Balay 11a7e14dcfSSatish Balay #define BFGS_SCALE_AHESS 0 12a7e14dcfSSatish Balay #define BFGS_SCALE_BFGS 1 13a7e14dcfSSatish Balay #define BFGS_SCALE_TYPES 2 14a7e14dcfSSatish Balay 15a7e14dcfSSatish Balay #define NTL_INIT_CONSTANT 0 16a7e14dcfSSatish Balay #define NTL_INIT_DIRECTION 1 17a7e14dcfSSatish Balay #define NTL_INIT_INTERPOLATION 2 18a7e14dcfSSatish Balay #define NTL_INIT_TYPES 3 19a7e14dcfSSatish Balay 20a7e14dcfSSatish Balay #define NTL_UPDATE_REDUCTION 0 21a7e14dcfSSatish Balay #define NTL_UPDATE_INTERPOLATION 1 22a7e14dcfSSatish Balay #define NTL_UPDATE_TYPES 2 23a7e14dcfSSatish Balay 2487f595a5SBarry Smith static const char *NTL_PC[64] = {"none","ahess","bfgs","petsc"}; 25a7e14dcfSSatish Balay 2687f595a5SBarry Smith static const char *BFGS_SCALE[64] = {"ahess","bfgs"}; 27a7e14dcfSSatish Balay 2887f595a5SBarry Smith static const char *NTL_INIT[64] = {"constant","direction","interpolation"}; 29a7e14dcfSSatish Balay 3087f595a5SBarry Smith static const char *NTL_UPDATE[64] = {"reduction","interpolation"}; 31a7e14dcfSSatish Balay 32a7e14dcfSSatish Balay /* Implements Newton's Method with a trust-region, line-search approach for 33a7e14dcfSSatish Balay solving unconstrained minimization problems. A More'-Thuente line search 34a7e14dcfSSatish Balay is used to guarantee that the bfgs preconditioner remains positive 35a7e14dcfSSatish Balay definite. */ 36a7e14dcfSSatish Balay 37a7e14dcfSSatish Balay #define NTL_NEWTON 0 38a7e14dcfSSatish Balay #define NTL_BFGS 1 39a7e14dcfSSatish Balay #define NTL_SCALED_GRADIENT 2 40a7e14dcfSSatish Balay #define NTL_GRADIENT 3 41a7e14dcfSSatish Balay 42*cd929ea3SAlp Dener PetscErrorCode TaoNTLPrecondBFGS(PC BFGSpc, Vec X, Vec Y) 43*cd929ea3SAlp Dener { 44*cd929ea3SAlp Dener PetscErrorCode ierr; 45*cd929ea3SAlp Dener Mat *M; 46*cd929ea3SAlp Dener 47*cd929ea3SAlp Dener PetscFunctionBegin; 48*cd929ea3SAlp Dener ierr = PCShellGetContext(BFGSpc, (void**)&M);CHKERRQ(ierr); 49*cd929ea3SAlp Dener ierr = MatSolve(*M, X, Y);CHKERRQ(ierr); 50*cd929ea3SAlp Dener PetscFunctionReturn(0); 51*cd929ea3SAlp Dener } 52*cd929ea3SAlp Dener 53441846f8SBarry Smith static PetscErrorCode TaoSolve_NTL(Tao tao) 54a7e14dcfSSatish Balay { 55a7e14dcfSSatish Balay TAO_NTL *tl = (TAO_NTL *)tao->data; 5655119615STodd Munson KSPType ksp_type; 5755119615STodd Munson PetscBool is_nash,is_stcg,is_gltr; 58a7e14dcfSSatish Balay KSPConvergedReason ksp_reason; 5955119615STodd Munson PC pc; 60e4cb33bbSBarry Smith TaoLineSearchConvergedReason ls_reason; 61a7e14dcfSSatish Balay 62a7e14dcfSSatish Balay PetscReal fmin, ftrial, prered, actred, kappa, sigma; 63a7e14dcfSSatish Balay PetscReal tau, tau_1, tau_2, tau_max, tau_min, max_radius; 64a7e14dcfSSatish Balay PetscReal f, fold, gdx, gnorm; 65a7e14dcfSSatish Balay PetscReal step = 1.0; 66a7e14dcfSSatish Balay 67a7e14dcfSSatish Balay PetscReal delta; 68a7e14dcfSSatish Balay PetscReal norm_d = 0.0; 69a7e14dcfSSatish Balay PetscErrorCode ierr; 70a7e14dcfSSatish Balay PetscInt stepType; 718931d482SJason Sarich PetscInt its; 72a7e14dcfSSatish Balay 73a7e14dcfSSatish Balay PetscInt bfgsUpdates = 0; 74a7e14dcfSSatish Balay PetscInt needH; 75a7e14dcfSSatish Balay 76a7e14dcfSSatish Balay PetscInt i_max = 5; 77a7e14dcfSSatish Balay PetscInt j_max = 1; 78a7e14dcfSSatish Balay PetscInt i, j, n, N; 79a7e14dcfSSatish Balay 80a7e14dcfSSatish Balay PetscInt tr_reject; 81a7e14dcfSSatish Balay 82a7e14dcfSSatish Balay PetscFunctionBegin; 83a7e14dcfSSatish Balay if (tao->XL || tao->XU || tao->ops->computebounds) { 84a7e14dcfSSatish Balay ierr = PetscPrintf(((PetscObject)tao)->comm,"WARNING: Variable bounds have been set but will be ignored by ntl algorithm\n");CHKERRQ(ierr); 85a7e14dcfSSatish Balay } 86a7e14dcfSSatish Balay 8755119615STodd Munson ierr = KSPGetType(tao->ksp,&ksp_type);CHKERRQ(ierr); 8855119615STodd Munson ierr = PetscStrcmp(ksp_type,KSPCGNASH,&is_nash);CHKERRQ(ierr); 8955119615STodd Munson ierr = PetscStrcmp(ksp_type,KSPCGSTCG,&is_stcg);CHKERRQ(ierr); 9055119615STodd Munson ierr = PetscStrcmp(ksp_type,KSPCGGLTR,&is_gltr);CHKERRQ(ierr); 9155119615STodd Munson if (!is_nash && !is_stcg && !is_gltr) { 9255119615STodd Munson SETERRQ(PETSC_COMM_SELF,1,"TAO_NTR requires nash, stcg, or gltr for the KSP"); 9355119615STodd Munson } 94a7e14dcfSSatish Balay 9555119615STodd Munson /* Initialize the radius and modify if it is too large or small */ 9655119615STodd Munson tao->trust = tao->trust0; 97a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tl->min_radius); 98a7e14dcfSSatish Balay tao->trust = PetscMin(tao->trust, tl->max_radius); 99a7e14dcfSSatish Balay 100a7e14dcfSSatish Balay if (NTL_PC_BFGS == tl->pc_type && !tl->M) { 101a7e14dcfSSatish Balay ierr = VecGetLocalSize(tao->solution,&n);CHKERRQ(ierr); 102a7e14dcfSSatish Balay ierr = VecGetSize(tao->solution,&N);CHKERRQ(ierr); 103*cd929ea3SAlp Dener ierr = MatCreateLBFGS(((PetscObject)tao)->comm,n,N,&tl->M);CHKERRQ(ierr); 104*cd929ea3SAlp Dener ierr = MatLMVMAllocate(tl->M,tao->solution,tao->gradient);CHKERRQ(ierr); 105a7e14dcfSSatish Balay } 106a7e14dcfSSatish Balay 107a7e14dcfSSatish Balay /* Check convergence criteria */ 108a7e14dcfSSatish Balay ierr = TaoComputeObjectiveAndGradient(tao, tao->solution, &f, tao->gradient);CHKERRQ(ierr); 109a7e14dcfSSatish Balay ierr = VecNorm(tao->gradient, NORM_2, &gnorm);CHKERRQ(ierr); 11053506e15SBarry Smith if (PetscIsInfOrNanReal(f) || PetscIsInfOrNanReal(gnorm)) SETERRQ(PETSC_COMM_SELF,1, "User provided compute function generated Inf or NaN"); 111a7e14dcfSSatish Balay needH = 1; 112a7e14dcfSSatish Balay 1133ecd9318SAlp Dener tao->reason = TAO_CONTINUE_ITERATING; 1143ecd9318SAlp Dener ierr = TaoLogConvergenceHistory(tao,f,gnorm,0.0,tao->ksp_its);CHKERRQ(ierr); 1153ecd9318SAlp Dener ierr = TaoMonitor(tao,tao->niter,f,gnorm,0.0,step);CHKERRQ(ierr); 1163ecd9318SAlp Dener ierr = (*tao->ops->convergencetest)(tao,tao->cnvP);CHKERRQ(ierr); 1173ecd9318SAlp Dener if (tao->reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0); 118a7e14dcfSSatish Balay 119a7e14dcfSSatish Balay /* Create vectors for the limited memory preconditioner */ 12053506e15SBarry Smith if ((NTL_PC_BFGS == tl->pc_type) && (BFGS_SCALE_BFGS != tl->bfgs_scale_type)) { 121a7e14dcfSSatish Balay if (!tl->Diag) { 122a7e14dcfSSatish Balay ierr = VecDuplicate(tao->solution, &tl->Diag);CHKERRQ(ierr); 123a7e14dcfSSatish Balay } 124a7e14dcfSSatish Balay } 125a7e14dcfSSatish Balay 126a7e14dcfSSatish Balay /* Modify the preconditioner to use the bfgs approximation */ 127a7e14dcfSSatish Balay ierr = KSPGetPC(tao->ksp, &pc);CHKERRQ(ierr); 128a7e14dcfSSatish Balay switch(tl->pc_type) { 129a7e14dcfSSatish Balay case NTL_PC_NONE: 130a7e14dcfSSatish Balay ierr = PCSetType(pc, PCNONE);CHKERRQ(ierr); 1311a1499c8SBarry Smith ierr = PCSetFromOptions(pc);CHKERRQ(ierr); 132a7e14dcfSSatish Balay break; 133a7e14dcfSSatish Balay 134a7e14dcfSSatish Balay case NTL_PC_AHESS: 135a7e14dcfSSatish Balay ierr = PCSetType(pc, PCJACOBI);CHKERRQ(ierr); 1361a1499c8SBarry Smith ierr = PCSetFromOptions(pc);CHKERRQ(ierr); 137baa89ecbSBarry Smith ierr = PCJacobiSetUseAbs(pc,PETSC_TRUE);CHKERRQ(ierr); 138a7e14dcfSSatish Balay break; 139a7e14dcfSSatish Balay 140a7e14dcfSSatish Balay case NTL_PC_BFGS: 141a7e14dcfSSatish Balay ierr = PCSetType(pc, PCSHELL);CHKERRQ(ierr); 1421a1499c8SBarry Smith ierr = PCSetFromOptions(pc);CHKERRQ(ierr); 143a7e14dcfSSatish Balay ierr = PCShellSetName(pc, "bfgs");CHKERRQ(ierr); 144a7e14dcfSSatish Balay ierr = PCShellSetContext(pc, tl->M);CHKERRQ(ierr); 145*cd929ea3SAlp Dener ierr = PCShellSetApply(pc,TaoNTLPrecondBFGS);CHKERRQ(ierr); 146a7e14dcfSSatish Balay break; 147a7e14dcfSSatish Balay 148a7e14dcfSSatish Balay default: 149a7e14dcfSSatish Balay /* Use the pc method set by pc_type */ 150a7e14dcfSSatish Balay break; 151a7e14dcfSSatish Balay } 152a7e14dcfSSatish Balay 15355119615STodd Munson /* Initialize trust-region radius */ 154a7e14dcfSSatish Balay switch(tl->init_type) { 155a7e14dcfSSatish Balay case NTL_INIT_CONSTANT: 156a7e14dcfSSatish Balay /* Use the initial radius specified */ 157a7e14dcfSSatish Balay break; 158a7e14dcfSSatish Balay 159a7e14dcfSSatish Balay case NTL_INIT_INTERPOLATION: 160a7e14dcfSSatish Balay /* Use the initial radius specified */ 161a7e14dcfSSatish Balay max_radius = 0.0; 162a7e14dcfSSatish Balay 163a7e14dcfSSatish Balay for (j = 0; j < j_max; ++j) { 164a7e14dcfSSatish Balay fmin = f; 165a7e14dcfSSatish Balay sigma = 0.0; 166a7e14dcfSSatish Balay 167a7e14dcfSSatish Balay if (needH) { 168ffad9901SBarry Smith ierr = TaoComputeHessian(tao,tao->solution,tao->hessian,tao->hessian_pre);CHKERRQ(ierr); 169a7e14dcfSSatish Balay needH = 0; 170a7e14dcfSSatish Balay } 171a7e14dcfSSatish Balay 172a7e14dcfSSatish Balay for (i = 0; i < i_max; ++i) { 173a7e14dcfSSatish Balay ierr = VecCopy(tao->solution, tl->W);CHKERRQ(ierr); 174a7e14dcfSSatish Balay ierr = VecAXPY(tl->W, -tao->trust/gnorm, tao->gradient);CHKERRQ(ierr); 175a7e14dcfSSatish Balay 176a7e14dcfSSatish Balay ierr = TaoComputeObjective(tao, tl->W, &ftrial);CHKERRQ(ierr); 177a7e14dcfSSatish Balay if (PetscIsInfOrNanReal(ftrial)) { 178a7e14dcfSSatish Balay tau = tl->gamma1_i; 17953506e15SBarry Smith } else { 180a7e14dcfSSatish Balay if (ftrial < fmin) { 181a7e14dcfSSatish Balay fmin = ftrial; 182a7e14dcfSSatish Balay sigma = -tao->trust / gnorm; 183a7e14dcfSSatish Balay } 184a7e14dcfSSatish Balay 185a7e14dcfSSatish Balay ierr = MatMult(tao->hessian, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 186a7e14dcfSSatish Balay ierr = VecDot(tao->gradient, tao->stepdirection, &prered);CHKERRQ(ierr); 187a7e14dcfSSatish Balay 188a7e14dcfSSatish Balay prered = tao->trust * (gnorm - 0.5 * tao->trust * prered / (gnorm * gnorm)); 189a7e14dcfSSatish Balay actred = f - ftrial; 19053506e15SBarry Smith if ((PetscAbsScalar(actred) <= tl->epsilon) && (PetscAbsScalar(prered) <= tl->epsilon)) { 191a7e14dcfSSatish Balay kappa = 1.0; 19253506e15SBarry Smith } else { 193a7e14dcfSSatish Balay kappa = actred / prered; 194a7e14dcfSSatish Balay } 195a7e14dcfSSatish Balay 196a7e14dcfSSatish Balay tau_1 = tl->theta_i * gnorm * tao->trust / (tl->theta_i * gnorm * tao->trust + (1.0 - tl->theta_i) * prered - actred); 197a7e14dcfSSatish Balay tau_2 = tl->theta_i * gnorm * tao->trust / (tl->theta_i * gnorm * tao->trust - (1.0 + tl->theta_i) * prered + actred); 198a7e14dcfSSatish Balay tau_min = PetscMin(tau_1, tau_2); 199a7e14dcfSSatish Balay tau_max = PetscMax(tau_1, tau_2); 200a7e14dcfSSatish Balay 201a7e14dcfSSatish Balay if (PetscAbsScalar(kappa - 1.0) <= tl->mu1_i) { 202a7e14dcfSSatish Balay /* Great agreement */ 203a7e14dcfSSatish Balay max_radius = PetscMax(max_radius, tao->trust); 204a7e14dcfSSatish Balay 205a7e14dcfSSatish Balay if (tau_max < 1.0) { 206a7e14dcfSSatish Balay tau = tl->gamma3_i; 20753506e15SBarry Smith } else if (tau_max > tl->gamma4_i) { 208a7e14dcfSSatish Balay tau = tl->gamma4_i; 20953506e15SBarry Smith } else if (tau_1 >= 1.0 && tau_1 <= tl->gamma4_i && tau_2 < 1.0) { 210a7e14dcfSSatish Balay tau = tau_1; 21153506e15SBarry Smith } else if (tau_2 >= 1.0 && tau_2 <= tl->gamma4_i && tau_1 < 1.0) { 212a7e14dcfSSatish Balay tau = tau_2; 21353506e15SBarry Smith } else { 214a7e14dcfSSatish Balay tau = tau_max; 215a7e14dcfSSatish Balay } 21653506e15SBarry Smith } else if (PetscAbsScalar(kappa - 1.0) <= tl->mu2_i) { 217a7e14dcfSSatish Balay /* Good agreement */ 218a7e14dcfSSatish Balay max_radius = PetscMax(max_radius, tao->trust); 219a7e14dcfSSatish Balay 220a7e14dcfSSatish Balay if (tau_max < tl->gamma2_i) { 221a7e14dcfSSatish Balay tau = tl->gamma2_i; 22253506e15SBarry Smith } else if (tau_max > tl->gamma3_i) { 223a7e14dcfSSatish Balay tau = tl->gamma3_i; 22453506e15SBarry Smith } else { 225a7e14dcfSSatish Balay tau = tau_max; 226a7e14dcfSSatish Balay } 22753506e15SBarry Smith } else { 228a7e14dcfSSatish Balay /* Not good agreement */ 229a7e14dcfSSatish Balay if (tau_min > 1.0) { 230a7e14dcfSSatish Balay tau = tl->gamma2_i; 23153506e15SBarry Smith } else if (tau_max < tl->gamma1_i) { 232a7e14dcfSSatish Balay tau = tl->gamma1_i; 23353506e15SBarry Smith } else if ((tau_min < tl->gamma1_i) && (tau_max >= 1.0)) { 234a7e14dcfSSatish Balay tau = tl->gamma1_i; 23553506e15SBarry Smith } else if ((tau_1 >= tl->gamma1_i) && (tau_1 < 1.0) && ((tau_2 < tl->gamma1_i) || (tau_2 >= 1.0))) { 236a7e14dcfSSatish Balay tau = tau_1; 23753506e15SBarry Smith } else if ((tau_2 >= tl->gamma1_i) && (tau_2 < 1.0) && ((tau_1 < tl->gamma1_i) || (tau_2 >= 1.0))) { 238a7e14dcfSSatish Balay tau = tau_2; 23953506e15SBarry Smith } else { 240a7e14dcfSSatish Balay tau = tau_max; 241a7e14dcfSSatish Balay } 242a7e14dcfSSatish Balay } 243a7e14dcfSSatish Balay } 244a7e14dcfSSatish Balay tao->trust = tau * tao->trust; 245a7e14dcfSSatish Balay } 246a7e14dcfSSatish Balay 247a7e14dcfSSatish Balay if (fmin < f) { 248a7e14dcfSSatish Balay f = fmin; 249a7e14dcfSSatish Balay ierr = VecAXPY(tao->solution, sigma, tao->gradient);CHKERRQ(ierr); 250a7e14dcfSSatish Balay ierr = TaoComputeGradient(tao, tao->solution, tao->gradient);CHKERRQ(ierr); 251a7e14dcfSSatish Balay 252a7e14dcfSSatish Balay ierr = VecNorm(tao->gradient, NORM_2, &gnorm);CHKERRQ(ierr); 25353506e15SBarry Smith if (PetscIsInfOrNanReal(f) || PetscIsInfOrNanReal(gnorm)) SETERRQ(PETSC_COMM_SELF,1, "User provided compute function generated Inf or NaN"); 254a7e14dcfSSatish Balay needH = 1; 255a7e14dcfSSatish Balay 2563ecd9318SAlp Dener ierr = TaoLogConvergenceHistory(tao,f,gnorm,0.0,tao->ksp_its);CHKERRQ(ierr); 2573ecd9318SAlp Dener ierr = TaoMonitor(tao,tao->niter,f,gnorm,0.0,step);CHKERRQ(ierr); 2583ecd9318SAlp Dener ierr = (*tao->ops->convergencetest)(tao,tao->cnvP);CHKERRQ(ierr); 2593ecd9318SAlp Dener if (tao->reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0); 260a7e14dcfSSatish Balay } 261a7e14dcfSSatish Balay } 262a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, max_radius); 263a7e14dcfSSatish Balay 264a7e14dcfSSatish Balay /* Modify the radius if it is too large or small */ 265a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tl->min_radius); 266a7e14dcfSSatish Balay tao->trust = PetscMin(tao->trust, tl->max_radius); 267a7e14dcfSSatish Balay break; 268a7e14dcfSSatish Balay 269a7e14dcfSSatish Balay default: 270a7e14dcfSSatish Balay /* Norm of the first direction will initialize radius */ 271a7e14dcfSSatish Balay tao->trust = 0.0; 272a7e14dcfSSatish Balay break; 273a7e14dcfSSatish Balay } 274a7e14dcfSSatish Balay 275a7e14dcfSSatish Balay /* Set initial scaling for the BFGS preconditioner 276a7e14dcfSSatish Balay This step is done after computing the initial trust-region radius 277a7e14dcfSSatish Balay since the function value may have decreased */ 278a7e14dcfSSatish Balay if (NTL_PC_BFGS == tl->pc_type) { 279*cd929ea3SAlp Dener delta = 2.0 * PetscMax(1.0, PetscAbsScalar(f)) / (gnorm*gnorm); 280*cd929ea3SAlp Dener ierr = MatLMVMSetJ0Scale(tl->M, delta);CHKERRQ(ierr); 281a7e14dcfSSatish Balay } 282a7e14dcfSSatish Balay 283a7e14dcfSSatish Balay /* Set counter for gradient/reset steps */ 284a7e14dcfSSatish Balay tl->ntrust = 0; 285a7e14dcfSSatish Balay tl->newt = 0; 286a7e14dcfSSatish Balay tl->bfgs = 0; 287a7e14dcfSSatish Balay tl->sgrad = 0; 288a7e14dcfSSatish Balay tl->grad = 0; 289a7e14dcfSSatish Balay 290a7e14dcfSSatish Balay /* Have not converged; continue with Newton method */ 2913ecd9318SAlp Dener while (tao->reason == TAO_CONTINUE_ITERATING) { 2928931d482SJason Sarich ++tao->niter; 293ae93cb3cSJason Sarich tao->ksp_its=0; 294a7e14dcfSSatish Balay /* Compute the Hessian */ 295a7e14dcfSSatish Balay if (needH) { 296ffad9901SBarry Smith ierr = TaoComputeHessian(tao,tao->solution,tao->hessian,tao->hessian_pre);CHKERRQ(ierr); 297a7e14dcfSSatish Balay } 298a7e14dcfSSatish Balay 299a7e14dcfSSatish Balay if (NTL_PC_BFGS == tl->pc_type) { 300a7e14dcfSSatish Balay if (BFGS_SCALE_AHESS == tl->bfgs_scale_type) { 301a7e14dcfSSatish Balay /* Obtain diagonal for the bfgs preconditioner */ 302a7e14dcfSSatish Balay ierr = MatGetDiagonal(tao->hessian, tl->Diag);CHKERRQ(ierr); 303a7e14dcfSSatish Balay ierr = VecAbs(tl->Diag);CHKERRQ(ierr); 304a7e14dcfSSatish Balay ierr = VecReciprocal(tl->Diag);CHKERRQ(ierr); 305*cd929ea3SAlp Dener ierr = MatLMVMSetJ0Diag(tl->M, tl->Diag);CHKERRQ(ierr); 306a7e14dcfSSatish Balay } 307a7e14dcfSSatish Balay 308a7e14dcfSSatish Balay /* Update the limited memory preconditioner */ 309a7e14dcfSSatish Balay ierr = MatLMVMUpdate(tl->M,tao->solution, tao->gradient);CHKERRQ(ierr); 310a7e14dcfSSatish Balay ++bfgsUpdates; 311a7e14dcfSSatish Balay } 31223ee1639SBarry Smith ierr = KSPSetOperators(tao->ksp, tao->hessian, tao->hessian_pre);CHKERRQ(ierr); 313a7e14dcfSSatish Balay /* Solve the Newton system of equations */ 314ba7fe8fbSTodd Munson ierr = KSPCGSetRadius(tao->ksp,tl->max_radius);CHKERRQ(ierr); 315a7e14dcfSSatish Balay ierr = KSPSolve(tao->ksp, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 316a7e14dcfSSatish Balay ierr = KSPGetIterationNumber(tao->ksp,&its);CHKERRQ(ierr); 317a7e14dcfSSatish Balay tao->ksp_its+=its; 318ae93cb3cSJason Sarich tao->ksp_tot_its+=its; 319ba7fe8fbSTodd Munson ierr = KSPCGGetNormD(tao->ksp, &norm_d);CHKERRQ(ierr); 320a7e14dcfSSatish Balay 321a7e14dcfSSatish Balay if (0.0 == tao->trust) { 322a7e14dcfSSatish Balay /* Radius was uninitialized; use the norm of the direction */ 323a7e14dcfSSatish Balay if (norm_d > 0.0) { 324a7e14dcfSSatish Balay tao->trust = norm_d; 325a7e14dcfSSatish Balay 326a7e14dcfSSatish Balay /* Modify the radius if it is too large or small */ 327a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tl->min_radius); 328a7e14dcfSSatish Balay tao->trust = PetscMin(tao->trust, tl->max_radius); 32953506e15SBarry Smith } else { 330a7e14dcfSSatish Balay /* The direction was bad; set radius to default value and re-solve 331a7e14dcfSSatish Balay the trust-region subproblem to get a direction */ 332a7e14dcfSSatish Balay tao->trust = tao->trust0; 333a7e14dcfSSatish Balay 334a7e14dcfSSatish Balay /* Modify the radius if it is too large or small */ 335a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tl->min_radius); 336a7e14dcfSSatish Balay tao->trust = PetscMin(tao->trust, tl->max_radius); 337a7e14dcfSSatish Balay 338ba7fe8fbSTodd Munson ierr = KSPCGSetRadius(tao->ksp,tl->max_radius);CHKERRQ(ierr); 339a7e14dcfSSatish Balay ierr = KSPSolve(tao->ksp, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 340a7e14dcfSSatish Balay ierr = KSPGetIterationNumber(tao->ksp,&its);CHKERRQ(ierr); 341a7e14dcfSSatish Balay tao->ksp_its+=its; 3422d9aa51bSJason Sarich tao->ksp_tot_its+=its; 343ba7fe8fbSTodd Munson ierr = KSPCGGetNormD(tao->ksp, &norm_d);CHKERRQ(ierr); 344a7e14dcfSSatish Balay 34553506e15SBarry Smith if (norm_d == 0.0) SETERRQ(PETSC_COMM_SELF,1, "Initial direction zero"); 346a7e14dcfSSatish Balay } 347a7e14dcfSSatish Balay } 348a7e14dcfSSatish Balay 349a7e14dcfSSatish Balay ierr = VecScale(tao->stepdirection, -1.0);CHKERRQ(ierr); 350a7e14dcfSSatish Balay ierr = KSPGetConvergedReason(tao->ksp, &ksp_reason);CHKERRQ(ierr); 35153506e15SBarry Smith if ((KSP_DIVERGED_INDEFINITE_PC == ksp_reason) && (NTL_PC_BFGS == tl->pc_type) && (bfgsUpdates > 1)) { 352a7e14dcfSSatish Balay /* Preconditioner is numerically indefinite; reset the 353a7e14dcfSSatish Balay approximate if using BFGS preconditioning. */ 354a7e14dcfSSatish Balay 355*cd929ea3SAlp Dener delta = 2.0 * PetscMax(1.0, PetscAbsScalar(f)) / (gnorm*gnorm); 356*cd929ea3SAlp Dener ierr = MatLMVMSetJ0Scale(tl->M, delta);CHKERRQ(ierr); 357*cd929ea3SAlp Dener ierr = MatLMVMReset(tl->M, PETSC_FALSE);CHKERRQ(ierr); 358a7e14dcfSSatish Balay ierr = MatLMVMUpdate(tl->M, tao->solution, tao->gradient);CHKERRQ(ierr); 359a7e14dcfSSatish Balay bfgsUpdates = 1; 360a7e14dcfSSatish Balay } 361a7e14dcfSSatish Balay 362a7e14dcfSSatish Balay /* Check trust-region reduction conditions */ 363a7e14dcfSSatish Balay tr_reject = 0; 364a7e14dcfSSatish Balay if (NTL_UPDATE_REDUCTION == tl->update_type) { 365a7e14dcfSSatish Balay /* Get predicted reduction */ 366ba7fe8fbSTodd Munson ierr = KSPCGGetObjFcn(tao->ksp,&prered);CHKERRQ(ierr); 367a7e14dcfSSatish Balay if (prered >= 0.0) { 368a7e14dcfSSatish Balay /* The predicted reduction has the wrong sign. This cannot 369a7e14dcfSSatish Balay happen in infinite precision arithmetic. Step should 370a7e14dcfSSatish Balay be rejected! */ 371a7e14dcfSSatish Balay tao->trust = tl->alpha1 * PetscMin(tao->trust, norm_d); 372a7e14dcfSSatish Balay tr_reject = 1; 37353506e15SBarry Smith } else { 374a7e14dcfSSatish Balay /* Compute trial step and function value */ 375a7e14dcfSSatish Balay ierr = VecCopy(tao->solution, tl->W);CHKERRQ(ierr); 376a7e14dcfSSatish Balay ierr = VecAXPY(tl->W, 1.0, tao->stepdirection);CHKERRQ(ierr); 377a7e14dcfSSatish Balay ierr = TaoComputeObjective(tao, tl->W, &ftrial);CHKERRQ(ierr); 378a7e14dcfSSatish Balay 379a7e14dcfSSatish Balay if (PetscIsInfOrNanReal(ftrial)) { 380a7e14dcfSSatish Balay tao->trust = tl->alpha1 * PetscMin(tao->trust, norm_d); 381a7e14dcfSSatish Balay tr_reject = 1; 38253506e15SBarry Smith } else { 383a7e14dcfSSatish Balay /* Compute and actual reduction */ 384a7e14dcfSSatish Balay actred = f - ftrial; 385a7e14dcfSSatish Balay prered = -prered; 386a7e14dcfSSatish Balay if ((PetscAbsScalar(actred) <= tl->epsilon) && 387a7e14dcfSSatish Balay (PetscAbsScalar(prered) <= tl->epsilon)) { 388a7e14dcfSSatish Balay kappa = 1.0; 38953506e15SBarry Smith } else { 390a7e14dcfSSatish Balay kappa = actred / prered; 391a7e14dcfSSatish Balay } 392a7e14dcfSSatish Balay 393a7e14dcfSSatish Balay /* Accept of reject the step and update radius */ 394a7e14dcfSSatish Balay if (kappa < tl->eta1) { 395a7e14dcfSSatish Balay /* Reject the step */ 396a7e14dcfSSatish Balay tao->trust = tl->alpha1 * PetscMin(tao->trust, norm_d); 397a7e14dcfSSatish Balay tr_reject = 1; 39853506e15SBarry Smith } else { 399a7e14dcfSSatish Balay /* Accept the step */ 400a7e14dcfSSatish Balay if (kappa < tl->eta2) { 401a7e14dcfSSatish Balay /* Marginal bad step */ 402a7e14dcfSSatish Balay tao->trust = tl->alpha2 * PetscMin(tao->trust, norm_d); 40353506e15SBarry Smith } else if (kappa < tl->eta3) { 404a7e14dcfSSatish Balay /* Reasonable step */ 405a7e14dcfSSatish Balay tao->trust = tl->alpha3 * tao->trust; 40653506e15SBarry Smith } else if (kappa < tl->eta4) { 407a7e14dcfSSatish Balay /* Good step */ 408a7e14dcfSSatish Balay tao->trust = PetscMax(tl->alpha4 * norm_d, tao->trust); 40953506e15SBarry Smith } else { 410a7e14dcfSSatish Balay /* Very good step */ 411a7e14dcfSSatish Balay tao->trust = PetscMax(tl->alpha5 * norm_d, tao->trust); 412a7e14dcfSSatish Balay } 413a7e14dcfSSatish Balay } 414a7e14dcfSSatish Balay } 415a7e14dcfSSatish Balay } 41653506e15SBarry Smith } else { 417a7e14dcfSSatish Balay /* Get predicted reduction */ 418ba7fe8fbSTodd Munson ierr = KSPCGGetObjFcn(tao->ksp,&prered);CHKERRQ(ierr); 419a7e14dcfSSatish Balay if (prered >= 0.0) { 420a7e14dcfSSatish Balay /* The predicted reduction has the wrong sign. This cannot 421a7e14dcfSSatish Balay happen in infinite precision arithmetic. Step should 422a7e14dcfSSatish Balay be rejected! */ 423a7e14dcfSSatish Balay tao->trust = tl->gamma1 * PetscMin(tao->trust, norm_d); 424a7e14dcfSSatish Balay tr_reject = 1; 42553506e15SBarry Smith } else { 426a7e14dcfSSatish Balay ierr = VecCopy(tao->solution, tl->W);CHKERRQ(ierr); 427a7e14dcfSSatish Balay ierr = VecAXPY(tl->W, 1.0, tao->stepdirection);CHKERRQ(ierr); 428a7e14dcfSSatish Balay ierr = TaoComputeObjective(tao, tl->W, &ftrial);CHKERRQ(ierr); 429a7e14dcfSSatish Balay if (PetscIsInfOrNanReal(ftrial)) { 430a7e14dcfSSatish Balay tao->trust = tl->gamma1 * PetscMin(tao->trust, norm_d); 431a7e14dcfSSatish Balay tr_reject = 1; 43253506e15SBarry Smith } else { 433a7e14dcfSSatish Balay ierr = VecDot(tao->gradient, tao->stepdirection, &gdx);CHKERRQ(ierr); 434a7e14dcfSSatish Balay 435a7e14dcfSSatish Balay actred = f - ftrial; 436a7e14dcfSSatish Balay prered = -prered; 437a7e14dcfSSatish Balay if ((PetscAbsScalar(actred) <= tl->epsilon) && 438a7e14dcfSSatish Balay (PetscAbsScalar(prered) <= tl->epsilon)) { 439a7e14dcfSSatish Balay kappa = 1.0; 44053506e15SBarry Smith } else { 441a7e14dcfSSatish Balay kappa = actred / prered; 442a7e14dcfSSatish Balay } 443a7e14dcfSSatish Balay 444a7e14dcfSSatish Balay tau_1 = tl->theta * gdx / (tl->theta * gdx - (1.0 - tl->theta) * prered + actred); 445a7e14dcfSSatish Balay tau_2 = tl->theta * gdx / (tl->theta * gdx + (1.0 + tl->theta) * prered - actred); 446a7e14dcfSSatish Balay tau_min = PetscMin(tau_1, tau_2); 447a7e14dcfSSatish Balay tau_max = PetscMax(tau_1, tau_2); 448a7e14dcfSSatish Balay 449a7e14dcfSSatish Balay if (kappa >= 1.0 - tl->mu1) { 450a7e14dcfSSatish Balay /* Great agreement; accept step and update radius */ 451a7e14dcfSSatish Balay if (tau_max < 1.0) { 452a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tl->gamma3 * norm_d); 45353506e15SBarry Smith } else if (tau_max > tl->gamma4) { 454a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tl->gamma4 * norm_d); 45553506e15SBarry Smith } else { 456a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tau_max * norm_d); 457a7e14dcfSSatish Balay } 45853506e15SBarry Smith } else if (kappa >= 1.0 - tl->mu2) { 459a7e14dcfSSatish Balay /* Good agreement */ 460a7e14dcfSSatish Balay 461a7e14dcfSSatish Balay if (tau_max < tl->gamma2) { 462a7e14dcfSSatish Balay tao->trust = tl->gamma2 * PetscMin(tao->trust, norm_d); 46353506e15SBarry Smith } else if (tau_max > tl->gamma3) { 464a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tl->gamma3 * norm_d); 465a7e14dcfSSatish Balay } else if (tau_max < 1.0) { 466a7e14dcfSSatish Balay tao->trust = tau_max * PetscMin(tao->trust, norm_d); 46753506e15SBarry Smith } else { 468a7e14dcfSSatish Balay tao->trust = PetscMax(tao->trust, tau_max * norm_d); 469a7e14dcfSSatish Balay } 47053506e15SBarry Smith } else { 471a7e14dcfSSatish Balay /* Not good agreement */ 472a7e14dcfSSatish Balay if (tau_min > 1.0) { 473a7e14dcfSSatish Balay tao->trust = tl->gamma2 * PetscMin(tao->trust, norm_d); 47453506e15SBarry Smith } else if (tau_max < tl->gamma1) { 475a7e14dcfSSatish Balay tao->trust = tl->gamma1 * PetscMin(tao->trust, norm_d); 47653506e15SBarry Smith } else if ((tau_min < tl->gamma1) && (tau_max >= 1.0)) { 477a7e14dcfSSatish Balay tao->trust = tl->gamma1 * PetscMin(tao->trust, norm_d); 47853506e15SBarry Smith } else if ((tau_1 >= tl->gamma1) && (tau_1 < 1.0) && ((tau_2 < tl->gamma1) || (tau_2 >= 1.0))) { 479a7e14dcfSSatish Balay tao->trust = tau_1 * PetscMin(tao->trust, norm_d); 48053506e15SBarry Smith } else if ((tau_2 >= tl->gamma1) && (tau_2 < 1.0) && ((tau_1 < tl->gamma1) || (tau_2 >= 1.0))) { 481a7e14dcfSSatish Balay tao->trust = tau_2 * PetscMin(tao->trust, norm_d); 48253506e15SBarry Smith } else { 483a7e14dcfSSatish Balay tao->trust = tau_max * PetscMin(tao->trust, norm_d); 484a7e14dcfSSatish Balay } 485a7e14dcfSSatish Balay tr_reject = 1; 486a7e14dcfSSatish Balay } 487a7e14dcfSSatish Balay } 488a7e14dcfSSatish Balay } 489a7e14dcfSSatish Balay } 490a7e14dcfSSatish Balay 491a7e14dcfSSatish Balay if (tr_reject) { 492a7e14dcfSSatish Balay /* The trust-region constraints rejected the step. Apply a linesearch. 493a7e14dcfSSatish Balay Check for descent direction. */ 494a7e14dcfSSatish Balay ierr = VecDot(tao->stepdirection, tao->gradient, &gdx);CHKERRQ(ierr); 495a7e14dcfSSatish Balay if ((gdx >= 0.0) || PetscIsInfOrNanReal(gdx)) { 496a7e14dcfSSatish Balay /* Newton step is not descent or direction produced Inf or NaN */ 497a7e14dcfSSatish Balay 498a7e14dcfSSatish Balay if (NTL_PC_BFGS != tl->pc_type) { 499a7e14dcfSSatish Balay /* We don't have the bfgs matrix around and updated 500a7e14dcfSSatish Balay Must use gradient direction in this case */ 501a7e14dcfSSatish Balay ierr = VecCopy(tao->gradient, tao->stepdirection);CHKERRQ(ierr); 502a7e14dcfSSatish Balay ierr = VecScale(tao->stepdirection, -1.0);CHKERRQ(ierr); 503a7e14dcfSSatish Balay ++tl->grad; 504a7e14dcfSSatish Balay stepType = NTL_GRADIENT; 50553506e15SBarry Smith } else { 506a7e14dcfSSatish Balay /* Attempt to use the BFGS direction */ 507*cd929ea3SAlp Dener ierr = MatSolve(tl->M, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 508a7e14dcfSSatish Balay ierr = VecScale(tao->stepdirection, -1.0);CHKERRQ(ierr); 509a7e14dcfSSatish Balay 510a7e14dcfSSatish Balay /* Check for success (descent direction) */ 511a7e14dcfSSatish Balay ierr = VecDot(tao->stepdirection, tao->gradient, &gdx);CHKERRQ(ierr); 512a7e14dcfSSatish Balay if ((gdx >= 0) || PetscIsInfOrNanReal(gdx)) { 513a7e14dcfSSatish Balay /* BFGS direction is not descent or direction produced not a number 514a7e14dcfSSatish Balay We can assert bfgsUpdates > 1 in this case because 515a7e14dcfSSatish Balay the first solve produces the scaled gradient direction, 516a7e14dcfSSatish Balay which is guaranteed to be descent */ 517a7e14dcfSSatish Balay 518a7e14dcfSSatish Balay /* Use steepest descent direction (scaled) */ 519*cd929ea3SAlp Dener delta = 2.0 * PetscMax(1.0, PetscAbsScalar(f)) / (gnorm*gnorm); 520*cd929ea3SAlp Dener ierr = MatLMVMSetJ0Scale(tl->M, delta);CHKERRQ(ierr); 521*cd929ea3SAlp Dener ierr = MatLMVMReset(tl->M, PETSC_FALSE);CHKERRQ(ierr); 522a7e14dcfSSatish Balay ierr = MatLMVMUpdate(tl->M, tao->solution, tao->gradient);CHKERRQ(ierr); 523*cd929ea3SAlp Dener ierr = MatSolve(tl->M, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 524a7e14dcfSSatish Balay ierr = VecScale(tao->stepdirection, -1.0);CHKERRQ(ierr); 525a7e14dcfSSatish Balay 526a7e14dcfSSatish Balay bfgsUpdates = 1; 527a7e14dcfSSatish Balay ++tl->sgrad; 528a7e14dcfSSatish Balay stepType = NTL_SCALED_GRADIENT; 52953506e15SBarry Smith } else { 530a7e14dcfSSatish Balay if (1 == bfgsUpdates) { 531a7e14dcfSSatish Balay /* The first BFGS direction is always the scaled gradient */ 532a7e14dcfSSatish Balay ++tl->sgrad; 533a7e14dcfSSatish Balay stepType = NTL_SCALED_GRADIENT; 53453506e15SBarry Smith } else { 535a7e14dcfSSatish Balay ++tl->bfgs; 536a7e14dcfSSatish Balay stepType = NTL_BFGS; 537a7e14dcfSSatish Balay } 538a7e14dcfSSatish Balay } 539a7e14dcfSSatish Balay } 54053506e15SBarry Smith } else { 541a7e14dcfSSatish Balay /* Computed Newton step is descent */ 542a7e14dcfSSatish Balay ++tl->newt; 543a7e14dcfSSatish Balay stepType = NTL_NEWTON; 544a7e14dcfSSatish Balay } 545a7e14dcfSSatish Balay 546a7e14dcfSSatish Balay /* Perform the linesearch */ 547a7e14dcfSSatish Balay fold = f; 548a7e14dcfSSatish Balay ierr = VecCopy(tao->solution, tl->Xold);CHKERRQ(ierr); 549a7e14dcfSSatish Balay ierr = VecCopy(tao->gradient, tl->Gold);CHKERRQ(ierr); 550a7e14dcfSSatish Balay 551a7e14dcfSSatish Balay step = 1.0; 552a7e14dcfSSatish Balay ierr = TaoLineSearchApply(tao->linesearch, tao->solution, &f, tao->gradient, tao->stepdirection, &step, &ls_reason);CHKERRQ(ierr); 553a7e14dcfSSatish Balay ierr = TaoAddLineSearchCounts(tao);CHKERRQ(ierr); 554a7e14dcfSSatish Balay 55553506e15SBarry Smith while (ls_reason != TAOLINESEARCH_SUCCESS && ls_reason != TAOLINESEARCH_SUCCESS_USER && stepType != NTL_GRADIENT) { /* Linesearch failed */ 556a7e14dcfSSatish Balay /* Linesearch failed */ 557a7e14dcfSSatish Balay f = fold; 558a7e14dcfSSatish Balay ierr = VecCopy(tl->Xold, tao->solution);CHKERRQ(ierr); 559a7e14dcfSSatish Balay ierr = VecCopy(tl->Gold, tao->gradient);CHKERRQ(ierr); 560a7e14dcfSSatish Balay 561a7e14dcfSSatish Balay switch(stepType) { 562a7e14dcfSSatish Balay case NTL_NEWTON: 563a7e14dcfSSatish Balay /* Failed to obtain acceptable iterate with Newton step */ 564a7e14dcfSSatish Balay 565a7e14dcfSSatish Balay if (NTL_PC_BFGS != tl->pc_type) { 566a7e14dcfSSatish Balay /* We don't have the bfgs matrix around and being updated 567a7e14dcfSSatish Balay Must use gradient direction in this case */ 568a7e14dcfSSatish Balay ierr = VecCopy(tao->gradient, tao->stepdirection);CHKERRQ(ierr); 569a7e14dcfSSatish Balay ++tl->grad; 570a7e14dcfSSatish Balay stepType = NTL_GRADIENT; 57153506e15SBarry Smith } else { 572a7e14dcfSSatish Balay /* Attempt to use the BFGS direction */ 573*cd929ea3SAlp Dener ierr = MatSolve(tl->M, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 574a7e14dcfSSatish Balay 575a7e14dcfSSatish Balay 576a7e14dcfSSatish Balay /* Check for success (descent direction) */ 577a7e14dcfSSatish Balay ierr = VecDot(tao->stepdirection, tao->gradient, &gdx);CHKERRQ(ierr); 578a7e14dcfSSatish Balay if ((gdx <= 0) || PetscIsInfOrNanReal(gdx)) { 579a7e14dcfSSatish Balay /* BFGS direction is not descent or direction produced 580a7e14dcfSSatish Balay not a number. We can assert bfgsUpdates > 1 in this case 581a7e14dcfSSatish Balay Use steepest descent direction (scaled) */ 582a7e14dcfSSatish Balay 583*cd929ea3SAlp Dener delta = 2.0 * PetscMax(1.0, PetscAbsScalar(f)) / (gnorm*gnorm); 584*cd929ea3SAlp Dener ierr = MatLMVMSetJ0Scale(tl->M, delta);CHKERRQ(ierr); 585*cd929ea3SAlp Dener ierr = MatLMVMReset(tl->M, PETSC_FALSE);CHKERRQ(ierr); 586a7e14dcfSSatish Balay ierr = MatLMVMUpdate(tl->M, tao->solution, tao->gradient);CHKERRQ(ierr); 587*cd929ea3SAlp Dener ierr = MatSolve(tl->M, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 588a7e14dcfSSatish Balay 589a7e14dcfSSatish Balay bfgsUpdates = 1; 590a7e14dcfSSatish Balay ++tl->sgrad; 591a7e14dcfSSatish Balay stepType = NTL_SCALED_GRADIENT; 59253506e15SBarry Smith } else { 593a7e14dcfSSatish Balay if (1 == bfgsUpdates) { 594a7e14dcfSSatish Balay /* The first BFGS direction is always the scaled gradient */ 595a7e14dcfSSatish Balay ++tl->sgrad; 596a7e14dcfSSatish Balay stepType = NTL_SCALED_GRADIENT; 59753506e15SBarry Smith } else { 598a7e14dcfSSatish Balay ++tl->bfgs; 599a7e14dcfSSatish Balay stepType = NTL_BFGS; 600a7e14dcfSSatish Balay } 601a7e14dcfSSatish Balay } 602a7e14dcfSSatish Balay } 603a7e14dcfSSatish Balay break; 604a7e14dcfSSatish Balay 605a7e14dcfSSatish Balay case NTL_BFGS: 606a7e14dcfSSatish Balay /* Can only enter if pc_type == NTL_PC_BFGS 607a7e14dcfSSatish Balay Failed to obtain acceptable iterate with BFGS step 608a7e14dcfSSatish Balay Attempt to use the scaled gradient direction */ 609a7e14dcfSSatish Balay 610*cd929ea3SAlp Dener delta = 2.0 * PetscMax(1.0, PetscAbsScalar(f)) / (gnorm*gnorm); 611*cd929ea3SAlp Dener ierr = MatLMVMSetJ0Scale(tl->M, delta);CHKERRQ(ierr); 612*cd929ea3SAlp Dener ierr = MatLMVMReset(tl->M, PETSC_FALSE);CHKERRQ(ierr); 613a7e14dcfSSatish Balay ierr = MatLMVMUpdate(tl->M, tao->solution, tao->gradient);CHKERRQ(ierr); 614*cd929ea3SAlp Dener ierr = MatSolve(tl->M, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 615a7e14dcfSSatish Balay 616a7e14dcfSSatish Balay bfgsUpdates = 1; 617a7e14dcfSSatish Balay ++tl->sgrad; 618a7e14dcfSSatish Balay stepType = NTL_SCALED_GRADIENT; 619a7e14dcfSSatish Balay break; 620a7e14dcfSSatish Balay 621a7e14dcfSSatish Balay case NTL_SCALED_GRADIENT: 622a7e14dcfSSatish Balay /* Can only enter if pc_type == NTL_PC_BFGS 623a7e14dcfSSatish Balay The scaled gradient step did not produce a new iterate; 624a7e14dcfSSatish Balay attemp to use the gradient direction. 625a7e14dcfSSatish Balay Need to make sure we are not using a different diagonal scaling */ 626*cd929ea3SAlp Dener ierr = MatLMVMSetJ0Scale(tl->M, 1.0);CHKERRQ(ierr); 627*cd929ea3SAlp Dener ierr = MatLMVMReset(tl->M, PETSC_FALSE);CHKERRQ(ierr); 628a7e14dcfSSatish Balay ierr = MatLMVMUpdate(tl->M, tao->solution, tao->gradient);CHKERRQ(ierr); 629*cd929ea3SAlp Dener ierr = MatSolve(tl->M, tao->gradient, tao->stepdirection);CHKERRQ(ierr); 630a7e14dcfSSatish Balay 631a7e14dcfSSatish Balay bfgsUpdates = 1; 632a7e14dcfSSatish Balay ++tl->grad; 633a7e14dcfSSatish Balay stepType = NTL_GRADIENT; 634a7e14dcfSSatish Balay break; 635a7e14dcfSSatish Balay } 636a7e14dcfSSatish Balay ierr = VecScale(tao->stepdirection, -1.0);CHKERRQ(ierr); 637a7e14dcfSSatish Balay 638a7e14dcfSSatish Balay /* This may be incorrect; linesearch has values for stepmax and stepmin 639a7e14dcfSSatish Balay that should be reset. */ 640a7e14dcfSSatish Balay step = 1.0; 641a7e14dcfSSatish Balay ierr = TaoLineSearchApply(tao->linesearch, tao->solution, &f, tao->gradient, tao->stepdirection, &step, &ls_reason);CHKERRQ(ierr); 642a7e14dcfSSatish Balay ierr = TaoAddLineSearchCounts(tao);CHKERRQ(ierr); 643a7e14dcfSSatish Balay } 644a7e14dcfSSatish Balay 64553506e15SBarry Smith if (ls_reason != TAOLINESEARCH_SUCCESS && ls_reason != TAOLINESEARCH_SUCCESS_USER) { 646a7e14dcfSSatish Balay /* Failed to find an improving point */ 647a7e14dcfSSatish Balay f = fold; 648a7e14dcfSSatish Balay ierr = VecCopy(tl->Xold, tao->solution);CHKERRQ(ierr); 649a7e14dcfSSatish Balay ierr = VecCopy(tl->Gold, tao->gradient);CHKERRQ(ierr); 650a7e14dcfSSatish Balay tao->trust = 0.0; 651a7e14dcfSSatish Balay step = 0.0; 652a7e14dcfSSatish Balay tao->reason = TAO_DIVERGED_LS_FAILURE; 653a7e14dcfSSatish Balay break; 65453506e15SBarry Smith } else if (stepType == NTL_NEWTON) { 655a7e14dcfSSatish Balay if (step < tl->nu1) { 656a7e14dcfSSatish Balay /* Very bad step taken; reduce radius */ 657a7e14dcfSSatish Balay tao->trust = tl->omega1 * PetscMin(norm_d, tao->trust); 65853506e15SBarry Smith } else if (step < tl->nu2) { 659a7e14dcfSSatish Balay /* Reasonably bad step taken; reduce radius */ 660a7e14dcfSSatish Balay tao->trust = tl->omega2 * PetscMin(norm_d, tao->trust); 66153506e15SBarry Smith } else if (step < tl->nu3) { 662a7e14dcfSSatish Balay /* Reasonable step was taken; leave radius alone */ 663a7e14dcfSSatish Balay if (tl->omega3 < 1.0) { 664a7e14dcfSSatish Balay tao->trust = tl->omega3 * PetscMin(norm_d, tao->trust); 66553506e15SBarry Smith } else if (tl->omega3 > 1.0) { 666a7e14dcfSSatish Balay tao->trust = PetscMax(tl->omega3 * norm_d, tao->trust); 667a7e14dcfSSatish Balay } 66853506e15SBarry Smith } else if (step < tl->nu4) { 669a7e14dcfSSatish Balay /* Full step taken; increase the radius */ 670a7e14dcfSSatish Balay tao->trust = PetscMax(tl->omega4 * norm_d, tao->trust); 67153506e15SBarry Smith } else { 672a7e14dcfSSatish Balay /* More than full step taken; increase the radius */ 673a7e14dcfSSatish Balay tao->trust = PetscMax(tl->omega5 * norm_d, tao->trust); 674a7e14dcfSSatish Balay } 67553506e15SBarry Smith } else { 676a7e14dcfSSatish Balay /* Newton step was not good; reduce the radius */ 677a7e14dcfSSatish Balay tao->trust = tl->omega1 * PetscMin(norm_d, tao->trust); 678a7e14dcfSSatish Balay } 67953506e15SBarry Smith } else { 680a7e14dcfSSatish Balay /* Trust-region step is accepted */ 681a7e14dcfSSatish Balay ierr = VecCopy(tl->W, tao->solution);CHKERRQ(ierr); 682a7e14dcfSSatish Balay f = ftrial; 683a7e14dcfSSatish Balay ierr = TaoComputeGradient(tao, tao->solution, tao->gradient);CHKERRQ(ierr); 684a7e14dcfSSatish Balay ++tl->ntrust; 685a7e14dcfSSatish Balay } 686a7e14dcfSSatish Balay 687a7e14dcfSSatish Balay /* The radius may have been increased; modify if it is too large */ 688a7e14dcfSSatish Balay tao->trust = PetscMin(tao->trust, tl->max_radius); 689a7e14dcfSSatish Balay 690e4cb33bbSBarry Smith /* Check for converged */ 691a7e14dcfSSatish Balay ierr = VecNorm(tao->gradient, NORM_2, &gnorm);CHKERRQ(ierr); 69253506e15SBarry Smith if (PetscIsInfOrNanReal(f) || PetscIsInfOrNanReal(gnorm)) SETERRQ(PETSC_COMM_SELF,1,"User provided compute function generated Not-a-Number"); 693a7e14dcfSSatish Balay needH = 1; 694a7e14dcfSSatish Balay 6953ecd9318SAlp Dener ierr = TaoLogConvergenceHistory(tao,f,gnorm,0.0,tao->ksp_its);CHKERRQ(ierr); 6963ecd9318SAlp Dener ierr = TaoMonitor(tao,tao->niter,f,gnorm,0.0,step);CHKERRQ(ierr); 6973ecd9318SAlp Dener ierr = (*tao->ops->convergencetest)(tao,tao->cnvP);CHKERRQ(ierr); 698a7e14dcfSSatish Balay } 699a7e14dcfSSatish Balay PetscFunctionReturn(0); 700a7e14dcfSSatish Balay } 701a7e14dcfSSatish Balay 702a7e14dcfSSatish Balay /* ---------------------------------------------------------- */ 703441846f8SBarry Smith static PetscErrorCode TaoSetUp_NTL(Tao tao) 704a7e14dcfSSatish Balay { 705a7e14dcfSSatish Balay TAO_NTL *tl = (TAO_NTL *)tao->data; 706a7e14dcfSSatish Balay PetscErrorCode ierr; 707a7e14dcfSSatish Balay 708a7e14dcfSSatish Balay PetscFunctionBegin; 709a7e14dcfSSatish Balay if (!tao->gradient) {ierr = VecDuplicate(tao->solution, &tao->gradient);CHKERRQ(ierr); } 710a7e14dcfSSatish Balay if (!tao->stepdirection) {ierr = VecDuplicate(tao->solution, &tao->stepdirection);CHKERRQ(ierr);} 711a7e14dcfSSatish Balay if (!tl->W) { ierr = VecDuplicate(tao->solution, &tl->W);CHKERRQ(ierr);} 712a7e14dcfSSatish Balay if (!tl->Xold) { ierr = VecDuplicate(tao->solution, &tl->Xold);CHKERRQ(ierr);} 713a7e14dcfSSatish Balay if (!tl->Gold) { ierr = VecDuplicate(tao->solution, &tl->Gold);CHKERRQ(ierr);} 714a7e14dcfSSatish Balay tl->Diag = 0; 715a7e14dcfSSatish Balay tl->M = 0; 716a7e14dcfSSatish Balay PetscFunctionReturn(0); 717a7e14dcfSSatish Balay } 718a7e14dcfSSatish Balay 719a7e14dcfSSatish Balay /*------------------------------------------------------------*/ 720441846f8SBarry Smith static PetscErrorCode TaoDestroy_NTL(Tao tao) 721a7e14dcfSSatish Balay { 722a7e14dcfSSatish Balay TAO_NTL *tl = (TAO_NTL *)tao->data; 723a7e14dcfSSatish Balay PetscErrorCode ierr; 724a7e14dcfSSatish Balay 725a7e14dcfSSatish Balay PetscFunctionBegin; 726a7e14dcfSSatish Balay if (tao->setupcalled) { 727a7e14dcfSSatish Balay ierr = VecDestroy(&tl->W);CHKERRQ(ierr); 728a7e14dcfSSatish Balay ierr = VecDestroy(&tl->Xold);CHKERRQ(ierr); 729a7e14dcfSSatish Balay ierr = VecDestroy(&tl->Gold);CHKERRQ(ierr); 730a7e14dcfSSatish Balay } 731a7e14dcfSSatish Balay ierr = VecDestroy(&tl->Diag);CHKERRQ(ierr); 732a7e14dcfSSatish Balay ierr = MatDestroy(&tl->M);CHKERRQ(ierr); 733a7e14dcfSSatish Balay ierr = PetscFree(tao->data);CHKERRQ(ierr); 734a7e14dcfSSatish Balay PetscFunctionReturn(0); 735a7e14dcfSSatish Balay } 736a7e14dcfSSatish Balay 737a7e14dcfSSatish Balay /*------------------------------------------------------------*/ 7384416b707SBarry Smith static PetscErrorCode TaoSetFromOptions_NTL(PetscOptionItems *PetscOptionsObject,Tao tao) 739a7e14dcfSSatish Balay { 740a7e14dcfSSatish Balay TAO_NTL *tl = (TAO_NTL *)tao->data; 741a7e14dcfSSatish Balay PetscErrorCode ierr; 742a7e14dcfSSatish Balay 743a7e14dcfSSatish Balay PetscFunctionBegin; 7441a1499c8SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"Newton trust region with line search method for unconstrained optimization");CHKERRQ(ierr); 74594ae4db5SBarry Smith ierr = PetscOptionsEList("-tao_ntl_pc_type", "pc type", "", NTL_PC, NTL_PC_TYPES, NTL_PC[tl->pc_type], &tl->pc_type,NULL);CHKERRQ(ierr); 74694ae4db5SBarry Smith ierr = PetscOptionsEList("-tao_ntl_bfgs_scale_type", "bfgs scale type", "", BFGS_SCALE, BFGS_SCALE_TYPES, BFGS_SCALE[tl->bfgs_scale_type], &tl->bfgs_scale_type,NULL);CHKERRQ(ierr); 74794ae4db5SBarry Smith ierr = PetscOptionsEList("-tao_ntl_init_type", "radius initialization type", "", NTL_INIT, NTL_INIT_TYPES, NTL_INIT[tl->init_type], &tl->init_type,NULL);CHKERRQ(ierr); 74894ae4db5SBarry Smith ierr = PetscOptionsEList("-tao_ntl_update_type", "radius update type", "", NTL_UPDATE, NTL_UPDATE_TYPES, NTL_UPDATE[tl->update_type], &tl->update_type,NULL);CHKERRQ(ierr); 74994ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_eta1", "poor steplength; reduce radius", "", tl->eta1, &tl->eta1,NULL);CHKERRQ(ierr); 75094ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_eta2", "reasonable steplength; leave radius alone", "", tl->eta2, &tl->eta2,NULL);CHKERRQ(ierr); 75194ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_eta3", "good steplength; increase radius", "", tl->eta3, &tl->eta3,NULL);CHKERRQ(ierr); 75294ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_eta4", "excellent steplength; greatly increase radius", "", tl->eta4, &tl->eta4,NULL);CHKERRQ(ierr); 75394ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_alpha1", "", "", tl->alpha1, &tl->alpha1,NULL);CHKERRQ(ierr); 75494ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_alpha2", "", "", tl->alpha2, &tl->alpha2,NULL);CHKERRQ(ierr); 75594ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_alpha3", "", "", tl->alpha3, &tl->alpha3,NULL);CHKERRQ(ierr); 75694ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_alpha4", "", "", tl->alpha4, &tl->alpha4,NULL);CHKERRQ(ierr); 75794ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_alpha5", "", "", tl->alpha5, &tl->alpha5,NULL);CHKERRQ(ierr); 75894ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_nu1", "poor steplength; reduce radius", "", tl->nu1, &tl->nu1,NULL);CHKERRQ(ierr); 75994ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_nu2", "reasonable steplength; leave radius alone", "", tl->nu2, &tl->nu2,NULL);CHKERRQ(ierr); 76094ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_nu3", "good steplength; increase radius", "", tl->nu3, &tl->nu3,NULL);CHKERRQ(ierr); 76194ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_nu4", "excellent steplength; greatly increase radius", "", tl->nu4, &tl->nu4,NULL);CHKERRQ(ierr); 76294ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_omega1", "", "", tl->omega1, &tl->omega1,NULL);CHKERRQ(ierr); 76394ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_omega2", "", "", tl->omega2, &tl->omega2,NULL);CHKERRQ(ierr); 76494ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_omega3", "", "", tl->omega3, &tl->omega3,NULL);CHKERRQ(ierr); 76594ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_omega4", "", "", tl->omega4, &tl->omega4,NULL);CHKERRQ(ierr); 76694ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_omega5", "", "", tl->omega5, &tl->omega5,NULL);CHKERRQ(ierr); 76794ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_mu1_i", "", "", tl->mu1_i, &tl->mu1_i,NULL);CHKERRQ(ierr); 76894ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_mu2_i", "", "", tl->mu2_i, &tl->mu2_i,NULL);CHKERRQ(ierr); 76994ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma1_i", "", "", tl->gamma1_i, &tl->gamma1_i,NULL);CHKERRQ(ierr); 77094ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma2_i", "", "", tl->gamma2_i, &tl->gamma2_i,NULL);CHKERRQ(ierr); 77194ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma3_i", "", "", tl->gamma3_i, &tl->gamma3_i,NULL);CHKERRQ(ierr); 77294ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma4_i", "", "", tl->gamma4_i, &tl->gamma4_i,NULL);CHKERRQ(ierr); 77394ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_theta_i", "", "", tl->theta_i, &tl->theta_i,NULL);CHKERRQ(ierr); 77494ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_mu1", "", "", tl->mu1, &tl->mu1,NULL);CHKERRQ(ierr); 77594ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_mu2", "", "", tl->mu2, &tl->mu2,NULL);CHKERRQ(ierr); 77694ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma1", "", "", tl->gamma1, &tl->gamma1,NULL);CHKERRQ(ierr); 77794ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma2", "", "", tl->gamma2, &tl->gamma2,NULL);CHKERRQ(ierr); 77894ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma3", "", "", tl->gamma3, &tl->gamma3,NULL);CHKERRQ(ierr); 77994ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_gamma4", "", "", tl->gamma4, &tl->gamma4,NULL);CHKERRQ(ierr); 78094ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_theta", "", "", tl->theta, &tl->theta,NULL);CHKERRQ(ierr); 78194ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_min_radius", "lower bound on initial radius", "", tl->min_radius, &tl->min_radius,NULL);CHKERRQ(ierr); 78294ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_max_radius", "upper bound on radius", "", tl->max_radius, &tl->max_radius,NULL);CHKERRQ(ierr); 78394ae4db5SBarry Smith ierr = PetscOptionsReal("-tao_ntl_epsilon", "tolerance used when computing actual and predicted reduction", "", tl->epsilon, &tl->epsilon,NULL);CHKERRQ(ierr); 784a7e14dcfSSatish Balay ierr = PetscOptionsTail();CHKERRQ(ierr); 785a7e14dcfSSatish Balay ierr = TaoLineSearchSetFromOptions(tao->linesearch);CHKERRQ(ierr); 786a7e14dcfSSatish Balay ierr = KSPSetFromOptions(tao->ksp);CHKERRQ(ierr); 787a7e14dcfSSatish Balay PetscFunctionReturn(0); 788a7e14dcfSSatish Balay } 789a7e14dcfSSatish Balay 790a7e14dcfSSatish Balay /*------------------------------------------------------------*/ 791441846f8SBarry Smith static PetscErrorCode TaoView_NTL(Tao tao, PetscViewer viewer) 792a7e14dcfSSatish Balay { 793a7e14dcfSSatish Balay TAO_NTL *tl = (TAO_NTL *)tao->data; 794a7e14dcfSSatish Balay PetscInt nrejects; 795a7e14dcfSSatish Balay PetscBool isascii; 796a7e14dcfSSatish Balay PetscErrorCode ierr; 797a7e14dcfSSatish Balay 798a7e14dcfSSatish Balay PetscFunctionBegin; 799a7e14dcfSSatish Balay ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 800a7e14dcfSSatish Balay if (isascii) { 801a7e14dcfSSatish Balay ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 802a7e14dcfSSatish Balay if (NTL_PC_BFGS == tl->pc_type && tl->M) { 803*cd929ea3SAlp Dener ierr = MatLMVMGetRejectCount(tl->M, &nrejects);CHKERRQ(ierr); 804a7e14dcfSSatish Balay ierr = PetscViewerASCIIPrintf(viewer, "Rejected matrix updates: %D\n", nrejects);CHKERRQ(ierr); 805a7e14dcfSSatish Balay } 806a7e14dcfSSatish Balay ierr = PetscViewerASCIIPrintf(viewer, "Trust-region steps: %D\n", tl->ntrust);CHKERRQ(ierr); 807a7e14dcfSSatish Balay ierr = PetscViewerASCIIPrintf(viewer, "Newton search steps: %D\n", tl->newt);CHKERRQ(ierr); 808a7e14dcfSSatish Balay ierr = PetscViewerASCIIPrintf(viewer, "BFGS search steps: %D\n", tl->bfgs);CHKERRQ(ierr); 809a7e14dcfSSatish Balay ierr = PetscViewerASCIIPrintf(viewer, "Scaled gradient search steps: %D\n", tl->sgrad);CHKERRQ(ierr); 810a7e14dcfSSatish Balay ierr = PetscViewerASCIIPrintf(viewer, "Gradient search steps: %D\n", tl->grad);CHKERRQ(ierr); 811a7e14dcfSSatish Balay ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 812a7e14dcfSSatish Balay } 813a7e14dcfSSatish Balay PetscFunctionReturn(0); 814a7e14dcfSSatish Balay } 815a7e14dcfSSatish Balay 816a7e14dcfSSatish Balay /* ---------------------------------------------------------- */ 8171522df2eSJason Sarich /*MC 8181522df2eSJason Sarich TAONTR - Newton's method with trust region and linesearch 8191522df2eSJason Sarich for unconstrained minimization. 8201522df2eSJason Sarich At each iteration, the Newton trust region method solves the system for d 8211522df2eSJason Sarich and performs a line search in the d direction: 8221522df2eSJason Sarich 8231522df2eSJason Sarich min_d .5 dT Hk d + gkT d, s.t. ||d|| < Delta_k 8241522df2eSJason Sarich 8251522df2eSJason Sarich Options Database Keys: 82655119615STodd Munson + -tao_ntl_pc_type - "none","ahess","bfgs","petsc" 8271522df2eSJason Sarich . -tao_ntl_bfgs_scale_type - type of scaling with bfgs pc, "ahess" or "bfgs" 8281522df2eSJason Sarich . -tao_ntl_init_type - "constant","direction","interpolation" 8291522df2eSJason Sarich . -tao_ntl_update_type - "reduction","interpolation" 8301522df2eSJason Sarich . -tao_ntl_min_radius - lower bound on trust region radius 8311522df2eSJason Sarich . -tao_ntl_max_radius - upper bound on trust region radius 8321522df2eSJason Sarich . -tao_ntl_epsilon - tolerance for accepting actual / predicted reduction 8331522df2eSJason Sarich . -tao_ntl_mu1_i - mu1 interpolation init factor 8341522df2eSJason Sarich . -tao_ntl_mu2_i - mu2 interpolation init factor 8351522df2eSJason Sarich . -tao_ntl_gamma1_i - gamma1 interpolation init factor 8361522df2eSJason Sarich . -tao_ntl_gamma2_i - gamma2 interpolation init factor 8371522df2eSJason Sarich . -tao_ntl_gamma3_i - gamma3 interpolation init factor 8381522df2eSJason Sarich . -tao_ntl_gamma4_i - gamma4 interpolation init factor 8391522df2eSJason Sarich . -tao_ntl_theta_i - thetha1 interpolation init factor 8401522df2eSJason Sarich . -tao_ntl_eta1 - eta1 reduction update factor 8411522df2eSJason Sarich . -tao_ntl_eta2 - eta2 reduction update factor 8421522df2eSJason Sarich . -tao_ntl_eta3 - eta3 reduction update factor 8431522df2eSJason Sarich . -tao_ntl_eta4 - eta4 reduction update factor 8441522df2eSJason Sarich . -tao_ntl_alpha1 - alpha1 reduction update factor 8451522df2eSJason Sarich . -tao_ntl_alpha2 - alpha2 reduction update factor 8461522df2eSJason Sarich . -tao_ntl_alpha3 - alpha3 reduction update factor 8471522df2eSJason Sarich . -tao_ntl_alpha4 - alpha4 reduction update factor 8481522df2eSJason Sarich . -tao_ntl_alpha4 - alpha4 reduction update factor 8491522df2eSJason Sarich . -tao_ntl_mu1 - mu1 interpolation update 8501522df2eSJason Sarich . -tao_ntl_mu2 - mu2 interpolation update 8511522df2eSJason Sarich . -tao_ntl_gamma1 - gamma1 interpolcation update 8521522df2eSJason Sarich . -tao_ntl_gamma2 - gamma2 interpolcation update 8531522df2eSJason Sarich . -tao_ntl_gamma3 - gamma3 interpolcation update 8541522df2eSJason Sarich . -tao_ntl_gamma4 - gamma4 interpolation update 8551522df2eSJason Sarich - -tao_ntl_theta - theta1 interpolation update 8561522df2eSJason Sarich 8571eb8069cSJason Sarich Level: beginner 8581522df2eSJason Sarich M*/ 8591522df2eSJason Sarich 860728e0ed0SBarry Smith PETSC_EXTERN PetscErrorCode TaoCreate_NTL(Tao tao) 861a7e14dcfSSatish Balay { 862a7e14dcfSSatish Balay TAO_NTL *tl; 863a7e14dcfSSatish Balay PetscErrorCode ierr; 8648caf6e8cSBarry Smith const char *morethuente_type = TAOLINESEARCHMT; 86553506e15SBarry Smith 866a7e14dcfSSatish Balay PetscFunctionBegin; 8673c9e27cfSGeoffrey Irving ierr = PetscNewLog(tao,&tl);CHKERRQ(ierr); 868a7e14dcfSSatish Balay tao->ops->setup = TaoSetUp_NTL; 869a7e14dcfSSatish Balay tao->ops->solve = TaoSolve_NTL; 870a7e14dcfSSatish Balay tao->ops->view = TaoView_NTL; 871a7e14dcfSSatish Balay tao->ops->setfromoptions = TaoSetFromOptions_NTL; 872a7e14dcfSSatish Balay tao->ops->destroy = TaoDestroy_NTL; 873a7e14dcfSSatish Balay 8746552cf8aSJason Sarich /* Override default settings (unless already changed) */ 8756552cf8aSJason Sarich if (!tao->max_it_changed) tao->max_it = 50; 8766552cf8aSJason Sarich if (!tao->trust0_changed) tao->trust0 = 100.0; 8776552cf8aSJason Sarich 878a7e14dcfSSatish Balay tao->data = (void*)tl; 879a7e14dcfSSatish Balay 880a7e14dcfSSatish Balay /* Default values for trust-region radius update based on steplength */ 881a7e14dcfSSatish Balay tl->nu1 = 0.25; 882a7e14dcfSSatish Balay tl->nu2 = 0.50; 883a7e14dcfSSatish Balay tl->nu3 = 1.00; 884a7e14dcfSSatish Balay tl->nu4 = 1.25; 885a7e14dcfSSatish Balay 886a7e14dcfSSatish Balay tl->omega1 = 0.25; 887a7e14dcfSSatish Balay tl->omega2 = 0.50; 888a7e14dcfSSatish Balay tl->omega3 = 1.00; 889a7e14dcfSSatish Balay tl->omega4 = 2.00; 890a7e14dcfSSatish Balay tl->omega5 = 4.00; 891a7e14dcfSSatish Balay 892a7e14dcfSSatish Balay /* Default values for trust-region radius update based on reduction */ 893a7e14dcfSSatish Balay tl->eta1 = 1.0e-4; 894a7e14dcfSSatish Balay tl->eta2 = 0.25; 895a7e14dcfSSatish Balay tl->eta3 = 0.50; 896a7e14dcfSSatish Balay tl->eta4 = 0.90; 897a7e14dcfSSatish Balay 898a7e14dcfSSatish Balay tl->alpha1 = 0.25; 899a7e14dcfSSatish Balay tl->alpha2 = 0.50; 900a7e14dcfSSatish Balay tl->alpha3 = 1.00; 901a7e14dcfSSatish Balay tl->alpha4 = 2.00; 902a7e14dcfSSatish Balay tl->alpha5 = 4.00; 903a7e14dcfSSatish Balay 904a7e14dcfSSatish Balay /* Default values for trust-region radius update based on interpolation */ 905a7e14dcfSSatish Balay tl->mu1 = 0.10; 906a7e14dcfSSatish Balay tl->mu2 = 0.50; 907a7e14dcfSSatish Balay 908a7e14dcfSSatish Balay tl->gamma1 = 0.25; 909a7e14dcfSSatish Balay tl->gamma2 = 0.50; 910a7e14dcfSSatish Balay tl->gamma3 = 2.00; 911a7e14dcfSSatish Balay tl->gamma4 = 4.00; 912a7e14dcfSSatish Balay 913a7e14dcfSSatish Balay tl->theta = 0.05; 914a7e14dcfSSatish Balay 915a7e14dcfSSatish Balay /* Default values for trust region initialization based on interpolation */ 916a7e14dcfSSatish Balay tl->mu1_i = 0.35; 917a7e14dcfSSatish Balay tl->mu2_i = 0.50; 918a7e14dcfSSatish Balay 919a7e14dcfSSatish Balay tl->gamma1_i = 0.0625; 920a7e14dcfSSatish Balay tl->gamma2_i = 0.5; 921a7e14dcfSSatish Balay tl->gamma3_i = 2.0; 922a7e14dcfSSatish Balay tl->gamma4_i = 5.0; 923a7e14dcfSSatish Balay 924a7e14dcfSSatish Balay tl->theta_i = 0.25; 925a7e14dcfSSatish Balay 926a7e14dcfSSatish Balay /* Remaining parameters */ 927a7e14dcfSSatish Balay tl->min_radius = 1.0e-10; 928a7e14dcfSSatish Balay tl->max_radius = 1.0e10; 929a7e14dcfSSatish Balay tl->epsilon = 1.0e-6; 930a7e14dcfSSatish Balay 931a7e14dcfSSatish Balay tl->pc_type = NTL_PC_BFGS; 932a7e14dcfSSatish Balay tl->bfgs_scale_type = BFGS_SCALE_AHESS; 933a7e14dcfSSatish Balay tl->init_type = NTL_INIT_INTERPOLATION; 934a7e14dcfSSatish Balay tl->update_type = NTL_UPDATE_REDUCTION; 935a7e14dcfSSatish Balay 936a7e14dcfSSatish Balay ierr = TaoLineSearchCreate(((PetscObject)tao)->comm, &tao->linesearch);CHKERRQ(ierr); 93763b15415SAlp Dener ierr = PetscObjectIncrementTabLevel((PetscObject)tao->linesearch, (PetscObject)tao, 1);CHKERRQ(ierr); 938a7e14dcfSSatish Balay ierr = TaoLineSearchSetType(tao->linesearch, morethuente_type);CHKERRQ(ierr); 939441846f8SBarry Smith ierr = TaoLineSearchUseTaoRoutines(tao->linesearch, tao);CHKERRQ(ierr); 9405d527766SPatrick Farrell ierr = TaoLineSearchSetOptionsPrefix(tao->linesearch,tao->hdr.prefix);CHKERRQ(ierr); 941a7e14dcfSSatish Balay ierr = KSPCreate(((PetscObject)tao)->comm,&tao->ksp);CHKERRQ(ierr); 94263b15415SAlp Dener ierr = PetscObjectIncrementTabLevel((PetscObject)tao->ksp, (PetscObject)tao, 1);CHKERRQ(ierr); 9435d527766SPatrick Farrell ierr = KSPSetOptionsPrefix(tao->ksp,tao->hdr.prefix);CHKERRQ(ierr); 94455119615STodd Munson ierr = KSPSetType(tao->ksp,KSPCGSTCG);CHKERRQ(ierr); 945a7e14dcfSSatish Balay PetscFunctionReturn(0); 946a7e14dcfSSatish Balay } 947