Mixup Augmentation for Kidney and Kidney Tumor Segmentation

Matej Gazda, Peter Bugata, Jakub Gazda, David Hubacek, David Jozef Hresko, Peter Drotar

Keywords: nnU-Net, mixup, augmentation, kidney, CT, segmentation

Abstract: Abdominal computed tomography is frequently used to noninvasively map local conditions and to detect any benign or malign masses. However, ill-defined borders of malign objects, fuzzy texture, and time pressure in fact, make accurate segmentation in clinical settings a challenging task. In this paper, we propose a two-stage deep learning architecture for kidney and kidney masses segmentation, denoted as convolutional computer tomography network (CCTNet). The first stage locates volume bounding box containing both kidneys. The second stage performs the segmentation of kidney, kidney tumors and cysts. In the first stage, we use a pre-trained 3D low resolution nnU-Net. In the second stage, we employ a mixup augmentation to improve segmentation performance of the second 3D full resolution nnU-Net. The obtained results indicate that CCTNet can provide improved segmentation of kidney, kidney tumor and cyst.