CycleMix: A Holistic Strategy for Medical Image Segmentation from Scribble Supervision
This addresses the high cost of full annotations in medical imaging by enabling effective segmentation from weaker scribble labels, though it is incremental as it builds on existing scribble learning and mixup techniques.
The paper tackles the problem of medical image segmentation with limited scribble supervision by proposing CycleMix, a framework combining mix augmentation and cycle consistency, which achieved comparable or better accuracy than fully-supervised methods on ACDC and MSCMRseg datasets.
Curating a large set of fully annotated training data can be costly, especially for the tasks of medical image segmentation. Scribble, a weaker form of annotation, is more obtainable in practice, but training segmentation models from limited supervision of scribbles is still challenging. To address the difficulties, we propose a new framework for scribble learning-based medical image segmentation, which is composed of mix augmentation and cycle consistency and thus is referred to as CycleMix. For augmentation of supervision, CycleMix adopts the mixup strategy with a dedicated design of random occlusion, to perform increments and decrements of scribbles. For regularization of supervision, CycleMix intensifies the training objective with consistency losses to penalize inconsistent segmentation, which results in significant improvement of segmentation performance. Results on two open datasets, i.e., ACDC and MSCMRseg, showed that the proposed method achieved exhilarating performance, demonstrating comparable or even better accuracy than the fully-supervised methods. The code and expert-made scribble annotations for MSCMRseg are publicly available at https://github.com/BWGZK/CycleMix.