KiU-Net

KiU-Net implements an overcomplete convolutional architecture for biomedical image and volumetric segmentation, improving detection of small structures and boundary precision by constraining receptive field expansion.


Key Features:

  • Overcomplete convolutional architecture: Projects input images into a higher-dimensional space to constrain receptive field expansion in deep layers.
  • Dual-branch design: Integrates a Kite-Net branch for low-level detail extraction with a U-Net branch for high-level feature learning.
  • Kite-Net branch: Captures fine details and accurate edges by maintaining a smaller receptive field through overcomplete convolutional operations.
  • U-Net branch: Learns high-level abstract features to complement low-level detail extraction.
  • KiU-Net 3D: Extends the architecture to 3D volumetric segmentation for volumetric medical imaging data.
  • Computational efficiency: Employs a design with fewer parameters and faster convergence during training compared to traditional encoder–decoder architectures.
  • Architectural enhancements: Supports incorporation of residual blocks and dense blocks to improve segmentation performance.
  • Empirical evaluation: Evaluated on five datasets with reported improvements in accuracy and efficiency relative to traditional methods.

Scientific Applications:

  • Biomedical image segmentation: Precise segmentation of small anatomical structures and boundary regions in medical images.
  • Volumetric segmentation: 3D segmentation of volumetric medical imaging modalities.
  • Medical research and diagnostics: Generation of accurate segmentations to support analysis in medical research and diagnostic workflows.

Methodology:

Overcomplete convolutional operations project inputs into a higher-dimensional space to constrain receptive field expansion; a dual-branch architecture combines an overcomplete Kite-Net branch for small-structure and edge capture with a U-Net branch for high-level feature learning; the architecture is extended to KiU-Net 3D for volumetric data and can incorporate residual blocks and dense blocks; the model was evaluated on five datasets.

Topics

Details

License:
MIT
Cost:
Free of charge
Tool Type:
command-line tool
Operating Systems:
Mac, Linux, Windows
Programming Languages:
Python
Added:
4/26/2022
Last Updated:
4/26/2022

Operations

Publications

Valanarasu JMJ, Sindagi VA, Hacihaliloglu I, Patel VM. KiU-Net: Overcomplete Convolutional Architectures for Biomedical Image and Volumetric Segmentation. IEEE Transactions on Medical Imaging. 2022;41(4):965-976. doi:10.1109/tmi.2021.3130469. PMID:34813472.

PMID: 34813472
Funding: - NSF: 1910141

Links