arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.00587eess.IVcs.AI

基于深度学习、注意力机制和基于能量的不确定性预测的脑肿瘤分割

Brain Tumor Segmentation Based on Deep Learning, Attention Mechanisms, and Energy-Based Uncertainty Prediction

Mills E. Godwin High School(米尔斯·E·戈德温高中)

Zachary Schwehr, Sriman Achanta

AI总结:

本文提出一种结合感兴趣区域检测、软注意力全卷积自编码器及基于能量的不确定性预测的脑肿瘤分割方法,在BraTS基准上达到最先进性能。

摘要:

脑肿瘤是最致命的癌症之一,死亡率超过80%。快速准确的诊断对于提高生存机会至关重要。然而,在医学分析中,脑肿瘤的手动标注和分割可能是一项复杂的任务。通常需要分析多种MRI模态,因为它们提供了关于肿瘤区域的独特信息。尽管这些MRI模态有助于胶质瘤的分割,但它们往往会增加过拟合和计算量。本文提出了一种在数据预处理阶段实现的感兴趣区域检测算法,用于定位显著特征并去除无关的MRI数据。这减小了输入尺寸,允许更激进的数据增强和更深的神经网络。在MRI模态预处理之后,一个带有软注意力的全卷积自编码器对不同脑部MRI进行分割。当这些深度学习算法在实践中实施时,分析师和医生无法区分准确和不准确的预测。随后,使用测试时增强和基于能量的模型进行基于体素的不确定性预测。在BraTS基准上进行了实验,并取得了最先进的分割性能。此外,使用定性结果来评估分割模型和不确定性预测。

英文摘要:

Brain tumors are one of the deadliest forms of cancer with a mortality rate of over 80%. A quick and accurate diagnosis is crucial to increase the chance of survival. However, in medical analysis, the manual annotation and segmentation of a brain tumor can be a complicated task. Multiple MRI modalities are typically analyzed as they provide unique information regarding the tumor regions. Although these MRI modalities are helpful for segmenting gliomas, they tend to increase overfitting and computation. This paper proposes a region of interest detection algorithm that is implemented during data preprocessing to locate salient features and remove extraneous MRI data. This decreases the input size, allowing for more aggressive data augmentations and deeper neural networks. Following the preprocessing of the MRI modalities, a fully convolutional autoencoder with soft attention segments the different brain MRIs. When these deep learning algorithms are implemented in practice, analysts and physicians cannot differentiate between accurate and inaccurate predictions. Subsequently, test time augmentations and an energy-based model were used for voxel-based uncertainty predictions. Experimentation was conducted on the BraTS benchmarks and achieved state-of-the-art segmentation performance. Additionally, qualitative results were used to assess the segmentation models and uncertainty predictions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑