Efficient binary cnn for medical image segmentation

Kaustav Brahma, Viksit Kumar, Anthony E. Samir, Anantha P. Chandrakasan, Yonina C. Eldar

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In this work, we propose accurate binary Depthwise Separable Convolutional Neural Networks (DSCNNs) for medical image segmentation. The networks are binarized by learning the distribution of weights and activations, and by using parameter-free skip connections in their encoder and decoder structure. We design full precision DSCNNs based on a symmetric encoder-decoder, feature pyramid network with an asymmetric decoder, and spatial pyramid pooling with atrous convolutions strategies for image segmentation. The DSCNNs have 14 X and 8 X fewer number of model parameters and operations, respectively, than standard segmentation networks. The trained full precision DSCNNs are used as baselines to achieve accurate binary DSCNNs. The networks are trained on two medical ultrasound datasets, a public fetal skull dataset and a privileged bladder dataset. The accuracy of the binary DSCNNs are within a 3% drop from the full precision networks on both the medical datasets.

Original languageEnglish
Title of host publication2021 IEEE 18th International Symposium on Biomedical Imaging, ISBI 2021
PublisherIEEE Computer Society
Pages817-821
Number of pages5
ISBN (Electronic)9781665412469
DOIs
StatePublished - 13 Apr 2021
Event18th IEEE International Symposium on Biomedical Imaging, ISBI 2021 - Nice, France
Duration: 13 Apr 202116 Apr 2021

Publication series

NameProceedings - International Symposium on Biomedical Imaging
Volume2021-April
ISSN (Print)1945-7928

Conference

Conference18th IEEE International Symposium on Biomedical Imaging, ISBI 2021
Country/TerritoryFrance
CityNice
Period13/04/2116/04/21

Keywords

  • Binary CNN
  • Depthwise separable convolution
  • MobileNet
  • Weight quantization

All Science Journal Classification (ASJC) codes

  • Biomedical Engineering
  • Radiology Nuclear Medicine and imaging

Fingerprint

Dive into the research topics of 'Efficient binary cnn for medical image segmentation'. Together they form a unique fingerprint.

Cite this