首个涵盖多种腹部病变的3D多类别分割数据集,提升医学影像诊断准确率。
MSWAL: 3D Multi-class Segmentation of Whole Abdominal Lesions Dataset
- 构建首个覆盖多种常见腹部病变的3D多类别分割数据集。
- 在肝癌、肾癌等病灶上提升3%以上分割精度,跨数据集迁移效果显著。
- 适合医学影像分析、深度学习在医疗应用的研究者使用。
随着腹腔疾病发病率持续上升,亟需引入新技术辅助诊疗。尽管深度学习已在辅助诊断中取得进展,但现有模型因训练数据缺少典型腹腔病灶标注,难以有效分割常见病变。为此,我们提出MSWAL,首个针对全腹部病变的3D多类别分割数据集,涵盖胆结石、肾结石、肝肿瘤、肾肿瘤、胰腺癌、肝囊肿及肾囊肿等多种病灶类型。数据来自694名患者(共191,417张切片),覆盖不同性别与扫描阶段,具备强鲁棒性与泛化能力。将MSWAL用于迁移学习,在LiTS和KiTS两个公开数据集上分别实现肝肿瘤Dice系数提升3.00%、肾肿瘤提升0.89%,证明其丰富标注与多样病灶有助于跨域学习。此外,我们提出Inception nnU-Net框架,融合Inception模块与nnU-Net,通过多感受野特征提取,在体素级DSC与区域级F1上均优于当前先进算法。数据集将在论文接收后发布,代码已开源至https://github.com/tiuxuxsh76075/MSWAL-。
原文摘要 · Abstract (English)
With the significantly increasing incidence and prevalence of abdominal diseases, there is a need to embrace greater use of new innovations and technology for the diagnosis and treatment of patients. Although deep-learning methods have notably been developed to assist radiologists in diagnosing abdominal diseases, existing models have the restricted ability to segment common lesions in the abdomen due to missing annotations for typical abdominal pathologies in their training datasets. To address the limitation, we introduce MSWAL, the first 3D Multi-class Segmentation of the Whole Abdominal Lesions dataset, which broadens the coverage of various common lesion types, such as gallstones, kidney stones, liver tumors, kidney tumors, pancreatic cancer, liver cysts, and kidney cysts. With CT scans collected from 694 patients (191,417 slices) of different genders across various scanning phases, MSWAL demonstrates strong robustness and generalizability. The transfer learning experiment from MSWAL to two public datasets, LiTS and KiTS, effectively demonstrates consistent improvements, with Dice Similarity Coefficient (DSC) increase of 3.00% for liver tumors and 0.89% for kidney tumors, demonstrating that the comprehensive annotations and diverse lesion types in MSWAL facilitate effective learning across different domains and data distributions. Furthermore, we propose Inception nnU-Net, a novel segmentation framework that effectively integrates an Inception module with the nnU-Net architecture to extract information from different receptive fields, achieving significant enhancement in both voxel-level DSC and region-level F1 compared to the cutting-edge public algorithms on MSWAL. Our dataset will be released after being accepted, and the code is publicly released at https://github.com/tiuxuxsh76075/MSWAL-.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。