Spectral U-Net: Enhancing Medical Image Segmentation via Spectral Decomposition
This addresses medical image segmentation for healthcare applications, representing an incremental improvement over existing U-Net architectures.
The paper tackles medical image segmentation by introducing Spectral U-Net, which uses spectral decomposition via Dual Tree Complex Wavelet Transform to reduce information loss during down-sampling and improve detail reconstruction during up-sampling, achieving superior performance on Retina Fluid, Brain Tumor, and Liver Tumor datasets compared to nnU-Net.
This paper introduces Spectral U-Net, a novel deep learning network based on spectral decomposition, by exploiting Dual Tree Complex Wavelet Transform (DTCWT) for down-sampling and inverse Dual Tree Complex Wavelet Transform (iDTCWT) for up-sampling. We devise the corresponding Wave-Block and iWave-Block, integrated into the U-Net architecture, aiming at mitigating information loss during down-sampling and enhancing detail reconstruction during up-sampling. In the encoder, we first decompose the feature map into high and low-frequency components using DTCWT, enabling down-sampling while mitigating information loss. In the decoder, we utilize iDTCWT to reconstruct higher-resolution feature maps from down-sampled features. Evaluations on the Retina Fluid, Brain Tumor, and Liver Tumor segmentation datasets with the nnU-Net framework demonstrate the superiority of the proposed Spectral U-Net.