SegHeD+融合解剖约束与病灶增强,实现多中心脑卒中病灶精准分割。
SegHeD+: Segmentation of Heterogeneous Data for Multiple Sclerosis Lesions with Anatomical Constraints and Lesion-aware Augmentation
- 引入纵向、解剖和体积约束,融合医学先验知识提升分割精度
- 在5个数据集上对新发与消失病灶的分割准确率均超现有方法
- 适合处理多中心异构数据,助力临床随访与疾病评估
评估脑磁共振图像中多发性硬化(MS)病灶及其随时间演变对诊断与监测至关重要。机器学习模型在自动分割病灶方面展现出潜力,但通常需要大规模标注数据。然而,MS影像数据集规模有限,分布于多个医院,格式各异(如横断面或纵向),标注风格不一。这种数据异质性阻碍了统一模型的构建。为此,我们提出SegHeD+,一种可处理多数据集与多任务的分割模型,能适应异构输入,并对所有类型病灶(包括新发与消失病灶)进行分割。通过引入纵向、解剖与体积约束,融合疾病先验知识;同时采用病灶级数据增强扩充训练集,进一步提升性能。SegHeD+在五个MS数据集上评估,对各类病灶的分割表现均优于多个前沿方法。
原文摘要 · Abstract (English)
Assessing lesions and tracking their progression over time in brain magnetic resonance (MR) images is essential for diagnosing and monitoring multiple sclerosis (MS). Machine learning models have shown promise in automating the segmentation of MS lesions. However, training these models typically requires large, well-annotated datasets. Unfortunately, MS imaging datasets are often limited in size, spread across multiple hospital sites, and exhibit different formats (such as cross-sectional or longitudinal) and annotation styles. This data diversity presents a significant obstacle to developing a unified model for MS lesion segmentation. To address this issue, we introduce SegHeD+, a novel segmentation model that can handle multiple datasets and tasks, accommodating heterogeneous input data and performing segmentation for all lesions, new lesions, and vanishing lesions. We integrate domain knowledge about MS lesions by incorporating longitudinal, anatomical, and volumetric constraints into the segmentation model. Additionally, we perform lesion-level data augmentation to enlarge the training set and further improve segmentation performance. SegHeD+ is evaluated on five MS datasets and demonstrates superior performance in segmenting all, new, and vanishing lesions, surpassing several state-of-the-art methods in the field.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。