用真实CT数据集对比多种深度学习重建算法表现。
Benchmarking learned algorithms for computed tomography image reconstruction tasks
- 按方法类型分类,构建可复现的重建评估流程。
- 在低剂量、稀疏角度等任务中,部分算法提升30%以上信噪比。
- 适合想对比或改进医疗影像重建算法的研究者使用。
计算机断层扫描(CT)是广泛应用于多个领域的无创诊断技术。近年来,深度学习在CT图像重建方面取得显著进展,但缺乏大规模开源数据集限制了不同学习方法的比较。为此,我们采用2DeteCT这一真实实验采集的二维CT数据集,对基于机器学习的重建算法进行基准测试。将方法分为后处理网络、可学习/展开迭代法、可学习正则化方法和即插即用类方法,并提供可快速实现与评估的完整流程。通过结构相似性(SSIM)和峰值信噪比(PSNR)等关键指标,验证了各类算法在全数据重建、有限角度、稀疏角度、低剂量及束硬化校正重建任务中的有效性。本研究基于近期发布的实际测量数据集,系统评估了代表性学习重建方法的表现。开放源码工具箱支持方法与任务的可复现部署,便于后续新方法的添加与对比;同时支持以不同方式加载2DeteCT数据,拓展至其他重建问题与场景。
原文摘要 · Abstract (English)
Computed tomography (CT) is a widely used non-invasive diagnostic method in various fields, and recent advances in deep learning have led to significant progress in CT image reconstruction. However, the lack of large-scale, open-access datasets has hindered the comparison of different types of learned methods. To address this gap, we use the 2DeteCT dataset, a real-world experimental computed tomography dataset, for benchmarking machine learning based CT image reconstruction algorithms. We categorize these methods into post-processing networks, learned/unrolled iterative methods, learned regularizer methods, and plug-and-play methods, and provide a pipeline for easy implementation and evaluation. Using key performance metrics, including SSIM and PSNR, our benchmarking results showcase the effectiveness of various algorithms on tasks such as full data reconstruction, limited-angle reconstruction, sparse-angle reconstruction, low-dose reconstruction, and beam-hardening corrected reconstruction. With this benchmarking study, we provide an evaluation of a range of algorithms representative for different categories of learned reconstruction methods on a recently published dataset of real-world experimental CT measurements. The reproducible setup of methods and CT image reconstruction tasks in an open-source toolbox enables straightforward addition and comparison of new methods later on. The toolbox also provides the option to load the 2DeteCT dataset differently for extensions to other problems and different CT reconstruction tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。