PGKD-Net: Prior-guided and Knowledge Diffusive Network for Choroid Segmentation

Yaqi Wang,Zehua Yang,Xindi Liu,Zhi Li,Chengyu Wu,Yizhen Wang,Kai Jin,Dechao Chen,Gangyong Jia,Xiaodiao Chen,Juan Ye,Xingru Huang
DOI: https://doi.org/10.1016/j.artmed.2024.102837
IF: 7.011
2024-01-01
Artificial Intelligence in Medicine
Abstract:The thickness of the choroid is considered to be an important indicator of clinical diagnosis. Therefore, accurate choroid segmentation in retinal OCT images is crucial for monitoring various ophthalmic diseases. However, this is still challenging due to the blurry boundaries and interference from other lesions. To address these issues, we propose a novel prior-guided and knowledge diffusive network (PGKD-Net) to fully utilize retinal structural information to highlight choroidal region features and boost segmentation performance. Specifically, it is composed of two parts: a Prior-mask Guided Network (PG-Net) for coarse segmentation and a Knowledge Diffusive Network (KD-Net) for fine segmentation. In addition, we design two novel feature enhancement modules, Multi-Scale Context Aggregation (MSCA) and Multi-Level Feature Fusion (MLFF). The MSCA module captures the long-distance dependencies between features from different receptive fields and improves the model’s ability to learn global context. The MLFF module integrates the cascaded context knowledge learned from PG-Net to benefit fine-level segmentation. Comprehensive experiments are conducted to evaluate the performance of the proposed PGKD-Net. Experimental results show that our proposed method achieves superior segmentation accuracy over other state-of-the-art methods. Our code is made up publicly available at: https://github.com/yzh-hdu/choroid-segmentation.
What problem does this paper attempt to address?