返回列表

5th place solution

558. Vesuvius Challenge - Ink Detection | vesuvius-challenge-ink-detection

开始: 2023-03-15 结束: 2023-06-14 医学影像分析 数据算法赛
第五名解决方案

第五名解决方案

感谢主办方组织了这场精彩的竞赛。同时感谢用户 @tanakar@samfc10,他们公开的 notebooks 为我参赛提供了极佳的起点。我的解决方案并不复杂,基于3D Resnet模型构建,但在公共榜和私有榜上均取得了良好效果。

模型

采用3D Resnet架构,包括resnet18和resnet34两种模型。

数据预处理

  • 忽略不属于纸草组织的空白切片块
  • 数据划分:4折交叉验证(1, 2a, 2b, 3)
  • 切片块大小:256px
  • 滑动步长:128px
  • 在16至48索引间取32个连续切片

训练策略

  • 训练轮次:50 epochs
  • 混合精度训练:FP16
  • 损失函数:二元交叉熵(BCE)
  • 优化器:AdamW
  • 学习率调度:GradualWarmupScheduler
  • 强化数据增强:Mosaic增强 + Albumentations增强

Mosaic增强效果:

Mosaic增强示例

Albumentations增强效果:

Albumentations增强示例

推理过程

  • 最终集成方案:2折(1, 2a)的3D Resnet18模型 + 4折3D Resnet34模型
  • 推理切片块大小:1024px
  • 推理步长:512px
  • 忽略块边缘区域的预测结果
  • 阈值:0.5
  • 测试时增强(TTA):4种旋转 + 水平/垂直翻转
  • 后处理:去噪操作

最终成绩

公共榜得分:0.804,私有榜得分:0.668

同比赛其他方案