研究:EfficientNet 系列在固定分辨率下做 OCT 视网膜分类,B1…

论文AI 评分 58/100arXiv cs.AI
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

一篇 arXiv 论文在固定 224x224 分辨率下,用 ImageNet 预训练的 EfficientNet-B0 到 B7 八个变体(实例化主干参数 4.05M 至 64.10M)对 Kermany 数据集做四分类 SD-OCT 视网膜疾病分类。测试准确率并非随模型变大而单调上升:B0 为 95.93%,B1 最高达 96.95%,B7 为 96.73%。B1 的 Drusen F1 最高(0.8780),B7 的 Drusen 召回率最高(0.9177)。Drusen 类在所有变体上 F1 最低,占各类 F1 平方偏差总和的 82.2%。准确率与参数量的比值从 19.82 降至 1.48 个百分点每百万参数。作者指出,该结果与 ImageNet 上 B7 居首的排序不同,且结果依赖于所给数据划分。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

研究问题与实验设置

论文关注一个具体问题:EfficientNet 在自然图像上按模型规模递增的性能阶梯,能否在固定分辨率的医学图像迁移学习中复现。作者选取 SD-OCT 视网膜疾病四分类任务,使用 Kermany 数据集,把八个 ImageNet 预训练变体 B0 至 B7 统一放在 224x224 输入下比较,实例化后的主干参数量跨度为 4.05M 到 64.10M。灰度图被复制为三通道 RGB,预处理与数据增强由模型内置流程完成。训练分两阶段:先冻结主干只训练分类头,再做部分微调。评估覆盖分类指标、区分度指标、效率指标与可视化分析。

准确率与类别表现

结果没有呈现单调递增。测试准确率最低的是 B0 的 95.93%,最高的是 B1 的 96.95%,B7 回升到 96.73%,形成一次随规模增大的二次回升。按类别看,Drusen 在所有变体上都是最难的一类,F1 最低,并贡献了各类 F1 平方偏差总和的 82.2%;类别难度排序稳定为 Normal > CNV > DME > Drusen。B1 取得最高的 Drusen F1(0.8780),B7 的 Drusen F1 为 0.8720,但召回率最高(0.9177),说明大模型在这一类上偏向多召回。

效率代价与结论边界

效率方面,准确率与参数量的比值从 19.82 个百分点每百万参数降到 1.48,意味着模型越大,每增加单位参数换来的精度越少;B7 同时带来更高的 CPU 延迟。训练-测试差距与测试准确率之间只有弱且不显著的关联(r=0.26,p=0.53)。作者据此认为,EfficientNet 的规模效应依赖具体任务与训练协议,不能直接照搬自然图像的排序——在 ImageNet 上 B7 居首,而本任务中 B1 更优。论文同时声明,跨划分的重复样本尚未独立审计,结论以所给数据划分为条件。

为什么值得看

想用 EfficientNet 做医学影像迁移学习的人可参考:更大模型不一定更准,B1 在精度与效率间更划算。

论文数据集

信源1 家

  1. [1]arXiv cs.AI一手信源Within-Family EfficientNet Scaling under Fixed-Resolution Transfer Learning for Spectral-Domain Optical Coherence Tomography Retinal Disease Classification

关键事实

  • 八个 ImageNet 预训练 EfficientNet 变体在 224x224 分辨率下评测,实例化主干参数 4.05M 至 64.10M[1]

  • 测试准确率从 B0 的 95.93% 到 B1 的 96.95%,B7 为 96.73%,并非单调变化[1]

  • B1 的 Drusen F1 为 0.8780 最高,B7 的 Drusen 召回率 0.9177 最高[1]

  • Drusen 类占各类 F1 平方偏差总和的 82.2%,类别排序为 Normal > CNV > DME > Drusen[1]

  • 准确率与参数量比值由 19.82 降至 1.48 个百分点每百万参数,训练-测试差距与准确率关联不显著(r=0.26,p=0.53)[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。