1. 引言
1.1. 研究背景及意义
随着信息技术的高速发展,许多学科领域涌现出了许多大规模优化问题,因此对于优化问题的高效求解,是当今研究的热点问题之一。本文研究的是一类具有有限和形式的非凸非光滑复合优化问题,该问题的具体格式如下:
(1.1)
其中
和
是适当下半连续凸函数,
是连续可微函数。这类模型在许多领域都有应用,包括机器学习、统计学和图像处理等。典型的例子包括非负或稀疏矩阵分解[1]、稀疏(PCA) [2] [3]、鲁棒PCA [4]、最小二乘[5]和盲图像反卷积[6]等。下面我们给出稀疏非负矩阵分解(sparse-NMF)模型,该模型的具体形式为
其中
表示X的第i列。在字典学习和稀疏编码中,X被称为系数为Y的学习字典。在该模型中,X上的稀疏性使用非凸
约束。模型可以表述为问题(1.1)的形式,具体地,我们假设
,
,
,
,
,其中
是第j个元素为1,其他元素均为0的向量。形如问题(1.1)的有限和形式的非凸非光滑复合优化问题在机器学习,图像处理和计算机视觉等领域中存在广泛应用,因此研究计算成本低、收敛速度快的算法是很有必要的。
1.2. 研究现状
针对非凸非光滑复合优化问题(1.1),一个经典的求解算法是交替极小化算法(alternating minimization, AM) [7],当函数凸且连续可微时,若其中一个参数固定时,关于另外一个参数它是严格凸的,那么用这种方法生成的序列的每一个极限点都是的极小点。AM算法为一系列研究此问题的算法提供了理论或实践上的支撑。然而在许多实际问题中,目标函数通常不满足上述较为严格的条件。为克服收敛性条件较强的缺陷,一些学者通过在AM算法的子问题中引入邻近项,提出了PAM算法[8],在非凸情形下,当目标函数满足KL性质时,该算法的收敛性分析已被建立。然而,PAM算法涉及两个函数和的邻近算子的计算,这通常是难以求解的。Bolte等人[9]通过将目标函数中的光滑部分进行线性化处理,提出了邻近交替线性极小化算法(Proximal alternating linearized minimization, PALM),其具体迭代格式如下所示:
需要说明的是,PALM算法中子问题通常有显式解或易于求解。在非凸情形下,Bolte等人在KL框架下证明了PALM算法的全局收敛性。本文正是围绕PALM算法的改进展开的。Bolte等人的工作为本文的研究提供了算法设计的灵感,展示了如何通过交替优化策略来处理非凸非光滑问题,而且他们的全局收敛性证明为本文的算法分析提供了理论基础。
针对非凸非光滑复合优化问题(1.1),当n很大的时候,PALM算法在每次迭代计算成本将非常昂
贵,因为它需要涉及到对全梯度
的计算,这将导致较大的计算成本。为了克服这个问题,针对
,仅涉及变量x的情形,随机梯度算法(SGD) [10]被提出,但是SGD要求步长随迭代进程不断衰减到0以弥补随机梯度带来的方差。为提高SGD算法的求解效率,一系列方差缩减的随机梯度算法被提出,包括SVRG [11],SAGA [12],SARAH [13]等。随机梯度也被引入到邻近梯度算法,用以邻近梯度算法的降低计算成本,包括prox-SGD,prox-SVRG,prox-SAGA [14]和prox-SARAH [15]等。本文在处理大规模优化问题时,在设计算法的迭代步骤时引入随机梯度,大大降低计算成本。
近年来,针对非凸非光滑复合优化问题(1.1),已有学者将随机梯度策略引入到PALM算法中,提出了一系列随机的PALM算法。具体的,Xu和Yin [16]首先将简单随机梯度下降法(SGD)与PALM相结合,提出了块随机梯度法(BSG)并在对目标函数
的一些较为苛刻的假设条件下,证明了BSG方法的收敛性。基于此,Driggs等人[17]提出了一种随机邻近交替线性极小化(SPRING)的方法,其中他们使用了方差减少的随机梯度算法,而不是BSG方法中使用的简单的SGD算法。数值实验表明,SPRING的收敛速度比BSG方法快。值得注意的是,与Xu和Yin的先前工作相比,SPRING的收敛性是在对目标函数更弱的假设下建立的。
然而,针对病态问题,上述一阶算法存在收敛速度较慢的问题。一个自然的想法是在算法中引入目标函数的二阶信息,以提升算法的求解速度。针对邻近梯度算法,Yang等人[18]提出了一个随机外推拟牛顿算法(stochastic extra-step quasi-Newton method),并证明了该算法在期望意义下的次线性收敛率。本文受到了他们方法的启发,在算法中通过残差方程引入目标函数的二阶信息,这些策略有助于提高算法的收敛速度。
本文扩展了PALM算法,通过引入随机梯度和二阶信息,提出了随机邻近牛顿型交替极小化算法。在计算效率和收敛速度上都有显著提升,这对于解决大规模优化问题尤为重要。
1.3. 本文的动机与贡献
考虑到一阶算法在处理病态问题时,收敛速度较慢,甚至有时不收敛。针对非凸非光滑复合优化问题(1.1),我们希望引入高阶信息以提升一阶算法的收敛速度,此外引入随机梯度算法以实现计算量的降低。具体的,本文将随机邻近牛顿型算法与邻近交替线性极小化算法相结合,提出了随机邻近牛顿型交替极小化算法,并对该算法的收敛性分析进行了证明。
1.4. 文章框架
本文框架如下,在第二节中,给出了本文所需要用到的基本符号、定义以及相关引理。在第三节中,我们给出了随机邻近牛顿型交替极小化算法的具体迭代格式,并给出算法在基本假设下的收敛性分析。在第四节,我们进行了总结。
2. 预备知识
纸型
为便于下文研究本文所提出算法的收敛性,本节将介绍文中涉及到的符号、定义以及相关引理。首先,我们对文中所涉及到的符号做出如下定义:用
和
表示标准欧几里得内积和范数。对称正定
矩阵的集合用
表示。对于给定矩阵
,我们定义内积
,
。对于任意
,我们令
,
。设
为概率空间。我们将使用大写字母来描述随机变量
和
。而小写字母通常保留给随机变量
和
。我们用
,
来表示Ω上的标准
空间。我们用
表示X是可测量的。此外,
表示由随机变量族
生成。对于随机变量
和子
代数
,给定H的X的条件期望记为
。我们使用缩写“a.e.”和“a.s.”,分别代表“几乎处处”和“几乎一定”。
定义2.1 (凸函数的次微分) 设
为适当下半连续凸函数,对
,函数f在点x处的次微分记作
,定义为所有满足下述条件的
所构成的集合:
若
,定义f在该点的次微分
。
定义2.2 (L-光滑) 对于可微函数
,若存在常数
,使其满足下列不等式
则称函数f是梯度Lipschitz连续的,其中L为Lipschitz常数。
引理2.1 (下降引理) 函数
是可微的且L-光滑(L > 0),则它满足下列不等式:
3. 随机邻近牛顿型交替极小化算法及其收敛性分析
针对大规模非凸非光滑复合优化问题(1.1),许多确定型一阶算法已被设计用以对其进行求解。但确定型算法由于计算量较大导致计算成本昂贵,如PALM算法在处理有限和形式的非凸非光滑复合优化问题时,需计算光滑部分函数的所有梯度,这将导致较大的计算成本。此外,由于一阶算法在求解病态问题时,求解速度较慢,甚至不收敛,因此将二阶信息引入到算法中,以实现算法的快速高效求解,也是本文需要考虑的一点。
具体的,在算法设计方面,为降低计算成本,提高算法的求解速度,本文将提出一种新颖的算法,即将随机梯度思想引入到PALM算法中,并借助广义残差方程引入目标函数的二阶信息,从而设计出随机邻近牛顿型交替极小化算法。在理论分析方面,本文在非凸情形下建立目标函数在期望意义下的下降性分析,进而证明出该算法收敛性。
3.1. 随机邻近牛顿型交替极小化算法
在这一小节,我们给出本文所提算法的具体结构。为更好的描述算法,我们首先给出相关符号解释。
首先,问题(1.1)最优性条件可以等价地改写为一个不动点方程,即点
是(1.1)的稳定点当且仅当
邻近算子是稳定非扩张的,即它是一个常数为1的全局Lipschitz连续函数,且满足:
给定随机方向
,我们考虑不动点方程的变体
我们用
分别表示关于
的随机邻近梯度步。当
时,用
分别表示关于
的邻近梯度步。
下面给出算法的具体框架和步骤。
算法1 随机邻近牛顿型交替极小化算法 初始化:选择正的步长
,
,
,
,点
。 For
do 计算随机梯度
,残差
(3.1) 以及方向
。计算
及随机梯度
,并执行更新
(3.2) 计算随机梯度
,以及残差
(3.3) 和方向
。计算
及随机梯度
,并执行更新
(3.4) |
本文算法根据残差方程的随机变体生成方向
和
,基于此在算法中引入了目标函数的二阶信息。具体来说,本文考虑下述形式的方向
其中矩阵
用以对基本随机邻近梯度方向
进行细化和改进,
是对
的随机近似,
是对
的随机近似。针对每一迭代步x的更新,我们首先通过
计算一个新的点,然后执行额外的邻近梯度步骤,以获得下一个迭代点
:
其中
是合适的步长参数,
是梯度
的随机近似。y的更新过程与x的更新类似,这里不再赘述。
3.2. 基本假设
为了分析随机邻近牛顿型交替极小化算法的收敛性及收敛率分析,在这一小节我们首先给出保障算法收敛性的相关假设,具体如下。
假设1 给定函数
,设
(A.1)
关于
都连续可微,且梯度映射
和
在
上是Lipschitz连续的,模
。
(A.2) 目标函数
在
上下有界。
本文假设随机近似
和
对应于随机向量
和
的实现,
和
对应于随机向量
和
的实现。此外,我们假设概率空间
足够丰富,允许我们以统一的方式建模所涉及的随机过程。我们现在定义滤流
用
表示与算法中选择的方向
相关的随机过程,
表示与算法中选择的方向
相关的随机过程。我们给出一下随机假设。
假设2 (B.1) 映射
和
对于所有k是可测的。
(B.2) 存在
,使得对于所有
,都有
a.e.
(B.3) 对于所有
,
,,
与 a.e.,且存在
,使得
和 a.e,
和 a.e。
假设2 (B.3)中
和
,
和
的条件在随机优化中很常见[19]。第二个假设要求所选方向
和随机过程
与随机非光滑残差
,
有关,并且所选方向
和随机过程
与随机非光滑残差
,
相关。在假设(B.1)下,算法的设计意味着
和
适用于滤流
,
和
适用于滤流
。即我们有
并且
,
。
3.3. 算法的收敛性分析
本节对提出的算法的收敛性展开分析,首先给出4个辅助引理以建立目标函数
的近似下降性,最后在定理1中证明了算法的收敛性。下面首先给出引理1,在该引理中给出了目标函数f的近似下降性。
引理1:假设1成立,
为随机邻近牛顿型交替极小化算法生成的序列,则有
(3.5)
证明:由迭代
的最优性条件可得
进一步有
(3.6)
利用函数f的凸性可知,
(3.7)
结合公式(3.5)和(3.6)可得:
(3.8)
其中等式是由
得到的。
由
可得,
(3.9)
又由函数f的凸性可知
(3.10)
因此
(3.11)
由
的定义可得
(3.12)
又由函数f的凸性可知
(3.13)
由(3.11)、(3.12)得
(3.14)
结合公式(3.7),(3.10)和(3.13)可得(3.14)。
类似的,对于变量y,有以下结论。
引理2:假设1成立,
为随机邻近牛顿型交替极小化算法生成的序列,则有
(3.15)
基于引理1和引理2,可以建立
的近似下降性。首先,由
定义可得
(3.16)
下面首先给出
的近似下降性。
引理3:假设1成立,
为随机邻近牛顿型交替极小化算法生成的序列,则有
(3.17)
证明:关于H的部分我们使用下降引理可得
(3.18)
则由(3.5)和(3.18)可得:
整理可得
(3.19)
利用公式(3.18)和不等式
可得
其中第二个不等式利用了
。进一步,根据
关于变量x连续可微且
可得(3.17)。
类似地,我们可以建立
的近似下降性。
引理4:假设1成立,
为随机邻近牛顿型交替极小化算法生成的序列,则有
(3.20)
基于引理1~4,下面给出算法的收敛性分析。
定理1:令随机过程
为算法生成的序列。假设1和假设2成立,且步长
以及
满足对于所有的
和某个
有以下式子成立
(3.21)
则若
,
,
,
,
,有
,
,
,
a.s.,并且
,
。
证明:由
与可得下述结果几乎处处成立
其中我们用到了
与
分别是
与
的无偏估计。基于此,将公式(3.17)与公式(3.20)相加,求条件期望,并利用假设2可得以下结果几乎处处成立:
(3.23)
结合公式(3.21)与(3.23)可得
由假设1可知存在
满足
对于所有的
。对上式左右两边求全期望并移项可得
对上式左右两边同时累和可得对于所有的
有
因为
,则
并且
。又映射
是一个关于
的递减函数[20],因此,对于所有的
,有
进而我们可以得到
,
。又因为
,
,因此
,
。又根据Borel-Cantelli引理[21]可知:当
与
几乎处处成立时,可以推出
,
必定成立。
由公式(3.21)和(3.23)可得
记
,由上式可得
对上式左右两边求全期望并累和可得
由此我们可以得到
,
。进一步可以得到
,
。
4. 总结
本文提出一种随机邻近牛顿型交替极小化算法用以求解大规模有限和形式的非凸非光滑复合优化问题,相较于邻近交替线性极小化算法PALM,本文采用的随机梯度算法可大大降低计算成本。此外,本文通过残差方程引入目标函数的二阶信息以提升算法的收敛速度。在目标函数非凸的情形下,本文给出了算法的收敛性分析,建立了随机邻近牛顿型交替极小化算法在期望意义下的收敛性。