提出新型网络TfeNet,精准分割肺部气道树状结构。
Airway Segmentation Network for Enhanced Tubular Feature Extraction
- 设计方向感知卷积,沿气道方向动态调整采样位置
- 在ATM22数据集上达到94.95%最高分,连续性更优
- 适合需要高精度气道重建的临床导航与机器人系统
CT图像中气道区域的人工标注耗时且依赖经验。自动分割是实现快速支气管镜导航及支气管镜机器人系统临床应用的前提。尽管卷积神经网络已在该领域受到关注,但气道独特的树状结构对传统和可变形卷积构成挑战,常导致细小气道漏检或断裂。为此,本文提出新型管状特征提取网络TfeNet。TfeNet引入方向感知卷积,先对线性卷积核进行空间旋转以调整采样位置,再将变形核表示为3D空间中的线段或折线。此外,基于非对称卷积与残差连接设计了管状特征融合模块(TFFM),增强对细微气道结构的关注。在公开数据集及两个气道分割挑战赛数据集上的大量实验表明,TfeNet相比现有方法在准确性和结构连续性方面表现更优。尤其在当前最大气道分割数据集ATM22上取得94.95%的最高综合得分,并在肺纤维化数据集AIIB23上表现优异。代码已开源:https://github.com/QibiaoWu/TfeNet。
原文摘要 · Abstract (English)
Manual annotation of airway regions in computed tomography images is a time-consuming and expertise-dependent task. Automatic airway segmentation is therefore a prerequisite for enabling rapid bronchoscopic navigation and the clinical deployment of bronchoscopic robotic systems. Although convolutional neural network methods have gained considerable attention in airway segmentation, the unique tree-like structure of airways poses challenges for conventional and deformable convolutions, which often fail to focus on fine airway structures, leading to missed segments and discontinuities. To address this issue, this study proposes a novel tubular feature extraction network, named TfeNet. TfeNet introduces a novel direction-aware convolution operation that first applies spatial rotation transformations to adjust the sampling positions of linear convolution kernels. The deformed kernels are then represented as line segments or polylines in 3D space. Furthermore, a tubular feature fusion module (TFFM) is designed based on asymmetric convolution and residual connection strategies, enhancing the network's focus on subtle airway structures. Extensive experiments conducted on one public dataset and two datasets used in airway segmentation challenges demonstrate that the proposed TfeNet achieves more accuracy and continuous airway structure predictions compared with existing methods. In particular, TfeNet achieves the highest overall score of 94.95% on the current largest airway segmentation dataset, Airway Tree Modeling(ATM22), and demonstrates advanced performance on the lung fibrosis dataset(AIIB23). The code is available at https://github.com/QibiaoWu/TfeNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。