arXiv:2410.22057eess.IVcs.CV2024-10被引 1

用特征引导注意力和课程学习,提升小脑转移瘤分割精度

FANCL: Feature-Guided Attention Network with Curriculum Learning for Brain Metastases Segmentation

  • 通过输入图像与特征建立大小病灶间关联,补偿小病灶信息丢失
  • 在BraTS-METS 2023上Dice达0.872,显著优于基线模型
  • 适合需要高精度分割小病灶的临床影像分析任务

准确分割MR图像中的脑转移瘤(BMs)对患者诊断与随访至关重要。基于深度卷积神经网络(CNN)的方法虽已取得高分割性能,但因卷积和池化操作导致关键特征信息丢失,仍面临小病灶分割难题。此外,脑转移瘤形状不规则,易与正常组织混淆,使模型在训练中难以有效学习肿瘤结构。为此,本文提出一种新模型——特征引导注意力网络结合课程学习(FANCL)。该模型在CNN基础上,利用输入图像及其特征建立不同大小转移瘤间的内在联系,可有效利用大病灶信息补偿小病灶的高层特征损失。同时,采用体素级课程学习策略,帮助模型逐步学习转移瘤的结构与细节,并以不同深度的基线模型作为课程挖掘网络,组织学习进度。在BraTS-METS 2023数据集上的评估结果表明,FANCL显著提升了分割性能,验证了方法的有效性。

原文摘要 · Abstract (English)

Accurate segmentation of brain metastases (BMs) in MR image is crucial for the diagnosis and follow-up of patients. Methods based on deep convolutional neural networks (CNNs) have achieved high segmentation performance. However, due to the loss of critical feature information caused by convolutional and pooling operations, CNNs still face great challenges in small BMs segmentation. Besides, BMs are irregular and easily confused with healthy tissues, which makes it difficult for the model to effectively learn tumor structure during training. To address these issues, this paper proposes a novel model called feature-guided attention network with curriculum learning (FANCL). Based on CNNs, FANCL utilizes the input image and its feature to establish the intrinsic connections between metastases of different sizes, which can effectively compensate for the loss of high-level feature from small tumors with the information of large tumors. Furthermore, FANCL applies the voxel-level curriculum learning strategy to help the model gradually learn the structure and details of BMs. And baseline models of varying depths are employed as curriculum-mining networks for organizing the curriculum progression. The evaluation results on the BraTS-METS 2023 dataset indicate that FANCL significantly improves the segmentation performance, confirming the effectiveness of our method.

医学图像分割注意力机制课程学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。