Tribranchu-Net: A Size-Sensitive Network for Orbital Tumor Segmentation.

Yuchen He,Wenyu Wang,Zhiming Cheng,Wei Qi,Zhigang Duan,Kai Jin,Juan Ye,Shuai Wang
DOI: https://doi.org/10.1109/isbi53787.2023.10230640
2023-01-01
Abstract:The shape and location of the orbital tumor are precious for tumor evaluation, diagnosis, and treatment, so it is essential to accurately segment the orbital tumor to assist clinicians in making reasonable decisions. Nowadays, encoder-decoder-based convolutional neural networks have been widely used in medical image segmentation tasks. However, the commonly used single tensor flow architecture cannot achieve satisfactory performance for segmentation with significant size variations. In this paper, we propose a size-sensitive deep network named TriBranchU-Net, which consists of a Small Branch with fewer down-sampling layers to enhance the feature learning of small regions and a Large Branch with atrous convolutional residual connections to enhance the feature learning of large regions. Then a multi-branch fusion module is designed to fuse the learned knowledge from different branches for the final segmentation. We built a large dataset for evaluation, including 602 CT images from 64 patients with orbital tumors. The experimental results show that our method can achieve superior performance in different sizes. Moreover, additional experimental results on the CVC-ClinicDB dataset further demonstrate that our method can better handle the size variation in segmentation.
What problem does this paper attempt to address?