DIM-UNet: Boosting medical image segmentation via diffusion models and information bottleneck theory mixed with MLP

Guangju Li, Yuanjie Zheng, Jia Cui, Wei Gai, Meng Qi

Abstract: Highlights•We propose a Diffusion-MLP module that can denoise feature maps while capturing global features.•We propose an IB-MLP module that compresses irrelevant and retains only relevant features.•We studied the impact of various parameters in Diffusion-MLP and IB-MLP on model performance.