672. RSNA Intracranial Aneurysm Detection | rsna-intracranial-aneurysm-detection
基于给定的 train_localizers.csv 采样阳性(存在动脉瘤)和阴性(不存在动脉瘤)病例,训练一个简单的图像分类器。使用周围图像作为图像的 R 和 B 通道。使用两种不同的步长进行堆叠。步长 2 意味着不堆叠三个连续帧,而是留一个间隙取下一帧。骨干网络使用 Convnext Base Dinov3,配合一个简单的分类头。
valid_transforms = A.Compose([
A.CenterCrop(448, 448),
])
train_transforms = A.Compose([
A.ShiftScaleRotate(rotate_limit=(-5, 5), p=0.5),
A.RandomCrop(448, 448),
A.RandomRotate90(p=1.0),
A.OneOf([
A.GridDropout(ratio=0.4, p=0.5),
A.CoarseDropout(max_holes=25,
max_height=int(0.2*448),
max_width=int(0.2*448),
min_holes=10,
min_height=int(0.1*448),
min_width=int(0.1*448),
p=0.5),
A.GridDistortion(p=1.0),
], p=0.5),
])
图像在预处理期间被调整为 512x512。采用旋转不变训练,并通过同时翻转标签应用左右翻转。
为了采样第一阶段的负样本 spot,使用来自无动脉瘤序列的图像,以及来自有动脉瘤序列的图像。在第二种情况下,排除存在动脉瘤的 spot 及其周围边界。此外,基于早期训练分类器的 OOF 预测,模型预测为假阳性的 spot 会被更频繁地采样。
基于第一阶段提取的特征,针对每个 UID 的完整序列训练 Transformer。在此阶段进行增强时,提取步长 1 和步长 2 的序列并用于第二阶段的训练。