1. 引言
本文考虑如下带线性等式约束的两块可分结构非凸优化问题:
(1)
其中
是梯度Lipschitz连续可微的凸函数,
是适当下半连续函数,
。问题(1)广泛应用于信号处理[1]、图像处理[2]、机器学习[3]和经济调度[4]等领域,其中
通常是一个(二次或逻辑)损失函数,
通常是一个正则项,如
范数或
拟范数。问题(1)的增广拉格朗日函数定义为
其中
是拉格朗日乘子,
是一个惩罚参数。
求解问题(1)的经典ADMM的第
步迭代格式为
(2)
尽管ADMM在解决凸优化问题方面得到了广泛的应用,但对于非凸优化问题,ADMM的收敛性分析仍值得研究,目前一些学者已经得到了部分研究成果。如Deng Zhao等[5]针对非凸两分块优化问题,在增广拉格朗日函数满足KL性质且罚参数大于某阈值的条件下,证明了ADMM的收敛性。在此基础上,Wang Y等[6]进一步拓展了ADMM的理论框架,他们放松了文献[5]中对矩阵列满秩的严格限制,并在目标函数和系数矩阵满足一些常见约束条件下,证明了ADMM的全局收敛性。
一般将ADMM直接应用于非凸优化问题时,其收敛性不能保证,由此学者们对原始ADMM方法做了改进研究,如线性化技术、松弛技巧等。为了有利于迭代子问题的求解,Wang F等[7]针对非凸两分块优化问题(1)引入Bregman距离改善子问题目标函数的结构,提出Bregman ADMM (BADMM)算法。其过程如下
其中
是凸可微函数,其Bregman距离
定义为:
.
比如当
,其中
是半正定矩阵时有
。
在此基础上,陈建华等[8]的Majorized Bregman ADMM(MBADMM)算法针对非凸两分块优化问题,对目标函数中的光滑函数项进行了极大化线性处理,由于目标函数中的
是光滑凸函数,这意味着存在一个半正定矩阵
使得对任意
,都有:
,又由于函数
的梯度是Lipschitz连续的,所以可以找到一个半正定矩阵
,且
,使得对任意的
都有
.
于是针对问题(1),文献[8]构建Majorized增广拉格朗日函数为
(3)
相应的MBADMM算法迭代过程如下:
(4)
另一方面,注意到在迭代算法中,增加惯性力或惯性项可以在产生新方向时充分利用部分旧方向的信息,从而对算法起到加速的作用[9]-[11],Xu J等[12]提出带惯性项的Bregman Generalized ADMM (IBGADMM)算法。针对线性约束
非凸优化问题,[12]结合惯性步和Bregman距离思想给出迭代过程
并建立了算法的子序列收敛性和全局收敛性。
受上述文献的启发,对目标函数中的光滑函数项进行极大化线性处理,可以简化关于该函数子问题的求解,针对目标函数加入Bregman距离可以提升算法在求解非凸优化问题时的性能和收敛性,引入惯性项也可以缩短收敛所需时间。本文提出了针对非凸优化问题(1)的带惯性项Majorized-Bregman-ADMM (IMBADMM)算法,该算法涵盖了ADMM和MBADMM作为其特殊情形。与文献[13]有所区别的是,本文在两个子问题中引入了惯性力[9]-[11]。此外,效益函数的充分递减特性并不需要假设
或
是满秩。在IMBADMM生成的序列具有有界性且效益函数满足KL性质的条件下,我们证明了IMBADMM的子序列收敛性和全局收敛性。最后运用所提出的IMBADMM算法对稀疏逻辑回归问题进行了数值实验比较。
2. 预备知识
2.1. 基本概念与结论
对任意
,定义它们的内积为
,定义
的
范数
,其中
表明
是半正定(正定)矩阵。
表示矩阵
的最大特征值。
任意点
到集合
的距离定义如下
定义1 ([14],引理2.2) 称可微函数
是
-强凸函数,若如下不等式成立:
关于Bregman距离,有下述性质。
性质1 ([15],命题2.3) 令
是可微凸函数,
是与它相关的Bregman距离,则
1) 对任意的
,有:
;
2) 对于固定的
,
是凸的;
3) 对于任意
,
,其中
是
的强凸系数;
4) 如果
是Lipschitz连续的,且其Lipschitz常数为
,则对任意
,有
.
定义2 ([14],定义8.5]) 定义满足以下条件的凹连续函数
的集合为
函数类:
1)
;
2)
在
内连续可微,在点
处连续;
3) 对任意
,都有
。
根据上面的定义我们可以引入KL性质。
定义3 ([14],定义8.6) (KL性质) 设
为适当下半连续函数。
1) 称函数
在给定点处具有KL性质,若存在
,
的一个邻域
及函数
,使得对
,
有以下不等式成立:
.
2) 若
在
上处处都满足KL性质,则称
是一个KL函数。
性质2 ([14],引理8.4) (一致KL性质) 设
为紧集,
为适当下半连续函数,在
上为常数且在
的每个点处都满足KL性质,则存在
,使得对任意
和所有如下集合中的
:
,
有
.
性质3 ([16],引理A.2) 设
是一个非零矩阵,
表示
的最小正特征值,
是到
的欧几里得投影,则有
.
2.2. 主要结论与论文结构
本文主要贡献集中在第3部分和第4部分,分别给出新的算法及其收敛性证明和数值实验。
第3部分提出求解非凸优化问题的带惯性项Majorized Bregman ADMM (IMBADMM)算法,针对两个子问题,在每一步迭代中引入惯性力,在保证算法收敛性的前提下,有效地提高了算法的运算速度。在较弱的条件下,证明了算法的全局收敛性。
第4部分我们将IMBADMM算法应用到稀疏逻辑回归问题中,并分别与MBADMM、ADMM、SGD和LBADMM算法进行比较,数值实验都表明了算法的有效性。
最后,在第5部分给出本文总结。
3. 求解非凸优化问题的带惯性Majorized Bregman ADMM算法
对优化问题(1),我们先给出带惯性项的Majorized Bregman ADMM算法,再讨论该算法的收敛性。
3.1. IMBADMM算法描述
算法1:带惯性项的Majorized Bregman ADMM(IMBADMM)算法 |
输入:
,
,
,
,选取合适的凸可微函数
用于构造Bregman距离; 初始化:
;令
。 步1:更新参数,选取适当
和
满足
; 步2:计算
(5a) 步3:固定其他,计算
(5b) 步4:更新乘子
(5c) 步5:若
,则停止迭代,令
,输出
,
;否则令
,转步1。 |
其中
,
是两个可微凸函数,
是Bregman距离,
,
为惯性项。
3.2. 收敛性分析
在分析收敛性之前,我们需要如下假设。
假设1 1)
;
2)
是连续可微凸函数,
是
-Lipschitz连续的,且
有界,
是适当下半连续函数;
3) 构造Bregman距离函数
是
-强凸的,且
是
-Lipschitz连续的,
且
,其中
是矩阵
的最小严格正特征值,
是矩阵
的最大特征值。
为了简化表达,记
,
的临界点
,
,
引理1 若假设1成立,令
是由IMBADMM算法生成的序列,则对于所有
,有
(6)
对IMBADMM算法的收敛性分析,构造效益函数
,
, (7)
其中
。
以下所述的充分下降性质将在我们的收敛性分析中占据关键地位,为了分析
的单调性,我们记
证明:因为
,所以
,由性质3得
。
根据(5b)式的最优性条件得
(9a)
(9a)式移项可得
(9b)
令
,有
(9c)
(9b)和(9c)式相减可得
由于
是
-Lipschitz连续的,
是
-Lipschitz连续的,且
是矩阵
的最大特征值,则
上述不等式两边乘
得
引理2 若假设1成立,令
是由IMBADMM算法生成的序列,那么以下结论成立:
1) 对于所有
,有
; (10)
2) 如果
是有界的,则
。
证明:
1) (充分下降性质)
由于
是(5a)式的最优解,所以
(11)
其中第2个不等式根据性质1的3)得出,由于
是(5b)式的最优解,故有
再利用
和性质1的3),可推出
(12)
根据Majorized增广拉格朗日函数
的定义得
(13)
根据
的定义及引理1,得
(14)
根据
的定义及拉格朗日中值定理,得
(15)
由(11)、(13)、(14)、(15)式相加可得
(16)
由算法过程可知
和
,其中最后一个不等式由此得出。
根据(16)式,得
则
。
2) 由于
有界,则
有界并且至少有一个聚点,即存在子序列
,使得
。因为
是连续可微函数,
是下半连续函数,所以
为下半连续函数,因此,
,
有下界。
根据假设1的3)知
,因此
单调非增,由于单调序列的子列收敛等价于全序列收敛,即有
收敛且
。由(10)式,得
.
基于
,可推导出
与
。
进而由(6)式,可得
。因此
。
注1 由引理2的1)可知,若
,则序列
是非增的。
下面的引理给出效益函数次梯度的上限估计。
引理3 若假设1成立,令
是由IMBADMM算法生成的序列,当它有界时,对于每一个
,有
,
其中。
且存在
,使得
。
证明:根据(7)式效益函数的定义得
(17)
对
各分量求偏导,
可以表示为
根据(5b)式的最优性条件得
代入(18a)式有
由于
是
-Lipschitz连续的,则
(19)
根据(5a)式的最优性条件得
代入(18b)式有
由于
是
-Lipschitz连续的,则
(20)
根据(5c)式得
,则
. (21)
因此,
,根据(18d~18g)式及(19~21)式,存在
使得
.
根据引理1,存在
使得
。
综合上述两个不等式,当
时,可得
其中
。
定理1 (子序列收敛性) 若假设1成立,
为IMBADMM算法生成的序列,且假设
有界。令
表示序列
的聚点集,
表示序列
的聚点集,则以下结论成立:
1)
是一个非空紧集,且当
时,
;
里面的点都是
的临界点;
,当且仅当
且
;
2)
是有限的且在
上为常数,
。
接下来的结果显示,当效益函数是KL函数时,序列
将收敛到优化问题(1)的临界点。
证明:
1) 因
有界,故
至少存在一个聚点,其聚点集
非空且为紧集。
令
,那么存在
的一个子序列
,使得:
。
根据引理2的2),有
,进而可得
。
由于
且
,从而
。
又由于
是(5a)的最优解,故有
(22)
因为效益函数
关于变量
是连续的,所以
(23)
由于
是连续函数且
是下半连续函数,则
是下半连续函数。因此,
(24)
根据(23)式和(24)式可得
(25)
由此可得
。利用
的连续性和
的闭合性质,结合(5)式的最优性条件得
。
这表明
是
的临界点。
2) 根据(25)式有
,又由引理2,
是单调非增的,故有
。
定理2 (全局收敛性) 若假设1成立,
是IMBADMM算法生成的序列,且假设
有界。设
表示序列
的聚点集,如果
满足
上的KL性质,则以下结论成立:
1) 序列
具有有限长度,即:
。
2) 序列
收敛到优化问题(1)的临界点。
证明:根据定理1的2)得
,对所有
,我们考虑以下两种情况。
a) 假设存在
,满足
。由引理2的1)知
当
且
时,不等式
成立。由于假设了
,则对于任意
,有
。结合
且
,可推出对任意
,有
且
。
再结合引理1,可知对任意
,
,综上有
。
b) 假设对于任意
,均有
。因为
,对于给定
,存在
,当
时,有
。又因为
,对于给定
,存在
,当
时,有。如果,那么
,
。根据性质2,对任意的
,存在可微凹函数
,有
。因为:
,所以
(26)
根据
的凹性得
由此可推导出
.
为了简化表达,定义
,以及
.
由(26)式及引理3,对于任意
,有
.
根据引理2的1)及上述不等式,可知对任意
,有
.
上述分析表明,存在
,使得
.
利用不等式
,有
.
可推出
.
由于
,
根据上述公式得
.
两边取极限
,得
因此,
。
根据引理1及
,我们推导出
。
进一步观察到
。
由此可得
,这表明
是一个柯西序列,从而
收敛。
从定理1,我们知
收敛到
的临界点。
4. 数值实验
下面通过数值实验对比IMBADMM、MBADMM [8]、ADMM [17]、SGD [18]与LBADMM [19]算法的性能。所有实验均在配备16 GB内存、3.20 GHz、AMD Ryzen 7 6800H with Radeon Graphics CPU的联想电脑上进行,操作系统为Windows 11,使用Matlab R2023a版本进行编码和执行。
针对问题(1)考虑稀疏逻辑回归问题,基本模型如下:
(27)
其中:
是样本
的特征向量,
是相应的二分类,
是一个正则化参数,
是
范数,对任意的
有
,
。令
,
,则
是训练集
的一个样本。该问题有
个样本,维数为
。
在实际应用中,我们将(27)式转化为:
(28)
于是(28)式即为优化问题(1)的形式,其中
,
。易知:
是Lipschitz连续的,其Lipschitz连续常数
。所以存在一个半正定矩阵
使得对任意给定的
,有:
,
令
,经计算有:
,因此,
的近似项可以选择为:,其中
。对五种算法最大迭代次数设为5000,取
。
本实验数据来源于LIBSVM数据库(https://www.csie.ntu.edu.tw/cjlin/libsvmtools/)提供的a2000a数据集进行了数值实验。分别取不同的范数正则项q = 1和q = 1/2,表1展示了IMBADMM、MBADMM、ADMM、SGD以及LBADMM算法在处理稀疏逻辑回归问题(28)时的数值结果比较。观察表1中的数据,在计算效率方面,IMBADMM仅需0.07秒即可完成优化,相比MBADMM提速25~30倍;在收敛速度方面,IMBADMM所需迭代次数最少,相比ADMM减少约50%~65%的迭代次数;特别值得注意的是,在保持计算精度的前提下(q = 1时目标函数值616.01 vs MBADMM的616.02),IMBADMM实现了计算效率的显著提升。对于非光滑情况(q = 1/2),IMBADMM同样展现出强大优势,相比LBADMM将计算时间从11.37秒降至0.07秒,提速162倍,同时迭代次数从2326次减少至57次。这些实验结果充分证明了IMBADMM算法在处理大规模优化问题的卓越性能。与随机梯度下降(SGD)算法的对比显示:虽然SGD在计算时间上略有优势(0.02 s vs 0.07 s),但IMBADMM在优化质量和收敛性方面表现出优势。为了直观分析算法的收敛性能,我们分别绘制了IMBADMM、MBADMM和ADMM算法在参数取值为q = 1和q = 1/2时的收敛曲线。其中,图1展示了IMBADMM算法的误差随迭代次数的演化过程,图2则呈现了三种算法的目标函数值随迭代次数的收敛特性。通过图2可知,IMBADMM算法展现出最平滑的收敛轨迹,算法更具有稳定性且收敛速度快。
Table 1. Comparison of numerical results of the five algorithms
表1. 五个算法的数值结果对比
算法 |
维度大小 |
数据来源 |
迭代次数 |
计算时间/s |
目标函数值 |
IMBADMM (q = 1) |
999 × 60 |
a2000a |
55 |
0.07 |
616.01 |
MBADMM (q = 1) |
999 × 60 |
a2000a |
55 |
1.82 |
616.02 |
ADMM (q = 1) |
999 × 60 |
a2000a |
119 |
1.76 |
613.91 |
SGD (q = 1) |
999 × 60 |
a2000a |
150 |
0.02 |
962.71 |
LBADMM (q = 1) |
999 × 60 |
a2000a |
3006 |
1.18 |
620.00 |
IMBADMM (q = 1/2) |
999 × 60 |
a2000a |
57 |
0.07 |
672.07 |
MBADMM (q = 1/2) |
999 × 60 |
a2000a |
57 |
2.07 |
672.07 |
ADMM (q = 1/2) |
999 × 60 |
a2000a |
166 |
1.96 |
670.5 |
SGD (q = 1/2) |
999 × 60 |
a2000a |
150 |
0.02 |
1395.03 |
LBADMM (q = 1/2) |
999 × 60 |
a2000a |
2326 |
11.37 |
692.45 |
Figure 1. Evolution curve of the error with number of iterations when q = 1 and q = 1/2
图1. 当q = 1和q = 1/2时,误差随迭代次数的演化曲线
Figure 2. Evolution curve of the objective function value with the number of iterations when q = 1 and q = 1/2
图2. 当q = 1和q = 1/2时,目标函数值随迭代次数的演化曲线
5. 结论
本文针对非凸两分块优化问题,提出了一种带惯性项的Majorized Bregman交替方向乘子法(简称IMBADMM),新算法结合了Majorized Bregman ADMM和惯性Bregman generalized ADMM的基本思想。 在适当的条件下,我们建立了算法的子序列收敛性和全局收敛性。最后,初步数值实验结果表明IMBADMM算法的有效性和可靠性。
基金项目
广西自然科学基金(2022GXNSFAA035618)。
NOTES
*第一作者。
#通讯作者。