Search papers, labs, and topics across Lattice.
This paper introduces OTLesMix, a novel image synthesis method that utilizes Wasserstein barycenter and optimal transport to generate diverse and realistic synthetic lesions for medical imaging. The method addresses the limitations of existing data augmentation techniques, which often produce limited variability in lesion shapes and locations. Evaluations on three brain lesion segmentation tasks demonstrate that OTLesMix improves the Dice score by 2.9 to 6.6 points compared to models trained without synthetic data, outperforming state-of-the-art mixing strategies.
OTLesMix significantly enhances synthetic lesion diversity, boosting segmentation performance by up to 6.6 points on critical medical imaging tasks.
The development of deep learning over the past decade has revolutionized medical imaging segmentation, allowing the extraction of precise descriptors from large volumes to characterize pathologies. Data augmentation is a technique widely regarded as a way to improve model training. It includes simple transformations like spatial operations or intensity modifications, but also more advanced synthesis techniques. Their goal is to generate new realistic samples from an existing dataset to diversify the images used during training. Among them, several propose different mixing strategies to combine real samples. However, one of their major shortcomings is to yield limited variability in terms of generated lesion shapes and locations. In this work, we introduce a novel image synthesis method, called OTLesMix, that leverages Wasserstein barycenter and optimal transport plan to generate realistic and diverse samples. We evaluated our method on three brain lesion segmentation tasks, on which it improves the Dice score compared to a model trained without synthetic data by 2.9 to 6.6 points, and outperforms state-of-the-art mix-based methods.
