Artificial Intelligence-Based Kidney Segmentation With Modified Cycle-Consistent Generative Adversarial Network and Appearance-Based Shape Prior

Document Type

Article

Source of Publication

IEEE Access

Publication Date

1-1-2024

Abstract

This study presents an innovative deep learning framework for kidney segmentation in magnetic resonance imaging (MRI) data. The framework integrates both kidney appearance and prior shape information using a residual cycle-consistent generative adversarial network (CycleGAN). An appearance-based shape prior model is developed, utilizing iso-circular contours generated from the kidney centroid and employing the fast marching level sets method for shape extraction. By utilizing the kidney centroid and matching cross-circular iso-circular contours' appearance, the proposed appearance-based shape prior model remains invariant to translation, rotation, and scaling, eliminating the need for alignment. Additionally, a novel weighted loss function, the H-Loss, is introduced to enhance segmentation performance and prevent overfitting. The proposed approach is tested on 34 blood-oxygen-level-dependent (BOLD) grafts from patients in our kidney transplant program, achieving an average dice score of 92%. These promising results validate the effectiveness of the approach, with optimized hyperparameters ensuring high segmentation quality.

ISSN

2169-3536

Volume

12

First Page

162536

Last Page

162548

Disciplines

Computer Sciences

Keywords

Appearance-based shape prior, CycleGAN, generative adversarial network, kidney segmentation

Scopus ID

85207904070

Indexed in Scopus

yes

Open Access

no

Share

COinS