4 个月前

基于深度局部-全局网络的肺结节分类

基于深度局部-全局网络的肺结节分类

摘要

目的:肺结节具有非常多样化的形状和大小,这使得将其分类为良性或恶性成为一个具有挑战性的问题。本文提出了一种新颖的方法,该方法能够通过全局特征提取器分析结节的形状和大小,同时利用局部特征提取器分析结节的密度和结构,从而预测结节的恶性程度。方法:我们建议使用带有3x3卷积核的残差块(Residual Blocks)进行局部特征提取,并使用非局部块(Non-Local Blocks)提取全局特征。非局部块能够在不使用大量参数的情况下提取全局特征。非局部块的核心思想是在同一特征图上的特征之间应用矩阵乘法。结果:我们在LIDC-IDRI数据集上训练并验证了所提出的方法,该数据集包含1,018个计算机断层扫描(CT)图像。实验设置遵循严格的程序,即10折交叉验证,并忽略了少于3位放射科医生标注的结节。所提出的方法取得了AUC=95.62%的最先进结果,显著优于其他基线方法。结论:我们提出的深度局部-全局网络具备准确提取局部和全局特征的能力。我们的新方法在迁移学习方面超越了包括Densenet和Resnet在内的最先进架构。

代码仓库

mundher/local-global
pytorch
GitHub 中提及

基准测试

基准方法指标
lung-nodule-classification-on-lidc-idriLocal-Global
AUC: 95.62
Accuracy: 88.46
Accuracy(10-fold): 88.46

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于深度局部-全局网络的肺结节分类 | 论文 | HyperAI超神经