返回列表

11th place solution

448. SIIM-FISABIO-RSNA COVID-19 Detection | siim-covid19-detection

开始: 2021-05-18 结束: 2021-08-09 医学影像分析 数据算法赛
第11名解决方案

第11名解决方案

作者:Roc
比赛:SIIM-FISABIO-RSNA COVID-19 Detection

首先,我要感谢主办方组织了这场非常有趣的比赛,并祝贺本次比赛的所有获胜者。同时感谢我们团队的所有成员 @quochungto@morizin@joven1997@socom20,让我能够在这次比赛中获得第一枚金牌。

研究级别 (Study Level)

  • 使用 SIIM COVID 数据集训练分类(4类:阴性、典型、不确定、非典型)+ 分割(边界框转掩码)任务。
  • 使用 SIIM COVID 数据集 + 测试数据集 + RICORD 数据集 + BIMCV 数据集,并结合伪标签,训练分类(4类)+ 分割任务。(在处理 BIMCV 外部数据时,感谢 @quochungto@joven1997 手动删除了噪声数据。)
  • 数据增强:Resize(调整大小)、HorizontalFlip(水平翻转)、RandomBrightness(随机亮度)、RandomContrast(随机对比度)、OpticalDistortion(光学畸变)、GridDistortion(网格畸变)、HueSaturationValue(色调饱和度亮度)、ShiftScaleRotate(平移缩放旋转)、CutoutV2。
  • 损失函数:
    分类损失 = BiTemperedLogisticLoss 链接
    辅助损失 = 0.9 * BCE + 0.1 * Lovasz。
  • 优化器:Adam,初始学习率为 0.0001。
  • 学习率调度器:GradualWarmupSchedulerV2 链接
  • 测试时增强 (TTA):原图 + 水平翻转。
  • 最终提交模型:包含 4 个模型,具有不同的编码器、解码器和输入尺寸:
    Efficientnet_b5 512, Efficientnet_b7 512, Efficientnetv2_m 512, Swim-Transformer 384。
  • 集成方法:更多细节请见 链接(感谢 @morizin 提供的大量思路)。

本次比赛中使用的研究级别部分源代码已上传至 GitHub:
https://github.com/ChenYingpeng/pl-siim-covid19-detection

  • 无效尝试:
    像 RANZCR 比赛那样分割通道图像;
    使用更多的 TTA(如 scale=1.25, 垂直翻转等);
    在研究级别训练 2 类(无和不透明度);
    在图像级别训练 2 类(无和不透明度)(将整张图像视为“无”作为其边界框)。
架构 CV (5折 mAP*(4/6)) LB
effnet-b5 0.387736 0.535
effnet-b5-pseudo 0.3925984 ---
effnet-b7 0.3821176 0.530
effnetv2-m 0.3869362 0.532
swim-transformer 0.3815662 ---