Holistic multitask regression network for multiapplication shape regression segmentation

A holistic multitask regression approach was implemented to tackle the limitations of clinical image analysis. Standard practice requires identifying multiple anatomic structures in multiple planes from multiple anatomic regions using multiple modalities. The proposed novel holistic multitask regression network (HMR-Net) formulates organ segmentation as a multitask learning problem. Multitask learning leverages the strength of joint task problem solving from capturing task correlations. HMR-Net performs multitask regression by estimating an organ's class, regional location, and precise contour coordinates. The estimation of each coordinate point also corresponds to another regression task. HMR-Net leverages hierarchical multiscale and fused organ features to handle nonlinear relationships between image appearance and distinct organ properties. Simultaneously, holistic shape information is captured by encoding coordinate correlations. The multitask pipeline enables the capturing of holistic organ information (e.g. class, location, shape) to perform shape regression for medical image segmentation. HMR-Net was validated on eight representative datasets obtained from a total of 222 subjects. A mean average precision and dice score reaching up to 0.81 and 0.93, respectively, was achieved on the representative multiapplication database. The generalized model demonstrates comparable or superior performance compared to state-of-the-art algorithms. The high-performance accuracy demonstrates our model as an effective general framework to perform organ shape regression in multiple applications. This method was proven to provide high-contrast sensitivity to delineate even the smallest and oddly shaped organs. HMR-Net's flexible framework holds great potential in providing a fully automatic preliminary analysis for multiple types of medical images.

[1]  Shu Liu,et al.  Path Aggregation Network for Instance Segmentation , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[2]  John J. Soraghan,et al.  Segmentation of Head and Neck Tumours Using Modified U-net , 2019, 2019 27th European Signal Processing Conference (EUSIPCO).

[3]  Kaiming He,et al.  Feature Pyramid Networks for Object Detection , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[4]  S. Li,et al.  Regression Segmentation for $M^{3}$ Spinal Images , 2015, IEEE Transactions on Medical Imaging.

[5]  Shuo Li,et al.  Spine‐GAN: Semantic segmentation of multiple spinal structures , 2018, Medical Image Anal..

[6]  Bill Triggs,et al.  Histograms of oriented gradients for human detection , 2005, 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'05).

[7]  Cristian Lorenz,et al.  Spine Segmentation Using Articulated Shape Models , 2008, MICCAI.

[8]  Zhang Yi,et al.  Multi-task Learning for the Segmentation of Thoracic Organs at Risk in CT images , 2019, SegTHOR@ISBI.

[9]  H. Redkey,et al.  A new approach. , 1967, Rehabilitation record.

[10]  Shuo Li,et al.  Glioma Segmentation With a Unified Algorithm in Multimodal MRI Images , 2018, IEEE Access.

[11]  Punsiri Boonyakiat,et al.  Segmentation of optic nerve head images , 2017, 2017 14th International Joint Conference on Computer Science and Software Engineering (JCSSE).

[12]  Vipin Chaudhary,et al.  A new approach to automatic disc localization in clinical lumbar MRI: Combining machine learning with heuristics , 2012, 2012 9th IEEE International Symposium on Biomedical Imaging (ISBI).

[13]  Håkon Olav Leira,et al.  Semantic segmentation and detection of mediastinal lymph nodes and anatomical structures in CT data for lung cancer staging , 2019, International Journal of Computer Assisted Radiology and Surgery.

[14]  Bruce I. Reiner,et al.  Evolution of the Digital Revolution: A Radiologist Perspective , 2003, Journal of Digital Imaging.

[15]  Jun Chen,et al.  Correlated Regression Feature Learning for Automated Right Ventricle Segmentation , 2018, IEEE Journal of Translational Engineering in Health and Medicine.

[16]  S. Butler Generalizing some results to the normalized Laplacian , 2006 .

[17]  G. Sharp,et al.  Vision 20/20: perspectives on automated image segmentation for radiotherapy. , 2014, Medical physics.

[18]  Pheng Ann Heng,et al.  Unpaired Multi-Modal Segmentation via Knowledge Distillation , 2020, IEEE Transactions on Medical Imaging.

[19]  Jean-François Daisne,et al.  Atlas-based automatic segmentation of head and neck organs at risk and nodal target volumes: a clinical validation , 2013, Radiation oncology.

[20]  Ronald M. Summers,et al.  A multi-center milestone study of clinical vertebral CT segmentation , 2016, Comput. Medical Imaging Graph..

[21]  Kensaku Mori,et al.  Spinal vertebrae segmentation and localization by transfer learning , 2019, Medical Imaging.

[22]  Yanping Zhang,et al.  Multitask Learning for Estimating Multitype Cardiac Indices in MRI and CT Based on Adversarial Reverse Mapping , 2020, IEEE Transactions on Neural Networks and Learning Systems.

[23]  Mikhail Belkin,et al.  Laplacian Eigenmaps for Dimensionality Reduction and Data Representation , 2003, Neural Computation.

[24]  Qianjin Feng,et al.  Direct automated quantitative measurement of spine by cascade amplifier regression network with manifold regularization , 2019, Medical Image Anal..

[25]  Ayse Betül Oktay,et al.  Simultaneous Localization of Lumbar Vertebrae and Intervertebral Discs With SVM-Based MRF , 2013, IEEE Transactions on Biomedical Engineering.

[26]  Bulat Ibragimov,et al.  Segmentation of organs‐at‐risks in head and neck CT images using convolutional neural networks , 2017, Medical physics.

[27]  Shang-Hong Lai,et al.  Learning-Based Vertebra Detection and Iterative Normalized-Cut Segmentation for Spinal MRI , 2009, IEEE Transactions on Medical Imaging.

[28]  Jun Chen,et al.  Deep Regression Segmentation for Cardiac Bi-Ventricle MR Images , 2018, IEEE Access.

[29]  Wei Shen,et al.  Semi-Supervised 3D Abdominal Multi-Organ Segmentation Via Deep Multi-Planar Co-Training , 2018, 2019 IEEE Winter Conference on Applications of Computer Vision (WACV).

[30]  Yang Lei,et al.  CT Prostate Segmentation Based on Synthetic MRI-aided Deep Attention Fully Convolution Network. , 2019, Medical physics.

[31]  Prof. Dr. Enrico Marani,et al.  Head and neck , 2022, Springer International Publishing.

[32]  Shuo Li,et al.  Graph Cuts with Invariant Object-Interaction Priors: Application to Intervertebral Disc Segmentation , 2011, IPMI.

[33]  Na Yang,et al.  Lung Nodule Detection With Deep Learning in 3D Thoracic MR Images , 2019, IEEE Access.

[34]  Yang Lei,et al.  Male pelvic multi-organ segmentation aided by CBCT-based synthetic MRI , 2019, Physics in medicine and biology.

[35]  Ronald M. Summers,et al.  Detection of Vertebral Body Fractures Based on Cortical Shell Unwrapping , 2012, MICCAI.

[36]  Vladimir Pekar,et al.  Auto-segmentation of normal and target structures in head and neck CT images: a feature-driven model-based approach. , 2011, Medical physics.

[37]  Gijs van Tulder,et al.  Learning Cross-Modality Representations From Multi-Modal Images , 2019, IEEE Transactions on Medical Imaging.

[38]  Qinghua He,et al.  A method of localization and segmentation of intervertebral discs in spine MRI based on Gabor filter bank , 2016, Biomedical engineering online.

[39]  Hui Yu,et al.  Segmentation of Lung Nodule in CT Images Based on Mask R-CNN , 2018, 2018 9th International Conference on Awareness Science and Technology (iCAST).

[40]  Yang Lei,et al.  Synthetic MRI-aided multi-organ segmentation on male pelvic CT using cycle consistent deep attention network. , 2019, Radiotherapy and oncology : journal of the European Society for Therapeutic Radiology and Oncology.

[41]  Jason Yosinski,et al.  An Intriguing Failing of Convolutional Neural Networks and the CoordConv Solution , 2018, NeurIPS.

[42]  H. Labelle,et al.  Spine Segmentation in Medical Images Using Manifold Embeddings and Higher-Order MRFs , 2013, IEEE Transactions on Medical Imaging.

[43]  Jianjiang Feng,et al.  CFUN: Combining Faster R-CNN and U-net Network for Efficient Whole Heart Segmentation , 2018, ArXiv.

[44]  Seungkyu Lee,et al.  Vertebrae localization in CT using both local and global symmetry features , 2017, Comput. Medical Imaging Graph..

[45]  Guy Engelhard,et al.  Lung Nodules Detection and Segmentation Using 3D Mask-RCNN , 2019, ArXiv.

[46]  Shuo Li,et al.  Multi-Modality Vertebra Recognition in Arbitrary Views Using 3D Deformable Hierarchical Model , 2015, IEEE Transactions on Medical Imaging.

[47]  Alina A. von Davier,et al.  Cross-Validation , 2014 .

[48]  Blake Hannaford,et al.  Evaluation of segmentation methods on head and neck CT: Auto‐segmentation challenge 2015 , 2017, Medical physics.

[49]  Nikos Paragios,et al.  Nonlinear Embedding towards Articulated Spine Shape Inference Using Higher-Order MRFs , 2010, MICCAI.

[50]  Shuo Li,et al.  Multi-modal vertebrae recognition using Transformed Deep Convolution Network , 2016, Comput. Medical Imaging Graph..

[51]  Yilong Yin,et al.  Multi-task Shape Regression for Medical Image Segmentation , 2016, MICCAI.

[52]  Vincent Lepetit,et al.  DAISY: An Efficient Dense Descriptor Applied to Wide-Baseline Stereo , 2010, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[53]  Wenwu Zhu,et al.  Structural Deep Network Embedding , 2016, KDD.

[54]  X. Jiang,et al.  Automated delineation of organs-at-risk in head and neck CT images using multi-output support vector regression , 2018, Medical Imaging.

[55]  Wenqing Sun,et al.  Fast and fully-automated detection and segmentation of pulmonary nodules in thoracic CT scans using deep convolutional neural networks , 2019, Comput. Medical Imaging Graph..

[56]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[57]  Marleen de Bruijne,et al.  Learning Cross-Modality Representations From Multi-Modal Images , 2019, IEEE Trans. Medical Imaging.

[58]  Cristian Lorenz,et al.  Automated model-based vertebra detection, identification, and segmentation in CT images , 2009, Medical Image Anal..

[59]  Ognjen Arandjelovic,et al.  Automatic vertebrae localization from CT scans using volumetric descriptors , 2017, 2017 39th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[60]  V. Pekar,et al.  Head and Neck Auto-segmentation Challenge , 2009, The MIDAS Journal.

[61]  Fan Chung,et al.  Spectral Graph Theory , 1996 .

[62]  Xiantong Zhen,et al.  Multitarget Sparse Latent Regression , 2018, IEEE Transactions on Neural Networks and Learning Systems.

[63]  Xiangzhi Bai,et al.  Efficient Multiple Organ Localization in CT Image Using 3D Region Proposal Network , 2019, IEEE Transactions on Medical Imaging.

[64]  U. Feige,et al.  Spectral Graph Theory , 2015 .