Semantic Segmentation of Optical Coherence Tomography Images Based on TUnet+ Modeling
Main Article Content
Abstract
Accurate segmentation of Optical Coherence Tomography images is critical for assisting clinicians in identifying lesion regions and evaluating disease progression in retinal and cardiovascular disorders. As OCT is an optical electromagnetic-wave imaging modality, segmentation performance is closely related to the interpretation of wave scattering, tissue-layer boundaries, and propagation-induced image features. Existing OCT segmentation models are often limited by high computational complexity and insufficient accuracy, which restrict efficient and reliable clinical application. To address these challenges, this study proposes a lightweight TUnet+ network incorporating an innovatively designed PSE_C2fCIB module. Extensive experiments were conducted on the publicly available GOALS2022 dataset, including 200 deidentified OCT images augmented for training and validation. The results show that TUnet+ achieves state-of-the-art performance, with an accuracy of 99.3%, an F1-score of 95.65%, and a mean Intersection over Union of 91.87%. Through multi-scale feature fusion, channel attention, and efficient contextual modeling, the proposed framework improves the recognition of fine structural boundaries in OCT images. The method provides a robust technical foundation for automated diagnostic support and quantitative disease assessment, and it also demonstrates the value of electromagnetic-wave imaging analysis in biomedical engineering.
Downloads
Article Details
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
References
X. Xu, L. Yu, and Z. Chen, “Effect of erythrocyte aggregation on hematocrit measurement using spectral-domain optical coherence tomography,” IEEE Transactions on Biomedical Engineering, vol. 55, no. 12, pp. 2753– 2758, 2008, doi: 10.1109/TBME.2008.2002134.
W. Xu, D. L. Mathine, and J. K. Barton, “Analog cmos design for optical coherencetomography signal detection and processing,” IEEE Transactions on Biomedical Engineering, vol. 55, no. 2, pp. 485–489, 2008, doi: 10.1109/TBME.2007.905402.
W. Jung, J. Kim, M. Jeon, E. J. Chaney, C. N. Stewart, and S. A. Boppart, “Hand-held optical coherence tomography scanner for primary care diagnostics,” IEEE Transactions on Biomedical Engineering, vol. 58, no. 3, pp. 741–744, 2011, doi: 10.1109/TBME.2010.2096816.
A. T. Zavareh and S. Hoyos, “Kalman-based real-time functional decomposition for the spectral calibration in swept source optical coherence tomography,” IEEE Transactions on Biomedical Circuits and Systems, vol. 14, no. 2, pp. 257–273, 2020, doi: 10.1109/TBCAS.2019.2953212.
W. Jung, J. Kim, M. Jeon, E. J. Chaney, C. N. Stewart, and S. A. Boppart, “Hand-held optical coherence tomography scanner for primary care diagnostics,” IEEE Transactions on Biomedical Engineering, vol. 58, no. 3, pp. 741–744, 2011, doi: 10.1109/TBME.2010.2096816.
C. S. Premachandran, A. Khairyanto, K. C. W. Sheng, J. Singh, J. Teo, X. Ying-shun, C. Nanguang, C. Sheppard, and M. Olivo, “Design, fabrication, and assembly of an optical biosensor probe package for OCT (optical coherence tomography) application,” IEEE Transactions on Advanced Packaging, vol. 32, no. 2, pp. 417–422, 2009, doi: 10.1109/TADVP.2009.2013658.
N. A. Patel, J. Zoeller, D. L. Stamper, J. G. Fujimoto, and M. E. Brezinski, “Monitoring osteoarthritis in the rat model using optical coherence tomography,” IEEE Transactions on Medical Imaging, vol. 24, no. 2, pp. 155–159, 2005, doi: 10.1109/TMI.2004.839360.
S. Lu, C. Y.-l. Cheung, J. Liu, J. H. Lim, C. K.-s. Leung, and T. Y. Wong, “Automated layer segmentation of optical coherence tomography images,” IEEE Transactions on Biomedical Engineering, vol. 57, no. 10, pp. 2605– 2608, 2010, doi: 10.1109/TBME.2010.2055057.
B. Hassan, S. Qin, T. Hassan, R. Ahmed, and N. Werghi, “Joint segmentation and quantification of chorioretinal biomarkers in optical coherence tomography scans: A deep learning approach,” IEEE Transactions on Instrumentation and Measurement, vol. 70, pp. 1–17, 2021, doi: 10.1109/TIM.2021.3077988.
G. Litjens, T. Kooi, B. E. Bejnordi, A. A. A. Setio, F. Ciompi, M. Ghafoorian, J. A. W. M. van der Laak, B. van Ginneken, and C. I. Snchez, “A survey on deep learning in medical image analysis,” Medical Image Analysis, vol. 42, pp. 60–88, 2017, doi: 10.1016/j.media.2017.07.005.
O. Oktay, J. Schlemper, L. L. Folgoc, M. Lee, M. Heinrich, K. Misawa, K. Mori, S. McDonagh, N. Y. Hammerla, B. Kainz, B. Glocker, and D. Rueckert, “Attention U-Net: Learning Where to Look for the Pancreas,” 2018. [Online]. Available: https://arxiv.org/abs/1804.03999
L. Ngo, J. Cha, and J.-H. Han, “Deep neural network regression for automated retinal layer segmentation in optical coherence tomography images,” IEEE Transactions on Image Processing, vol. 29, pp. 303–312, 2020, doi: 10.1109/TIP.2019.2931461.
Y. Tan, W.-D. Shen, M.-Y. Wu, G.-N. Liu, S.-X. Zhao, Y. Chen, K.-F. Yang, and Y.-J. Li, “Retinal layer segmentation in OCT images with boundary regression and feature polarization,” IEEE Transactions on Medical Imaging, vol. 43, no. 2, pp. 686–700, 2024, doi: 10.1109/TMI.2023.3317072.
J. Chen, Y. Lu, Q. Yu, et al., “Transunet: Transformers make strong encoders for medical image segmentation,” arXiv preprint arXiv:2102.04306, 2021.
H. Cao, Y. Wang, J. Chen, et al., “Swin-unet: Unet-like pure transformer for medical image segmentation,” in European Conference on Computer Vision. Cham, Switzerland: Springer Nature Switzerland, 2022, pp. 205– 218.
Z. Liu, Y. Lin, Y. Cao, et al., “Swin transformer: Hierarchical vision transformer using shifted windows,” in Proceedings of the IEEE/CVF International Conference on Computer Vision, 2021, pp. 10012–10022.