SASAN: Spectrum-Axial Spatial Approach Networks for Medical Image Segmentation

• Published in: IEEE transactions on medical imaging 2024-08, Vol.43 (8), p.3044-3056
• Main Authors: Huang, Xingru, Huang, Jian, Zhao, Kai, Zhang, Tianyun, Li, Zhi, Yue, Changpeng, Chen, Wenhao, Wang, Ruihao, Chen, Xuanbin, Zhang, Qianni, Fu, Ying, Wang, Yangyundou, Guo, Yihao
• Format: Article
• Language: English
• Subjects: Algorithms; Central Serous Chorioretinopathy - diagnostic imaging; Choroid - diagnostic imaging; choroid identification; Databases, Factual; Decoding; Feature extraction; Fourier Analysis; Fourier transform; Frequency-domain analysis; Fundus OCT segmentation; Humans; Image Interpretation, Computer-Assisted - methods; Image segmentation; Imaging, Three-Dimensional - methods; macular edema; Macular Edema - diagnostic imaging; Medical diagnostic imaging; Neural Networks, Computer; Noise; Three-dimensional displays; Tomography, Optical Coherence - methods
• ISSN: 0278-0062; 1558-254X; 1558-254X
• DOI: 10.1109/TMI.2024.3383466

Ophthalmic diseases such as central serous chorioretinopathy (CSC) significantly impair the vision of millions of people globally. Precise segmentation of choroid and macular edema is critical for diagnosing and treating these conditions. However, existing 3D medical image segmentation methods often fall short due to the heterogeneous nature and blurry features of these conditions, compounded by medical image clarity issues and noise interference arising from equipment and environmental limitations. To address these challenges, we propose the Spectrum Analysis Synergy Axial-Spatial Network (SASAN), an approach that innovatively integrates spectrum features using the Fast Fourier Transform (FFT). SASAN incorporates two key modules: the Frequency Integrated Neural Enhancer (FINE), which mitigates noise interference, and the Axial-Spatial Elementum Multiplier (ASEM), which enhances feature extraction. Additionally, we introduce the Self-Adaptive Multi-Aspect Loss ( \mathcal {L}_{\textit {SM}} ), which balances image regions, distribution, and boundaries, adaptively updating weights during training. We compiled and meticulously annotated the Choroid and Macular Edema OCT Mega Dataset (CMED-18k), currently the world's largest dataset of its kind. Comparative analysis against 13 baselines shows our method surpasses these benchmarks, achieving the highest Dice scores and lowest HD95 in the CMED and OIMHS datasets. Our code is publicly available at https://github.com/IMOP-lab/SASAN-Pytorch .