U-Net Using Stacked Dilated Convolutions for Medical Image Segmentation
Authors:
Shuhang Wang,
Szu-Yeu Hu,
Eugene Cheah,
Xiaohong Wang,
Jingchao Wang,
Lei Chen,
Masoud Baikpour,
Arinc Ozturk,
Qian Li,
Shinn-Huey Chou,
Constance D. Lehman,
Viksit Kumar,
Anthony Samir
Abstract:
This paper proposes a novel U-Net variant using stacked dilated convolutions for medical image segmentation (SDU-Net). SDU-Net adopts the architecture of vanilla U-Net with modifications in the encoder and decoder operations (an operation indicates all the processing for feature maps of the same resolution). Unlike vanilla U-Net which incorporates two standard convolutions in each encoder/decoder…
▽ More
This paper proposes a novel U-Net variant using stacked dilated convolutions for medical image segmentation (SDU-Net). SDU-Net adopts the architecture of vanilla U-Net with modifications in the encoder and decoder operations (an operation indicates all the processing for feature maps of the same resolution). Unlike vanilla U-Net which incorporates two standard convolutions in each encoder/decoder operation, SDU-Net uses one standard convolution followed by multiple dilated convolutions and concatenates all dilated convolution outputs as input to the next operation. Experiments showed that SDU-Net outperformed vanilla U-Net, attention U-Net (AttU-Net), and recurrent residual U-Net (R2U-Net) in all four tested segmentation tasks while using parameters around 40% of vanilla U-Net's, 17% of AttU-Net's, and 15% of R2U-Net's.
△ Less
Submitted 10 April, 2020; v1 submitted 7 April, 2020;
originally announced April 2020.
Expert identification of visual primitives used by CNNs during mammogram classification
Authors:
Jimmy Wu,
Diondra Peck,
Scott Hsieh,
Vandana Dialani,
Constance D. Lehman,
Bolei Zhou,
Vasilis Syrgkanis,
Lester Mackey,
Genevieve Patterson
Abstract:
This work interprets the internal representations of deep neural networks trained for classification of diseased tissue in 2D mammograms. We propose an expert-in-the-loop interpretation method to label the behavior of internal units in convolutional neural networks (CNNs). Expert radiologists identify that the visual patterns detected by the units are correlated with meaningful medical phenomena s…
▽ More
This work interprets the internal representations of deep neural networks trained for classification of diseased tissue in 2D mammograms. We propose an expert-in-the-loop interpretation method to label the behavior of internal units in convolutional neural networks (CNNs). Expert radiologists identify that the visual patterns detected by the units are correlated with meaningful medical phenomena such as mass tissue and calcificated vessels. We demonstrate that several trained CNN models are able to produce explanatory descriptions to support the final classification decisions. We view this as an important first step toward interpreting the internal representations of medical classification CNNs and explaining their predictions.
△ Less
Submitted 13 March, 2018;
originally announced March 2018.