广东工业大学学报2024,Vol.41Issue(6):60-68,9.DOI:10.12052/gdutxb.230156
基于方向约束强化学习的左心室内膜分割
Segmentation of Left Ventricular Endocardium Using Direction-constrained Reinforcement Learning
摘要
Abstract
Accurate segmentation of the left ventricular endocardium from cardiac magnetic resonance imaging to obtain the left ventricular region is an important step in the analysis of cardiac function.It is noted that reinforcement learning is prone to localization deviation in left ventricular endocardium segmentation by locating the left ventricular endocardial edges,leading to a performance decrease in the segmentation.To address this,this paper proposes a direction-constrained reinforcement learning method for left ventricular endocardium segmentation,which divides the segmentation task into two stages.In the first stage,the proposed method extracts global edge features of the endocardium,and in the second stage,reinforcement learning is used to iteratively locate the endocardial edge points to obtain the edge,obtaining the segmented left ventricular endocardium.The proposed method constrains the direction of agent positioning,which can reduce the localization deviation and overlap,such that the segmentation accuracy can be improved.Finally,the experimental results on two public datasets,including the Automated Cardiac Diagnosis Challenge(ACDC)and Sunnybrook Cardiac MR Left Ventricle Segmentation Challenge(Sunnybrook),show that the proposed method has higher accuracy than the compared methods.Specifically,the F1-score of the proposed method are 0.9482 and 0.9387,and the Average perpendicular distance(APD)are 3.5863 and 4.9447,which can effectively segment the left ventricular endocardium.关键词
图像分割/变换器/深度强化学习/边缘定位Key words
image segmentation/transformer/deep reinforcement learning/edge localization分类
信息技术与安全科学引用本文复制引用
曾安,庞耀幸,潘丹,赵靖亮..基于方向约束强化学习的左心室内膜分割[J].广东工业大学学报,2024,41(6):60-68,9.基金项目
国家自然科学基金资助项目(61976058,92267107) (61976058,92267107)
广东省重点领域研发计划项目(2021B0101220006) (2021B0101220006)
广东省科技计划项目(2019A050510041) (2019A050510041)
广东省自然科学基金资助项目(2021A1515012300) (2021A1515012300)
广州市科技计划项目(202103000034) (202103000034)