Skip to main content
Video s3
    Details
    Presenter(s)
    Liming Guo Headshot
    Display Name
    Liming Guo
    Affiliation
    Affiliation
    Shanghai Jiao Tong University
    Country
    Author(s)
    Display Name
    Liming Guo
    Affiliation
    Affiliation
    Shanghai Jiao Tong University
    Display Name
    Wen Fei
    Affiliation
    Affiliation
    Shanghai Jiao Tong University
    Display Name
    Wenrui Dai
    Affiliation
    Affiliation
    Shanghai Jiao Tong University
    Display Name
    Chenglin Li
    Affiliation
    Affiliation
    Shanghai Jiao Tong University
    Display Name
    Junni Zou
    Affiliation
    Affiliation
    Shanghai Jiao Tong University
    Display Name
    Hongkai Xiong
    Affiliation
    Affiliation
    Shanghai Jiao Tong University
    Abstract

    In this paper, we proposed a mixed-precision quantization of U-Net for medical image segmentation. There are two main contributions in our work: A split convolution for a fine-grained bitwidth allocation that addresses the difference between skip feature maps and up-sampled feature maps. Furthermore, we introduce supervision on skip connection for quantization aware training to compensate gradients in back propagation and enhance the skip features in finetuning. Extensive experiments are conducted to demonstrate the effectiveness of our proposed method.

    Slides
    • Mixed-Precision Quantization of U-Net for Medical Image Segmentation (application/pdf)