形式陈述
在有限维实空间 中,设 是 proper、闭凸函数:允许取 ,但不恒为 ,也不取 。考虑将同一个变量复制成两份的问题
约束使它与 等价。本页研究这一恒等一致性约束下的 ADMM,假设存在 满足
这里 是凸次微分公理库次梯度与次微分Subgradient · Subdifferential以全局仿射下界刻画凸函数在不可微点的支撑斜率集合。。这两个条件给出原问题的最优点和拆分问题的乘子证书;在拉格朗日对偶公理库拉格朗日对偶Lagrange duality通过拉格朗日函数构造原问题下界的对偶问题,并研究弱对偶、强对偶与最优性条件。语言中, 是未增广拉格朗日函数的鞍点。它比单独假设原问题有最小点更强,保证两部分在该点能够以相反的次梯度平衡。
固定 ,增广拉格朗日函数为
任取初值 ,令 。ADMM 每轮依次精确最小化 、精确最小化 ,再更新乘子:
按近端算子公理库近端算子Proximal operator · Proximity operator在降低凸函数值与保持靠近输入点之间取得精确平衡的单值算子。的约定, 最小化 。闭凸性与二次项保证每个子问题有唯一解。上述鞍点假设下,整个序列 收敛到某个 ,后者也满足鞍点条件;下文给出这一有限维结论的完整证明。
直觉
拆分的好处是分别利用 和 的计算结构。平方损失可能适合解线性系统,绝对值惩罚则适合逐坐标软阈值;复制变量后,两种操作可以各做一次,再逐渐让两份答案一致。
更新式可以直接从配方得到。记 ,则
固定 时,第一份二次中心是 ;固定刚得到的 时,第二份中心变为 。第二步必须使用第一步的新结果,所以这是一轮交替最小化,并没有联合求出整个增广子问题的最小点。
乘子记录的是历轮未消除的一致性误差:
当两份变量长期朝同一方向偏离时,这份累积会改变后续子问题的中心。固定的二次惩罚配合乘子更新,就能驱动约束误差趋零,无需仅靠不断增大惩罚系数。
两个残差分别检查什么
定义原始残差、相邻 位移和对偶残差
负号对应这里的约束写法 。从两个子问题分别得到
代入 ,便有精确关系
因此 的驻点条件在每轮结束时已经满足; 是集合 中一个明确的元素,衡量 驻点条件的偏差; 则衡量两份变量是否可行地一致。二者同时为零时就得到完整鞍点证书。二者仅仅很小时,首先得到的是近似最优性条件;若要换算成目标值或变量距离的误差,还需要额外的误差界或对偶证书。
例子与边界
同一个耦合 Lasso 的四轮精确计算
沿用Lasso 的最优性与对偶间隙公理库Lasso 的最优性与对偶间隙Lasso optimality conditions · Lasso duality gap · Lasso primal-dual certificate从残差相关性核验 Lasso 的零与非零坐标,并用可行对偶值认证剩余优化误差。中的数据:
记 、,取 、。第一步的最优性方程和第二步的软阈值为
其中 对各坐标施加阈值 。逆矩阵的非对角项保留了损失的耦合;可分的是后面的惩罚子问题。
第一轮解线性系统得 ,两坐标都落在阈值区间内,于是 、。第二轮的右端为 ,故 ;加上旧乘子后,阈值输入为 ,所以 。继续同样的有理数运算得到:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
第一轮已经有 ,但 ,两份变量尚未一致。只看对偶残差就会在这里错误停止。 的第二坐标从一开始就是零, 的第二坐标则逐渐减小;中间变量分别承担各自子问题的结构,无需每轮拥有相同的稀疏模式。
反过来,原始残差为零也未必已经最优。取一维 、、,从 开始,第一轮得到 、。此时 ,但 ,共同变量还没有到最小点零。
用证书确定极限
四行迭代可以展示算法如何运行,但极限应由最优性条件确定。取
直接计算 。同时, 要求绝对值次梯度的第一坐标为 , 允许第二坐标取 中任意值,所以 。这正是本页的鞍点假设;而 的特征值为 ,平方损失严格凸,故原变量的最优点唯一。
原回归残差为 ,目标最优值为
下文的收敛证明于是保证 。若在有限轮结束时认证目标误差,可以把 作为原问题候选,计算 ,再使用上述 Lasso 条目的可行对偶间隙。 使用了两份尚未一致的变量,不能直接当成某个原问题可行点的目标值。
推论与应用
从单调性到 Fejér 不等式
证明只需凸次微分的单调性公理库单调算子与极大单调性Monotone operator · Maximal monotone operator · 极大单调算子用图上的内积不等式统一凸次微分与旋转关系,并以极大性保证稳定隐式步对每个输入都有唯一解。:、 蕴含 。固定任意鞍点对 ,在一轮中简记 、、、、。对两个精确次梯度关系分别应用单调性,得到
相加并利用 ,得到控制原始与对偶误差的关键式
定义乘积空间中的加权平方距离
因为 、,直接展开平方有
这里尚有一个交叉项。当 时,上一轮的 更新已保证 ,本轮又有 ,故单调性给出
于是得到所需的 Fejér 不等式:
任意初值未必满足 ,所以证明从第一轮结束后开始。若初值恰好满足该关系,不等式也适用于 ;任意初值只多出有限的一轮,不影响以下结论。
为什么整个点序列收敛
从 到 求和,并用 ,得到
因此 、,从而 。同时 保证 有界; 也有界。在有限维空间中可以取子列 ,于是 。
闭凸函数的次微分图在有限维中闭合。这一点也能从 prox 的连续性看出: 等价于 ,若 ,取极限就有 ,即 。现在对迭代中的两个关系
取极限,得到 、。所以聚点本身就是鞍点,而不只是满足一致性约束的点。
最后,把 Fejér 不等式中的固定鞍点换成刚找到的 。相应的距离
从 起非增,又沿 趋于零,因此整个 都趋于零。这证明 、;再由 得 。这里不需要最优点唯一,也没有用到函数在有效域边界的连续性。
计算结构与适用范围
对一般 Lasso,固定 时的矩阵 在所有轮次中相同,可以预先分解后重复求解。若直接形成稠密的 系统,分解需要 ,已有分解后的每轮求解为 ,软阈值和乘子更新为 ;形成矩阵与计算 另有一次成本。与近端梯度法公理库近端梯度法Proximal gradient method · Forward-backward splitting对复合目标的光滑项取显式梯度步、对非光滑凸项取隐式近端步的算法。的矩阵向量乘法相比,是否划算取决于维度、稀疏性和能否复用分解。
本页证明依赖恒等约束 、固定 、精确子问题和鞍点存在。一般形式 会引入矩阵核与子问题可解性等问题,不能由这里的点收敛论证直接推出所有原变量都收敛。非凸项、近似内层求解或调整惩罚参数也需要各自的假设与分析;对当前模型,两次 prox、两种残差和上述加权距离已经构成一条完整的算法与收敛链条。
参考资料