用统一模型精准分割从胶质瘤到儿童脑瘤的多种MRI脑肿瘤。
Unified HT-CNNs Architecture: Transfer Learning for Segmenting Diverse Brain Tumors in MRI from Gliomas to Pediatric Tumors
- 融合混合注意力与卷积网络,通过迁移学习适应不同脑瘤类型。
- 在BraTS验证集上达最优分割效果,DSC与HD95指标领先。
- 适合需要跨瘤种、高泛化能力医学图像分割的研究者。
从3D多模态MRI中精确分割脑肿瘤对诊断和治疗规划至关重要,尤其针对从胶质瘤到儿童脑瘤等多样脑瘤。本文针对BraTS 2023挑战提出一种统一的迁移学习方法,适用于更广泛的脑瘤类型。我们引入HT-CNNs——一种由混合注意力网络与卷积神经网络组成的集成模型,通过迁移学习优化以适应不同脑瘤分割任务。该方法能有效捕捉MRI数据中的空间与上下文特征,基于涵盖常见肿瘤类型的多个国际数据集进行微调。利用预训练模型在大规模数据上的表征能力,并在特定瘤种上进一步微调,显著提升泛化性能。我们对来自多源分布的多样化数据集进行了预处理,确保代表性。通过标准化定量指标在所有瘤种上进行严格评估,结果表明该集成模型在BraTS验证集上优于以往优胜方法。综合定量分析显示,其在DSC和HD95指标上表现优异;定性预测也验证了输出质量。研究强调了迁移学习与集成策略在医学图像分割中的潜力,显著提升临床决策支持与患者照护水平。尽管存在后处理与领域差异挑战,本工作为未来脑瘤分割研究树立新基准。代码与模型已公开,可通过https://hub.docker.com/r/razeineldin/ht-cnns获取。
原文摘要 · Abstract (English)
Accurate segmentation of brain tumors from 3D multimodal MRI is vital for diagnosis and treatment planning across diverse brain tumors. This paper addresses the challenges posed by the BraTS 2023, presenting a unified transfer learning approach that applies to a broader spectrum of brain tumors. We introduce HT-CNNs, an ensemble of Hybrid Transformers and Convolutional Neural Networks optimized through transfer learning for varied brain tumor segmentation. This method captures spatial and contextual details from MRI data, fine-tuned on diverse datasets representing common tumor types. Through transfer learning, HT-CNNs utilize the learned representations from one task to improve generalization in another, harnessing the power of pre-trained models on large datasets and fine-tuning them on specific tumor types. We preprocess diverse datasets from multiple international distributions, ensuring representativeness for the most common brain tumors. Our rigorous evaluation employs standardized quantitative metrics across all tumor types, ensuring robustness and generalizability. The proposed ensemble model achieves superior segmentation results across the BraTS validation datasets over the previous winning methods. Comprehensive quantitative evaluations using the DSC and HD95 demonstrate the effectiveness of our approach. Qualitative segmentation predictions further validate the high-quality outputs produced by our model. Our findings underscore the potential of transfer learning and ensemble approaches in medical image segmentation, indicating a substantial enhancement in clinical decision-making and patient care. Despite facing challenges related to post-processing and domain gaps, our study sets a new precedent for future research for brain tumor segmentation. The docker image for the code and models has been made publicly available, https://hub.docker.com/r/razeineldin/ht-cnns.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。