【技术实现步骤摘要】
一种可微分的低秩学习网络图像修复方法
[0001]本专利技术涉及深度学习,低秩优化和图像修复领域,特别是一种可微分的低秩学习网络图像修复方法。
技术介绍
[0002]低秩矩阵优化是机器学习中的一项关键技术,在许多领域都有所研究,包括推荐系统、计算机视觉和特征表示等。因为低秩优化问题属于NP困难问题,所以许多算法使用一些非凸的替代函数来进行优化求解。例如,现有研究经常使用核范数来替代原始的秩函数,通过累加矩阵的奇异值来近似计算矩阵的秩。然而在一些实际应用中,基于核范数的优化算法往往只能得到次优解。为了解决这一问题,研究人员提出了许多针对低秩优化的非凸松弛函数,例如l
p
范数,mini-max concave plus(MCP)和smoothly clipped absolute deviation(SCAD)等函数。许多研究也进一步提出了基于截断核范数或加权核范数的优化方法。此外,Schatten-p范数作为对多种范数函数的统一表达式,其被应用在许多低秩优化问题上。例如,Zhang等人提出了一种Schatte ...
【技术保护点】
【技术特征摘要】
1.一种可微分的低秩学习网络图像修复方法,其特征在于,包括如下步骤:步骤S1、输入一个带噪声的图像矩阵M,将整个图像矩阵的低秩优化问题转换为多个子矩阵的优化子问题,并初始化各个优化子问题的相关参数和初始值;步骤S2、利用拉格朗日乘子法对每一个优化子问题进行求解,并将其中的拉格朗日乘子转换为可学习的神经网络参数,将相应的神经网络计算过程构建为一个神经网络模块;步骤S3、将步骤S2中构建的神经网络模块堆叠成一个完整的神经网络,每一个模块输出各自求解得到的最优解,并通过矩阵连乘得到低秩优化后的重构图像矩阵;步骤S4、根据步骤S3得到的重构图像矩阵,计算神经网络的损失函数,并根据该损失函数使用反向传播和梯度下降算法更新每个模块的参数W
i
以及η,若损失函数未收敛到稳定值则返回步骤S3;步骤S5、输出优化得到的重构图像矩阵。2.根据权利要求1所述的一种可微分的低秩学习网络图像修复方法,其特征在于,所述步骤S1通过Schatten-p范数替代算法将整个图像矩阵的低秩优化问题转换为多个子矩阵的优化子问题,并初始化各个优化子问题的相关参数和初始值,具体实现如下:Schatten-p范数替代算法的优化目标为:其中为分解得到的I个子矩阵,M为原始已观测到部分像素值的图像矩阵,P
Ω
为基于元素的投影操作,使得优化目标的前半部分仅计算未被噪声污染部分或已观测到的图像矩阵的误差;正则项计算多个子矩阵的Schatten-p范数和,p
i
为每个子矩阵对应的范数的超参数;根据优化目标,该算法随机初始化每个子矩阵同时初始化每个子矩阵所对应的Schatten-p范数的参数集合且每一个p
i
应满足0<p
i
≤2。3.根据权利要求2所述的一种可微分的低秩学习网络图像修复方法,其特征在于,所述步骤S2具体实现如下:根据拉格朗日乘子法和近端梯度下降...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。