arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.01643eess.IVcs.CVcs.LG
收藏

深入研究用于COVID-19检测的空间切片特征学习

A Closer Look at Spatial-Slice Features Learning for COVID-19 Detection

Institute of Data Science, National Cheng Kung University(成功大学数据科学研究所);Department of Electrical Engineering, National Cheng Kung University(成功大学电机工程学系)

Chih-Chung Hsu, Chia-Ming Lee, Yang Fan Chiang, Yi-Shiuan Chou, Chih-Yu Jiang, Shen-Chieh Tai, Chi-Han Tsai

AI总结:

针对CT扫描分辨率差异大和分布外切片冗余问题,提出SSFL++框架结合KDS采样方法,有效过滤OOD数据并减少70%冗余,在仅用1%训练数据下实现高效COVID-19检测。

摘要:

传统计算机断层扫描(CT)成像识别面临两个重大挑战:(1)每次CT扫描的分辨率和尺寸通常存在很大差异,这对模型的输入尺寸和适应性提出了严格要求。(2)CT扫描包含大量分布外(OOD)切片。关键特征可能仅存在于整个CT扫描的特定空间区域和切片中。我们如何有效地找出它们的位置?为了解决这个问题,我们引入了专为CT扫描设计的增强型空间切片特征学习(SSFL++)框架。它旨在过滤掉整个CT扫描中的OOD数据,使我们能够通过总共减少70%的冗余来选择关键的空间切片进行分析。同时,我们提出了基于核密度的切片采样(KDS)方法,以提高训练和推理阶段的稳定性,从而加快收敛速度并提升性能。结果表明,即使仅使用1%的训练数据,我们的模型使用简单的EfficientNet-2D(E2D)模型也展现出优异的性能。我们方法的有效性已在DEF-AI-MIA研讨会与CVPR 2024联合提供的COVID-19-CT-DB数据集上得到验证。源代码可在https://github.com/ming053l/E2D获取。

英文摘要:

Conventional Computed Tomography (CT) imaging recognition faces two significant challenges: (1) There is often considerable variability in the resolution and size of each CT scan, necessitating strict requirements for the input size and adaptability of models. (2) CT-scan contains large number of out-of-distribution (OOD) slices. The crucial features may only be present in specific spatial regions and slices of the entire CT scan. How can we effectively figure out where these are located? To deal with this, we introduce an enhanced Spatial-Slice Feature Learning (SSFL++) framework specifically designed for CT scan. It aim to filter out a OOD data within whole CT scan, enabling our to select crucial spatial-slice for analysis by reducing 70% redundancy totally. Meanwhile, we proposed Kernel-Density-based slice Sampling (KDS) method to improve the stability when training and inference stage, therefore speeding up the rate of convergence and boosting performance. As a result, the experiments demonstrate the promising performance of our model using a simple EfficientNet-2D (E2D) model, even with only 1% of the training data. The efficacy of our approach has been validated on the COVID-19-CT-DB datasets provided by the DEF-AI-MIA workshop, in conjunction with CVPR 2024. Our source code is available at https://github.com/ming053l/E2D

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

↑