QGD-Net: A Lightweight Model Utilizing Pixels of Affinity in Feature Layer for Dermoscopic Lesion Segmentation
Jingchao Wang,Guoheng Huang,Guo Zhong,Xiaochen Yuan,Chi-Man Pun,Jie Deng
DOI: https://doi.org/10.1109/jbhi.2023.3320953
IF: 7.7
2023-01-01
IEEE Journal of Biomedical and Health Informatics
Abstract:Pixels with location affinity, which can be also called "pixels of affinity", have similar semantic information. As a result, utilizing them can contribute to improving the capability of the model. However, group convolution and dilated convolution remain some problems related to it. For group convolution, it does not utilize pixels of affinity between layers. For dilated convolution, after multiple convolutions with the same dilated rate, the pixels utilized within each layer do not possess location affinity with each other. To solve the problem of group convolution, our proposed quaternion group convolution uses the quaternion convolution, which promotes the communication between layers via Hamilton product, to promote utilizing pixels of affinity between channels. It partially compensates for the limited utilization of pixel affinity between layers. In quaternion group convolution, the feature layers are divided into 4 layers per group, ensuring the quaternion convolution can be performed. To solve the problem of dilated convolution, we propose the quaternion sawtooth wavelike dilated convolutions module (QS module). QS module utilizes quaternion convolution with sawtooth wave-like dilated rates to effectively leverage the pixels that share the location affinity both between and within layers. This allows for an expanded receptive field, ultimately enhancing the performance of the model. In particular, we perform our quaternion group convolution in QS module to design the quaternion group dilated neutral network (QGD-Net). Extensive experiments on Dermoscopic Lesion Segmentation based on ISIC 2016 and ISIC 2017 indicate that our method has significantly reduced the model parameters and highly promoted the precision of the model in Dermoscopic Lesion Segmentation. And our method also shows generalizability in retinal vessel segmentation.
computer science, interdisciplinary applications,mathematical & computational biology,medical informatics, information systems