Automated segmentation of intracranial hemorrhage (ICH) is an important clinical task limited by the scarcity of high-quality annotated data and variability across datasets. To address these challenges, we propose a novel two-phase learning framework that leverages self-supervised pretraining and multi-label decoding. First, we pretrain a hierarchical vision transformer on over 25,000 head CT cases using a Masked Autoencoder (MAE) strategy to learn robust multi-scale feature representations. For the downstream task, we append a multi-label decoder with dataset-specific heads, which enables joint learning from four distinct ICH segmentation datasets with diverse annotation protocols. A tailored sampling strategy further addresses class imbalance and heterogeneity across datasets. Our results demonstrate superior performance, with the proposed model achieving a mean Dice score of 63.57% (±12.30), surpassing the next best-performing method by 2.69%. Notably, the proposed MAE-based model outperformed its BYOL-pretrained counterpart, suggesting that MAE’s reconstruction-based objective is a more effective pretraining strategy for learning the dense, spatially-rich representations required for this segmentation task.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Self-supervised Pretraining and Multi-label Decoding for Intracranial Hemorrhage Segmentation

  • Aloys Portafaix,
  • Laurent Létourneau-Guillon,
  • Samuel Kadoury

摘要

Automated segmentation of intracranial hemorrhage (ICH) is an important clinical task limited by the scarcity of high-quality annotated data and variability across datasets. To address these challenges, we propose a novel two-phase learning framework that leverages self-supervised pretraining and multi-label decoding. First, we pretrain a hierarchical vision transformer on over 25,000 head CT cases using a Masked Autoencoder (MAE) strategy to learn robust multi-scale feature representations. For the downstream task, we append a multi-label decoder with dataset-specific heads, which enables joint learning from four distinct ICH segmentation datasets with diverse annotation protocols. A tailored sampling strategy further addresses class imbalance and heterogeneity across datasets. Our results demonstrate superior performance, with the proposed model achieving a mean Dice score of 63.57% (±12.30), surpassing the next best-performing method by 2.69%. Notably, the proposed MAE-based model outperformed its BYOL-pretrained counterpart, suggesting that MAE’s reconstruction-based objective is a more effective pretraining strategy for learning the dense, spatially-rich representations required for this segmentation task.