Deep Frequency Re-calibration U-Net for Medical Image Segmentation

Azad R, Bozorgpour A, Asadi-Aghbolaghi M, Merhof D, Escalera S (2021)


Publication Type: Conference contribution

Publication year: 2021

Publisher: Institute of Electrical and Electronics Engineers Inc.

Book Volume: 2021-October

Pages Range: 3267-3276

Conference Proceedings Title: Proceedings of the IEEE International Conference on Computer Vision

Event location: Virtual, Online, CAN

ISBN: 9781665401913

DOI: 10.1109/ICCVW54120.2021.00366

Abstract

The human visual cortex is biased towards shape components while CNNs produce texture biased features. This fact may explain why the performance of CNN significantly degrades with low-labeled input data scenarios. In this paper, we propose a frequency re-calibration U-Net (FRCU-Net) for medical image segmentation. Representing an object in terms of frequency may reduce the effect of texture bias, resulting in better generalization for a low data regime. To do so, we apply the Laplacian pyramid in the bottleneck layer of the U-shaped structure. The Laplacian pyramid represents the object proposal in different frequency domains, where the high frequencies are responsible for the texture information and lower frequencies might be related to the shape. Adaptively re-calibrating these frequency representations can produce a more discriminative representation for describing the object of interest. To this end, we first propose to use a channel-wise attention mechanism to capture the relationship between the channels of a set of feature maps in one layer of the frequency pyramid. Second, the extracted features of each level of the pyramid are then combined through a non-linear function based on their impact on the final segmentation output. The proposed FRCU-Net is evaluated on five datasets ISIC 2017, ISIC 2018, the PH2, lung segmentation, and SegPC 2021 challenge datasets and compared to existing alternatives, achieving state-of-the-art results.

Involved external institutions

How to cite

APA:

Azad, R., Bozorgpour, A., Asadi-Aghbolaghi, M., Merhof, D., & Escalera, S. (2021). Deep Frequency Re-calibration U-Net for Medical Image Segmentation. In Proceedings of the IEEE International Conference on Computer Vision (pp. 3267-3276). Virtual, Online, CAN: Institute of Electrical and Electronics Engineers Inc..

MLA:

Azad, Reza, et al. "Deep Frequency Re-calibration U-Net for Medical Image Segmentation." Proceedings of the 18th IEEE/CVF International Conference on Computer Vision Workshops, ICCVW 2021, Virtual, Online, CAN Institute of Electrical and Electronics Engineers Inc., 2021. 3267-3276.

BibTeX: Download