Xiong Liang, Qin Xiaolin, Liu Xin
Chengdu Institute of Computer Application, Chinese Academy of Sciences, Chengdu 610041, P. R. China.
University of Chinese Academy of Sciences, Beijing 100049, P. R. China.
Sheng Wu Yi Xue Gong Cheng Xue Za Zhi. 2023 Oct 25;40(5):912-919. doi: 10.7507/1001-5515.202211079.
Precise segmentation of lung field is a crucial step in chest radiographic computer-aided diagnosis system. With the development of deep learning, fully convolutional network based models for lung field segmentation have achieved great effect but are poor at accurate identification of the boundary and preserving lung field consistency. To solve this problem, this paper proposed a lung segmentation algorithm based on non-local attention and multi-task learning. Firstly, an encoder-decoder convolutional network based on residual connection was used to extract multi-scale context and predict the boundary of lung. Secondly, a non-local attention mechanism to capture the long-range dependencies between pixels in the boundary regions and global context was proposed to enrich feature of inconsistent region. Thirdly, a multi-task learning to predict lung field based on the enriched feature was conducted. Finally, experiments to evaluate this algorithm were performed on JSRT and Montgomery dataset. The maximum improvement of Dice coefficient and accuracy were 1.99% and 2.27%, respectively, comparing with other representative algorithms. Results show that by enhancing the attention of boundary, this algorithm can improve the accuracy and reduce false segmentation.
肺野的精确分割是胸部X光计算机辅助诊断系统中的关键步骤。随着深度学习的发展,基于全卷积网络的肺野分割模型取得了很好的效果,但在准确识别边界和保持肺野一致性方面表现不佳。为了解决这个问题,本文提出了一种基于非局部注意力和多任务学习的肺分割算法。首先,使用基于残差连接的编码器-解码器卷积网络来提取多尺度上下文并预测肺的边界。其次,提出了一种非局部注意力机制,以捕捉边界区域像素之间的长程依赖关系和全局上下文,从而丰富不一致区域的特征。第三,基于丰富后的特征进行多任务学习以预测肺野。最后,在JSRT和蒙哥马利数据集上进行了评估该算法的实验。与其他代表性算法相比,Dice系数和准确率的最大提升分别为1.99%和2.27%。结果表明,通过增强对边界的注意力,该算法可以提高准确性并减少误分割。