xref: /petsc/src/tao/unconstrained/impls/ntl/ntl.c (revision cd929ea3f739fd9f7b6394f772cb40b9d4e6d97c)
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