角度受限下稀疏投影数据的改进粒子群优化随机CT重建 下载: 4158次
1 引言
计机断层成像(CT)是依据从多角度方向检测到的经过被探测物的X射线衰减量,重建出被探测物某一希望层面的轴向X射线图像的透视成像技术[1]。用于CT重建的投影数据通常需要满足Tuy-Smith数据完备性条件[2-3],即需要在360°或180°内均匀且密集地采集完整的投影数据,才能获得精确的重建效果。然而受X射线剂量、采集时间、扫描工作范围等的限制,在诸多实际应用中,只能在采集范围受限、采集角度稀疏的投影数据上实现CT重建。例如,脑CT中通过减少投影的采样角度来降低射线对脑的损伤,工业CT中对大型不规则工件进行内部缺陷的无损探伤,都会得到不完备投影数据[4-5]。
在进行不完备投影数据的重建时,通常要利用待重建图像的先验条件,如全变分(TV)约束等将不完备投影数据下的不适定问题转化为具有约束条件的优化问题,以降低重建对数据完备性的依赖[6-7]。针对角度受限型投影数据的重建问题,2006年Sidky等[8]利用CT图像梯度稀疏性先验知识提出基于凸集投影和TV最小化(POCS-TV)的算法,实现了角度受限下90°投影范围内64组投影数据的重建,但在重建轮廓处存在伪影。为了提高算法的重建质量,通过设计自适应松弛因子[9]或改进TV约束项等方法[10-11],抑制角度受限下不完备投影数据的重建伪影,但受制于梯度下降等确定型求解算法的局限,上述算法并不能在有限时间内得到高质量的重建结果[12]。在效率优化方面,研究者针对360°或180°投影范围内角度完备的稀疏采样数据开展了重建算法优化研究,针对梯度下降算法收敛速度慢的缺点,利用二阶梯度近似[13]、共轭梯度下降[14]和软阈值滤波[15]等方法进行改进,提高了角度完备下稀疏投影数据的重建效率。但当面对完备性更低的角度受限下稀疏投影数据的重建问题时,由于数据完备性、对称性更低,重建算法的搜索空间非常巨大,前述这些改进算法无法解决。传统的基于凸优化思路的确定型重建算法,从单一初始解开始沿一定的下降路径迭代求解,无法在有限时间内遍历这类不完备数据带来的巨大搜索空间,找到符合重建要求的解[16-18]。这使得角度受限下稀疏投影数据的求解成为CT重建的研究难点之一。
随机优化[19-20]通过随机生成样本保证解的多样性,搜索中用群体的随机采样来代表整个群体[16],由适应度评价解的品质,最终由种群进化得到全局最优解,更适于不完备数据下的全局搜索。然而,在求解数据完备性很低的不适定问题时,其计算成本过高;因此,虽然在CT图像复原、脑CT图像配准和分割等相关研究上,研究人员已开展了随机优化方法的探索[21-23],但在不完备投影数据的CT重建上仍未见到采用随机算法的研究报道。
针对角度受限下稀疏投影数据的CT重建问题,本文提出一种基于粒子群优化(PSO)的随机稀疏重建算法。首先,通过随机策略生成具有多样性的初始种群,以保证算法的搜索能力;然后,随机选择梯度下降或基于个体历史最优解和全局历史最优解的随机下降方向进行迭代,以兼顾算法效率和搜索方向的多样性;最后,基于适应度评价,有针对性地重新生成随机初始种群,强制跳离局部最优。角度受限下稀疏投影数据的重建实验表明,与现有CT重建算法相比,本文算法在保证算法效率的同时,明显提升了重建效果和稳健性,尤其在含噪投影数据的重建上,具有更强的适应性和稳定性。
2 不完备投影数据的重建模型
X射线的反投影重建是指X射线源和探测器同时绕待重建对象的中心旋转扫描,采集不同方向的衰减后的X射线投影数据,并据此反推算出反映对象密度分布的灰度图像的过程。离散化的投影过程如
X射线的反投影重建可归结为如下的线性方程求解[1]:
式中x=
为了获得精确的重建结果,通常要求在360°或180°内均匀且密集地采集完整的投影数据。然而,如前所述,医学和工业CT中都存在如
图 2. 不完备投影数据示意图。(a)投影角度受限;(b)投影角度稀疏;(c)投影角度受限且稀疏
Fig. 2. Illustration of incomplete projection data. (a) Limited-angle data; (b) sparse view data; (c) limited-angle sparse view data
为改善不完备数据带来的求解不适定性,Sidky等[8]基于CT重建图像的梯度稀疏性先验知识,引入TV正则化项,得到(3)式的不完备投影数据重建模型,可解决投影角度受限、稀疏或完备性更低的角度受限下稀疏投影数据的重建问题。
式中‖x‖TV=∑
在这一求解框架下,Sidky等[8]基于凸优化的思路提出了经典的传统的CT重建算法(POCS-TV算法),对
然而,在对
3 角度受限下稀疏投影数据的改进PSO随机重建算法
3.1 PSO算法
PSO是由Kennedy[26]提出的一种进化计算技术。其主要思想是利用群体中个体对信息的共享,使整个群体的运动在问题求解空间中从无序到有序演化,从而获得最优解[26-27]。算法主要步骤[28]如下:
1) 随机生成一定规模的初始种群,保证样本的多样性。
2) 计算粒子适应度值来评价解的品质,更新个体和全局历史最优解。
3) 粒子依据个体和群体的历史最优解更新自身的运动速度和位置。
4) 重复执行步骤2)~3),直至迭代次数达到设定值或相邻2次迭代的偏差小于给定阈值。
作为一种随机搜索策略,PSO避免了遗传算法中“交叉”和“变异”的复杂规则,种群更新方式简单,实现容易,精度高,收敛快[29]。但面对不完备数据带来的巨大搜索空间时,种群规模要保持在粒子自身维度的5~10倍以上,才能使粒子分散分布在搜索空间中,提供足够的搜索能力。以CT重建中常见的128 pixel×128 pixel图像为例,粒子群的种群规模要求在105以上,种群中每个粒子的维度在104以上,每次迭代的计算量和存储代价都很大,寻优效率低,并不能完全解决角度受限下稀疏投影数据的重建问题。
3.2 基于PSO的随机稀疏重建算法
对
3.2.1 基于随机策略的初始解生成
梯度下降算法在对由不完备投影数据带来的巨大搜索空间进行寻优时,算法初始解在解空间中的位置严重影响着最终解的质量。为保证初始解的多样性,本文对目标函数引入随机策略来生成规模为K的初始解种群。首先,对CT投影重建物理模型Ax=p求其伪逆,生成靠近最优解的初始解:
以保证算法的搜索效率。然后,为增加初始解的多样性,在
式中
3.2.2 基于随机策略的算法迭代方向更新
梯度下降算法中,每次的迭代方向即为目标函数的优化方向,但单一下降方向易陷入局部最优。为保证搜索方向的多样性,本文融合粒子群的随机更新策略,在每次迭代中随机更新算法的搜索方向,提高全局搜索能力。算法迭代更新的策略如下。
1) 对解的取值进行约束,保证随机解的有效性:
2) 依据(3)式得到对应的适应度函数:
对随机解进行评价,并计算得到当前的个体历史最优解
3) 对样本进行随机迭代更新:
式中Δ
在这一迭代策略下,当产生的随机数大于等于阈值时,采用梯度下降方向更新样本,降低了大规模空间遍历中的不确定性,使得粒子群规模K可保持在5~10的较低水平,一定程度上保证了算法的重建效率;否则,采用基于个体历史最优和全局历史最优的随机方向进行迭代更新,保证搜索方向的多样性。
阈值越大,粒子群随机方向的选择概率越大,全局搜索能力提高,但算法效率下降;反之,则更多选择确定性的梯度下降方向,以提高算法效率,但却牺牲了全局搜索能力。此外,在基于随机策略的初始解生成中,初始种群规模K越大,粒子群更新的计算量也会相应增加。此时,应减小阈值,增加确定性下降方向的选择概率,以保证算法效率。但如前所述,样本的随机迭代更新中引入了梯度下降方向,降低了对粒子群种群规模和迭代方向随机性的要求;因此,在实际应用中只需选取较小的K和V值,即可在保证算法效率的同时获得好的重建效果。
3.2.3 跳离局部最优的随机策略
对角度受限下的稀疏投影数据进行迭代重建时,常因搜索空间巨大,使解连续多次无法更新,陷入局部最优。实验观察发现,陷入局部最优的重建图像主要表现为平滑区出现噪声、边缘不清晰或整体过平滑3种情况。针对这一问题,本文提出如下的随机策略,来跳离局部最优:
1) 当连续3次迭代没有更新Pg时,执行如下操作
式中g为3×3的高斯滤波模板,⊗为卷积操作,Pg_smooth为滤波后的样本,edge(Pg)为对Pg进行sobel边缘检测的结果,Pg_sharpen为边缘增强后的样本,rand[length(Pg)]生成维度与Pg相同的随机向量,其元素为0~1之间的随机数,Pg_random为随机扰动后的样本。
2) 用适应度函数F(x)=‖Ax-p
作为新的初始解。
3) 应用(5)式重新生成随机初始种群。
在这一随机策略中,适应度函数是依据算法的目标函数而确定的,兼顾了投影重建的数据保真、噪声滤除和边缘增强。适应度最优保证了后续的种群更新朝向目标函数的下降方向。因此,通过比较图像滤波、边缘增强和随机扰动3种操作的适应度,可有选择性地确定当前局部最优下应采取的图像操作,再重新生成随机初始种群,使得平滑区出现噪声、边缘不清晰或者整体过平滑等情况强制跳离局部最优。
综合以上基于随机策略的初始解生成、算法迭代方向更新和跳离局部最优,本文提出的基于PSO的随机稀疏重建算法具体流程如下。
输入: 投影数据p,投影矩阵A
输出: 重建图像x
1) 初始化。置当前迭代次数t=0,
2) 根据(4)式计算初始解。
3) 根据(5)式生成随机初始种群。
4) 根据(6)式对
5) 根据(7)式计算解的适应度。
6) 计算得到当前的个体历史最优解
7) 更新进化停止计数值q。若步骤6中
8) 若q<3,执行样本随机迭代更新;否则执行跳离局部最优。
① 根据(11)式对样本
② 跳离局部最优。
a 对当前的
b 根据(15)式计算上述3种操作后适应度值最优的样本
c 重新初始化种群。
9) 迭代终止条件。若满足t=T或
4 角度受限下稀疏投影数据的重建实验结果及分析
以128 pixel×128 pixel的二维Shepp Logan模型、Forbild Head模型为CT重建对象,投影数据生成模型如下:
1) X射线探测器由256个接收器阵列组成,单个接收单元大小为0.5 mm。
2) 投影角度为(0°,90°),投影数据的采集组数为30。
3) 待重建图像的每一个像素的物理尺寸为0.5 mm×0.5 mm。
为了验证本文算法的有效性,在无噪声的X射线投影数据和含噪声的X射线投影数据上进行重建实验,通过重建效果、重建误差和重建效率来对比算法性能。
对比算法包括传统凸优化(L2-TV)算法[30-32]、基于凸优化的改进算法——随机平均梯度(SAG)算法[33-34]、POCS-TV算法[8]和PSO算法[35]。为更好地比较各算法的重建效果,设置PSO算法的种群规模K=10000,迭代次数小于等于5000。其他各算法的迭代停止条件:相邻2次迭代的重建结果差值小于10-3,最大迭代次数小于等于1000。
本文算法的参数设置为:T=1000,K=5,V=0.4,c1=c2=2,w=wmax-(wmax-wmin)(t/T),式中wmax=0.95,wmin=0.4,t表示当前迭代次数。
本文实验的软硬件环境为:Matlab R2016a,Intel(R)Core(TM)i7-4790K CPU@4.00 GHz,32 GB内存。
4.1 角度受限下无噪声稀疏投影数据的重建实验结果
由
图 3. Shepp Logan图像的无噪声稀疏投影数据重建结果。(a) Shepp Logan原图;(b) L2-TV算法;(c) SAG算法;(d) POCS-TV算法;(e) PSO算法;(f)本文算法
Fig. 3. Reconstruction results from noise-free Shepp sparse Logan projection data. (a) Original image of Shepp Logan; (b) L2-TV algorithm; (c) SAG algorithm; (d) POCS-TV algorithm; (e) PSO algorithm; (f) proposed algorithm
图 4. Forbild Head图像的无噪声稀疏投影数据重建结果。(a) Forbild Head原图;(b) L2-TV算法;(c) SAG算法;(d) POCS-TV算法;(e) PSO算法;(f)本文算
Fig. 4. Reconstruction results from noise-free Forbild Head sparse projection data. (a) Original image of Forbild Head; (b) L2-TV algorithm; (c) SAG algorithm; (d) POCS-TV algorithm; (e) PSO algorithm; (f) proposed algorithm
为更好地展示本文算法的重建效果,原始图像与各重建图的水平剖线及垂直剖线对比结果如
图 5. 无噪声情况下Shepp Logan图像投影数据的水平剖线(a)及垂直剖线(b)对比
Fig. 5. Comparison of horizontal (a) and vertical (b) profiles with noise-free Shepp Logan projection data
图 6. 无噪声情况下Forbild Head图像投影数据的水平剖线(a)及垂直剖线(b)对比
Fig. 6. Comparison of horizontal (a) and vertical (b) profiles with noise-free Forbild Head projection data
此外,为客观对比各算法的重建效果,采用重建结果与原始图像的误差值进行算法评价,结果如
表 1. 无噪声情况下各算法重建误差比较
Table 1. Comparison of reconstruction error with noise-free projection data
|
4.2 角度受限下含噪声稀疏投影数据的重建实验结果
为验证本文算法的稳健性和适应性,对角度受限下含噪声稀疏投影数据的重建结果,从重建图像效果、剖线对比和重建误差3个方面进行算法性能对比。含噪声投影数据由无噪声投影数据加上标准差为1.5的较强高斯噪声得到。
从
图 7. Shepp Logan图像的含噪声投影数据重建结果。(a) Shepp Logan原图;(b) L2-TV算法;(c) SAG算法;(d) POCS-TV算法;(e) PSO算法;(f)本文算法
Fig. 7. Reconstruction results from Shepp Logan projection data with Gaussian noise. (a) Original image of Shepp Logan; (b) L2-TV algorithm; (c) SAG algorithm; (d) POCS-TV algorithm; (e) PSO algorithm; (f) proposed algorithm
图 8. Forbild Head图像的含噪声投影数据重建结果。(a) Forbild Head原图;(b) L2-TV算法;(c) SAG算法;(d) POCS-TV算法;(e) PSO算法;(f)本文算法
Fig. 8. Reconstruction results from Forbild Head projection data with Gaussian noise. (a) Original image of Forbild Head; (b) L2-TV algorithm; (c) SAG algorithm; (d) POCS-TV algorithm; (e) PSO algorithm; (f) proposed algorithm
图 9. 含噪声情况下Shepp Logan图像投影数据的水平剖线(a)及垂直剖线(b)对比
Fig. 9. Comparison of horizontal (a) and vertical (b) profiles with noisy Shepp Logan projection data
图 10. 含噪声情况下Forbild Head图像投影数据的水平剖线(a)及垂直剖线(b)对比
Fig. 10. Comparison of horizontal (a) and vertical (b) profiles with noisy Forbild Head projection data
表 2. 含噪声情况下各算法重建误差比较
Table 2. Comparison of reconstruction error with noisy projection data
|
4.3 重建算法效率对比
为验证本文算法的效率,对PSO算法以外的各算法,对比其重建误差随迭代次数和时间变化的情况,因为PSO算法无法在有限的迭代次数、计算时间内获得重建结果,所以不参与对比。
图 11. 各算法重建误差随迭代次数变化的曲线。(a)无噪声的Shepp Logan投影数据;(b)无噪声的Forbild Head投影数据;(c)含噪声的Shepp Logan投影数据;(d)含噪声的Forbild Head投影数据
Fig. 11. Reconstruction error curves with the change of iteration number of different algorithms. (a) Noise-free Shepp Logan projection data; (b) noise-free Forbild Head projection data; (c) noisy Shepp Logan projection data; (d) noisy Forbild Head projection data
图 12. 各算法重建误差随迭代时间变化的曲线。(a)无噪声的Shepp Logan投影数据;(b)无噪声的Forbild Head投影数据;(c)含噪声的Shepp Logan投影数据;(d)含噪声的Forbild Head投影数据
Fig. 12. Reconstruction error curves with the change of iteration time of different algorithms. (a) Noise-free Shepp Logan projection data; (b) noise-free Forbild Head projection data; (c) noisy Shepp Logan projection data; (d) noisy Forbild Head projection data
由
5 结论
针对角度受限下稀疏投影数据的重建,将梯度下降法和PSO策略相融合,解决了初始解多样性、算法迭代方向的随机更新和陷入局部最优等问题。其中,梯度下降方向的引入降低了大规模空间遍历中的不确定性,使得算法能够在粒子群的种群规模较小时实现对不完备投影数据的高效、高质量重建。角度受限下无噪声和含噪声稀疏投影数据的对比实验结果显示,本文算法既保证了算法效率,又在重建质量和算法稳健性上有明显的优势。
本文的实验设计对投影的角度范围和稀疏性都提出了非常严苛的要求,得到的投影数据完备性和对称性很低;因此,在含噪声投影数据的重建图像中,与投影方向垂直的边缘上还存在一定的重建误差。后续研究将考虑投影数据在不同方向上的完备性,进一步提高算法的重建能力。算法迭代方向的随机更新中也可探索自适应地调整梯度下降与粒子群随机方向的比重,提高算法的重建效率。
此外,依据目标函数对本文算法中的适应度评价函数做相应修改,可将随机稀疏重建算法推广应用到其他不完备观测数据下的高维信号盲复原问题中,提升算法性能和效率。
[1] 庄天戈. CT原理与算法[M]. 上海: 上海交通大学出版社, 1992: 77- 97.
[5] 张朝宗. 工业CT技术和原理[M]. 北京: 科学出版社, 2009: 46.
[6] 王林元, 刘宏奎, 李磊, 等. 基于稀疏优化的计算机断层成像图像不完全角度重建综述[J]. 物理学报, 2014, 63(20): 208702.
[7] 郭红波, 贺小伟, 侯榆青, 等. 基于非凸稀疏正则的荧光分子断层成像[J]. 光学学报, 2015, 35(7): 0717001.
[11] 李毅. 有限角度三维CT图像重建算法研究[D]. 太原: 中北大学, 2011.
LiY. The study of limited angle three-dimensional CT image reconstruction algorithm[D]. Taiyuan: North University of China, 2011.
[14] 邹晶, 孙艳勤, 张朋. 由少量投影数据快速重建图像的迭代算法[J]. 光学学报, 2009, 29(5): 1198-1204.
[18] BianchiniS. On the Euler-Lagrange equation for a variational problem[M]. [S.l.]: Birkhäuser Basel, 2007: 61- 77.
[19] FreyJ. Introduction to stochastic search and optimization: estimation, simulation, and control[M]. Hoboken: John Wiley & Sons, 2003: 368- 369.
[20] Solis F J. Wets R J B. Minimization by random search techniques[J]. Mathematics of Operations Research, 1981, 6(1): 19-30.
[21] 陈昊, 颜秀铭, 曹福凯, 等. 粒子群优化算法在CT图像复原中的应用研究[ C]. 全国光谱仪器与分析学术研讨会, 2007.
[22] 于泽. 脑CT图像配准中自适应粒子群算法的应用研究[D]. 大连: 大连海事大学, 2010.
YuZ. Research on cerebral CT image registration by adaptive particle swarm optimization[D]. Dalian: Dalian Maritime University, 2010.
[24] IdierJ. Bayesian approach to inverse problems[M]. Hoboken: John Wiley & Sons, 2008.
[25] 卢孝强, 孙恰. 基于乘性正则化的有限角度CT重建算法[J]. 光学学报, 2010, 30(5): 1285-1290.
[26] KennedyJ. Particle swarm optimization[M] ∥ Encyclopedia of Machine Learning. New York: Springer, 2011: 760- 766.
[28] 高璇. 粒子群算法优化及其在图像检索中的应用研究[D]. 西安: 西安电子科技大学, 2013.
GaoX. Research on particle swarm optimization and its application in image retrieval[D]. Xi'an:Xidian University, 2013.
[29] 方峻. 粒子群算法及其应用研究[D]. 成都: 电子科技大学, 2006.
[30] NojimaY, Chen YW, Han XH. Image and video restoration with TV/L2-norm constraint[C]. International Conference on Computer Information Systems and Industrial Applications, 2015: 642- 644.
[34] SchmidtM, Roux NL, BachF. Minimizing finite sums with the stochastic average gradient[M]. New York: Springer-Verlag, 2017.
Article Outline
高红霞, 罗澜, 骆英浩, 陈展鸿, 马鸽. 角度受限下稀疏投影数据的改进粒子群优化随机CT重建[J]. 光学学报, 2018, 38(1): 0111003. Hongxia Gao, Lan Luo, Yinghao Luo, Zhanhong Chen, Ge Ma. Improved Stochastic CT Reconstruction Based on Particle Swarm Optimization for Limited-Angle Sparse Projection Data[J]. Acta Optica Sinica, 2018, 38(1): 0111003.