提升手术场景分割在未知中心和模态下的泛化能力。
RobustSurg: Tackling domain generalisation for out-of-distribution surgical scene segmentation
- 结合风格与内容信息,通过实例归一化和特征协方差映射增强鲁棒性。
- 在未见中心数据上相比基线提升23%,优于当前最优方法10%-32%。
- 新增多中心手术数据集,适合医疗视觉与领域泛化研究者。
尽管深度学习在单中心、单成像模态的手术场景分割上取得进展,但其在未见分布(如其他中心)和未见模态下泛化能力较差。现有针对自然场景的域泛化方法难以直接应用于手术场景,因手术图像视觉线索少且场景差异大。本文提出RobustSurg,通过挖掘手术场景中的风格与内容信息,降低外观变化影响(如出血、成像伪影),利用实例归一化与特征协方差映射实现更鲁棒的特征表示。为避免丢失关键任务特征,设计了恢复模块保留在残差网络主干中。同时构建新数据集CholecSeg8K+HeiCholSeg,用于评估跨中心泛化性能。实验表明,模型在未见中心数据HeiCholSeg上相较DeepLabv3+基线提升近23%,相比当前最优方法提升10%-32%;在EndoUDA结直肠镜数据集上,较基线提升约22%,较近期SOTA提升约11%。
原文摘要 · Abstract (English)
While recent advances in deep learning for surgical scene segmentation have demonstrated promising results on single-centre and single-imaging modality data, these methods usually do not generalise to unseen distribution (i.e., from other centres) and unseen modalities. Current literature for tackling generalisation on out-of-distribution data and domain gaps due to modality changes has been widely researched but mostly for natural scene data. However, these methods cannot be directly applied to the surgical scenes due to limited visual cues and often extremely diverse scenarios compared to the natural scene data. Inspired by these works in natural scenes to push generalisability on OOD data, we hypothesise that exploiting the style and content information in the surgical scenes could minimise the appearances, making it less variable to sudden changes such as blood or imaging artefacts. This can be achieved by performing instance normalisation and feature covariance mapping techniques for robust and generalisable feature representations. Further, to eliminate the risk of removing salient feature representation associated with the objects of interest, we introduce a restitution module within the feature learning ResNet backbone that can enable the retention of useful task-relevant features. To tackle the lack of multiclass and multicentre data for surgical scene segmentation, we also provide a newly curated dataset that can be vital for addressing generalisability in this domain. Our proposed RobustSurg obtained nearly 23% improvement on the baseline DeepLabv3+ and from 10-32% improvement on the SOTA in terms of mean IoU score on an unseen centre HeiCholSeg dataset when trained on CholecSeg8K. Similarly, RobustSurg also obtained nearly 22% improvement over the baseline and nearly 11% improvement on a recent SOTA method for the target set of the EndoUDA polyp dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。