Local Gamma Augmentation for Ischemic Stroke Lesion Segmentation on MRI
Jon Middleton, Marko Bauer, Kaining Sheng, Jacob Johansen, Mathias Perslev, Silvia Ingala, Mads Nielsen, Akshay Pai
TL;DR
The paper tackles data scarcity in ischemic stroke lesion segmentation on multi-modal MRI by introducing local gamma augmentation, a targeted intensity transformation applied only to pathologic regions identified by segmentation maps. The method defines a local gamma transform and blends augmented pathology back into normal tissue using a mask, with gamma parameters drawn from a fixed mixture distribution. Across three private MRI datasets, the approach yields consistent gains in image-level sensitivity (e.g., increases from 0.566 to 0.736 on the WUS set) with minimal or acceptable changes to specificity, demonstrating improved detection of ischemic lesions under limited voxel-level annotations. The contribution provides a simple, easily integrable augmentation technique that mitigates intensity bias and enhances data efficiency for pathology segmentation tasks in MRI, with potential applicability to other diseases characterized by region-specific intensity differences.
Abstract
The identification and localisation of pathological tissues in medical images continues to command much attention among deep learning practitioners. When trained on abundant datasets, deep neural networks can match or exceed human performance. However, the scarcity of annotated data complicates the training of these models. Data augmentation techniques can compensate for a lack of training samples. However, many commonly used augmentation methods can fail to provide meaningful samples during model fitting. We present local gamma augmentation, a technique for introducing new instances of intensities in pathological tissues. We leverage local gamma augmentation to compensate for a bias in intensities corresponding to ischemic stroke lesions in human brain MRIs. On three datasets, we show how local gamma augmentation can improve the image-level sensitivity of a deep neural network tasked with ischemic lesion segmentation on magnetic resonance images.
