Elongated Physiological Structure Segmentation via Spatial and Scale Uncertainty-aware Network
Authors:
Yinglin Zhang,
Ruiling Xi,
Huazhu Fu,
Dave Towey,
RuiBin Bai,
Risa Higashita,
Jiang Liu
Abstract:
Robust and accurate segmentation for elongated physiological structures is challenging, especially in the ambiguous region, such as the corneal endothelium microscope image with uneven illumination or the fundus image with disease interference. In this paper, we present a spatial and scale uncertainty-aware network (SSU-Net) that fully uses both spatial and scale uncertainty to highlight ambiguous…
▽ More
Robust and accurate segmentation for elongated physiological structures is challenging, especially in the ambiguous region, such as the corneal endothelium microscope image with uneven illumination or the fundus image with disease interference. In this paper, we present a spatial and scale uncertainty-aware network (SSU-Net) that fully uses both spatial and scale uncertainty to highlight ambiguous regions and integrate hierarchical structure contexts. First, we estimate epistemic and aleatoric spatial uncertainty maps using Monte Carlo dropout to approximate Bayesian networks. Based on these spatial uncertainty maps, we propose the gated soft uncertainty-aware (GSUA) module to guide the model to focus on ambiguous regions. Second, we extract the uncertainty under different scales and propose the multi-scale uncertainty-aware (MSUA) fusion module to integrate structure contexts from hierarchical predictions, strengthening the final prediction. Finally, we visualize the uncertainty map of final prediction, providing interpretability for segmentation results. Experiment results show that the SSU-Net performs best on cornea endothelial cell and retinal vessel segmentation tasks. Moreover, compared with counterpart uncertainty-based methods, SSU-Net is more accurate and robust.
△ Less
Submitted 30 May, 2023;
originally announced May 2023.
Machine Learning for Cataract Classification and Grading on Ophthalmic Imaging Modalities: A Survey
Authors:
Xiaoqing Zhang,
Yan Hu,
Zunjie Xiao,
Jiansheng Fang,
Risa Higashita,
Jiang Liu
Abstract:
Cataracts are the leading cause of visual impairment and blindness globally. Over the years, researchers have achieved significant progress in develo** state-of-the-art machine learning techniques for automatic cataract classification and grading, aiming to prevent cataracts early and improve clinicians' diagnosis efficiency. This survey provides a comprehensive survey of recent advances in mach…
▽ More
Cataracts are the leading cause of visual impairment and blindness globally. Over the years, researchers have achieved significant progress in develo** state-of-the-art machine learning techniques for automatic cataract classification and grading, aiming to prevent cataracts early and improve clinicians' diagnosis efficiency. This survey provides a comprehensive survey of recent advances in machine learning techniques for cataract classification/grading based on ophthalmic images. We summarize existing literature from two research directions: conventional machine learning methods and deep learning methods. This survey also provides insights into existing works of both merits and limitations. In addition, we discuss several challenges of automatic cataract classification/grading based on machine learning techniques and present possible solutions to these challenges for future research.
△ Less
Submitted 1 April, 2022; v1 submitted 8 December, 2020;
originally announced December 2020.