用分块拓扑分析提升CT图像分类,更快更准。
A Novel Patch-Based TDA Approach for Computed Tomography Imaging
- 将3D CT图像切块,用分块构造拓扑特征
- 准确率等五项指标平均提升超7%,计算更快
- 适合医学影像分析、想用拓扑方法的研究者
基于计算机断层扫描(CT)的机器学习模型在诊断、分期和预后评估中具有重要潜力,但常依赖手工特征工程。拓扑数据分析(TDA)从代数拓扑角度提取数据深层结构,其中持久同调(PH)可捕捉连通分量、环路和空洞等特征。传统3D立方体复形滤波法适用于网格数据,但高分辨率图像下性能差、计算成本高。本文提出一种新型分块式PH构建方法,专为体积CT数据设计,显著提升性能并降低计算时间。实验对比了立方体复形算法与放射组学特征,结果表明新方法在所有数据集上平均提升准确率7.2%、AUC 3.6%、敏感性2.7%、特异性8.0%、F1分数7.2%。研究还发布了便捷的Python工具包Patch-TDA,便于推广使用。
原文摘要 · Abstract (English)
The development of machine learning models based on computed tomography (CT) imaging has been a major focus due to the promise that imaging holds for diagnosis, staging, and prognostication. These models often rely on the extraction of hand-crafted features where incorporating robust feature engineering improves the performance of these models. Topological data analysis (TDA), based on the mathematical field of algebraic topology, focuses on data from a topological perspective, extracting deeper insight and higher dimensional structures. Persistent homology (PH), a fundamental tool in TDA, extracts topological features such as connected components, cycles, and voids. A popular approach to construct PH from 3D CT images is to utilize 3D cubical complex filtration, a method adapted for grid-structured data. However, this approach is subject to poor performance and high computational cost with higher resolution images. This study introduces a novel patch-based PH construction approach designed for volumetric CT imaging data that improves performance and reduces computational time. This study conducts a series of experiments to comprehensively analyze the performance of the proposed method and benchmarks against the cubical complex algorithm and radiomic features. Our results highlight the dominance of the patch-based TDA approach in terms of both classification performance and computational time. The proposed approach outperformed the cubical complex method and radiomic features, achieving average improvement of 7.2%, 3.6%, 2.7%, 8.0%, and 7.2% in accuracy, AUC, sensitivity, specificity, and F1 score, respectively, across all datasets. Finally, we provide a convenient Python package, Patch-TDA, to facilitate the utilization of the proposed approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。