LASSO 问题的连续化策略
- LASSO 问题的连续化策略
- 考虑 LASSO 问题
- 与 BP 问题罚函数的关系
- 初始化和迭代准备
- 连续化循环
- 辅助函数
LASSO 问题的连续化策略
考虑 LASSO 问题
连续化策略从较大的正则化参数 逐渐减小到 (即 ),并求解相应的 LASSO 问题:
这样做的好处是:在求解 对应的优化问题时,可以利用 对应优化问题的解( 子问题使用随机初始点)作为一个很好的逼近解以在较短的时间内完成求解过程; 分析可知 。因此,连续化策略相当于是通过快速求解一系列简单问题(复杂问题有了好的初始解也就变简单了)来加速求解原始问题。
这里,在调用迭代算法求解 对应的 LASSO 问题后,正则化参数
其中
与 BP 问题罚函数的关系
对于 BP 问题
利用二次罚函数法,令
令 ,不难看出 LASSO 问题的连续化策略与罚函数法的增大罚因子之间的对应。在实际应用中,连续化策略对于快速求解 LASSO 问题是非常重要的。
初始化和迭代准备
输入信息: , , ,迭代初始值
输出信息: 迭代得到的解
- out.fvec :每一步迭代的函数值
- out.itr_inn :总内层迭代次数
- out.fval :迭代终止时的目标函数值
- out.tt :运行时间
- out.itr :外层迭代次数
function [x, out] = LASSO_con(x0, A, b, mu0, opts)
从输入的结构体 opts 中读取参数或采取默认参数。
- opts.maxit :最大外层迭代次数
- opts.maxit_inn :最大内层迭代次数
- opts.ftol :针对函数值的停机判断条件
- opts.gtol :针对梯度的停机判断条件
- opts.factor :正则化系数的衰减率
- opts.verbose :不等于 0 时输出每步迭代信息,否则不输出
- opt.mu1 :初始的正则化系数(采用连续化策略,从更大的正则化系数开始)
- opts.alpha0 :初始步长
- opts.ftol_init_ratio :初始时停机准则 opts.ftol 的放大倍数
- opts.gtol_init_ratio :初始时停机准则 opts.gtol 的放大倍数
- opts.etaf :每步外层循环的停机判断标准 opts.ftol 的缩减
- opts.etag :每步外层循环的停机判断标准 opts.gtol 的缩减
- opts.opts1 :结构体,用于向内层算法提供其它具体的参数
if ~isfield(opts, 'maxit'); opts.maxit = 30; end
if ~isfield(opts, 'maxit_inn'); opts.maxit_inn = 200; end
if ~isfield(opts, 'ftol'); opts.ftol = 1e-8; end
if ~isfield(opts, 'gtol'); opts.gtol = 1e-6; end
if ~isfield(opts, 'factor'); opts.factor = 0.1; end
if ~isfield(opts, 'verbose'); opts.verbose = 1; end
if ~isfield(opts, 'mu1'); opts.mu1 = 100; end
if ~isfield(opts, 'gtol_init_ratio'); opts.gtol_init_ratio = 1/opts.gtol; end
if ~isfield(opts, 'ftol_init_ratio'); opts.ftol_init_ratio = 1e5; end
if ~isfield(opts, 'opts1'); opts.opts1 = struct(); end
if ~isfield(opts, 'etaf'); opts.etaf = 1e-1; end
if ~isfield(opts, 'etag'); opts.etag = 1e-1; end
L = eigs(A'*A,1);
if ~isfield(opts, 'alpha0'); opts.alpha0 = 1/L; end
通过 opts.method 选择求解正则化参数 对应的子问题的算法。
if ~isfield(opts, 'method')
error('Need opts.method');
end
algf = eval(sprintf('@LASSO_%s_inn',opts.method)); # 调用算法-BB步长+线搜索的近似点梯度法、FISTA算法、第二类Nesterov加速算法
添加所有子目录的路径到工作路径。
addpath(genpath(pwd));
迭代准备,注意到采取了连续化策略,因此
out = struct();
out.fvec = [];
k = 0;
x = x0;
mu_t = opts.mu1; % mu初始值为100,正则化参数逐渐减小
tt = tic;
当前正则化系数 和
f = Func(A, b, mu_t, x); % mu 初始值一般取1e-3
opts1 是固定某一正则化系数
在初始时,这两个阈值均选择较大的值。
opts1 = opts.opts1;
opts1.ftol = opts.ftol*opts.ftol_init_ratio; % opts.ftol_init_ratio:初始时停机准则 opts.ftol 的放大倍数
opts1.gtol = opts.gtol*opts.gtol_init_ratio;
out.itr_inn = 0; % 总内层迭代次数
连续化循环
求解 LASSO 问题的经典技巧,通过采用连续化策略,从较大的 逐渐减小到 ,以加速收敛。对每个
while k < opts.maxit
内层循环参数设置,记录在结构体 opts1 中
- opts1.itr:最大迭代次数,由 opts.maxit_inn 给出
- opts1.ftol:针对函数值的迭代终止条件
- opts1.gtol:针对梯度的迭代终止条件
- opts1.alpha0:初始步长
- opts1.verbose:当 ops.verbose 大于 1 时为真,此时详细输出内层迭代的信息
opts1.maxit = opts.maxit_inn; % 最大内层迭代次数
opts1.gtol = max(opts1.gtol * opts.etag, opts.gtol);
opts1.ftol = max(opts1.ftol * opts.etaf, opts.ftol); % opts.etaf:每步外层循环的停机判断标准 opts.ftol 的缩减
opts1.verbose = opts.verbose > 1; % 不等于 0 时输出每步迭代信息,否则不输出
opts1.alpha0 = opts.alpha0; % 初始步长
仅当 opts.method 为 ‘grad_huber’ 时, 光滑化参数opts1.sigma 给出 Huber 光滑化的范围;
if strcmp(opts.method, 'grad_huber')
opts1.sigma = 1e-3*mu_t; % 光滑化参数
end
调用内层循环函数,记录每一次内层循环的返回信息。out.fvec 记录每一步的 对应的原始函数值(正则化系数为 而非当前 )。
out.fvec = [out.fvec, out1.fvec];
fp = f;
[x, out1] = algf(x, A, b, mu_t, mu0, opts1); # algf-通过method控制选择不同的算法
f = out1.fvec(end);
out.fvec = [out.fvec, out1.fvec]; % 往右边添加out.fvec
k = k + 1;
由于 -范数不可导,这里 nrmG 表示 LASSO 问题的最优性条件的违反度来。
nrmG = norm(x - prox(x - A'*(A*x - b),mu0),2);
详细输出模式下打印每一次外层循环信息。
if opts.verbose
fprintf('itr: %d\tmu_t: %e\titr_inn: %d\tfval: %e\tnrmG: %.1e\n', k, mu_t, out1.itr, f, nrmG);
end
当内层循环因达到收敛条件而退出时,缩减当前正则化系数 ,并判断收敛。外层循环的收敛条件:当 已经减小到与 相同并且函数值或梯度满足收敛条件时,停止外层循环。
if ~out1.flag
mu_t = max(mu_t * opts.factor, mu0); % opts.factor:正则化系数的衰减率
end
if mu_t == mu0 && (nrmG < opts.gtol | abs(f-fp) < opts.ftol)
break;
end
更新总迭代次数。
out.itr_inn = out.itr_inn + out1.itr;
end
当外层循环终止时,记录当前的函数值、外层迭代次数和运行时间。
out.fval = f;
out.tt = toc(tt); % 读取时间
out.itr = k;
辅助函数
原始 LASSO 问题的目标函数。
function f = Func(A, b, mu0, x)
w = A * x - b;
f = 0.5 * (w' * w) + mu0 * norm(x, 1);
end
函数 对应的邻近算子 。
function y = prox(x, mu)
y = max(abs(x) - mu, 0);
y = sign(x) .* y;
end
end