Segmentation of head and neck tumours using modified U-net

Zhao, Baixiang and Soraghan, John and Di Caterina, Gaetano and Grose, Derek (2019) Segmentation of head and neck tumours using modified U-net. In: 27th European Signal Processing Conference, 2019-09-02 - 2019-09-06.

[thumbnail of Zhao-etal-EUSIPCO2019-Segmentation-of-head-and-neck-tumours-using-modified-U-net]
Preview
Text. Filename: Zhao_etal_EUSIPCO2019_Segmentation_of_head_and_neck_tumours_using_modified_U_net.pdf
Accepted Author Manuscript

Download (813kB)| Preview

Abstract

A new neural network for automatic head and neck cancer (HNC) segmentation from magnetic resonance imaging (MRI) is presented. The proposed neural network is based on U-net, which combines features from different resolutions to achieve end-to-end locating and segmentation of medical images. In this work, the dilated convolution is introduced into U-net, to obtain larger receptive field so that extract multi-scale features. Also, this network uses Dice loss to reduce the imbalance between classes. The proposed algorithm is trained and tested on real MRI data. The cross-validation results show that the new network outperformed the original Unet by 5% (Dice score) on head and neck tumour segmentation.