Automated Machine Learning Algorithm for Kidney, Kidney Tumor, Kidney Cyst Segmentation in Computed Tomography Scans

In this paper, we have described an automated algorithm for accurate segmentation of kidney, kidney tumors, and kidney cysts from CT scans. The Dataset for this problem was made available online as part of KiTS21 Challenge. Our approach was placed 13th in the official leaderboard of the competition. Our model uses a 2 stage Residual Unet architecture. The first network is designed to predict (Kidney + Tumor + Cyst) regions. The second network predicts segmented tumor and cyst regions from the output of the first network. The paper contains implementation details along with results on the official test and internal set.