Integration- and separation-aware adversarial model for cerebrovascular segmentation from TOF-MRA

Comput Methods Programs Biomed. 2023 May:233:107475. doi: 10.1016/j.cmpb.2023.107475. Epub 2023 Mar 11.

Abstract

Purpose: Cerebrovascular segmentation from time-of-flight magnetic resonance angiography (TOF-MRA) is important but challenging for the simulation and measurement of cerebrovascular diseases. Recently, deep learning has promoted the rapid development of cerebrovascular segmentation. However, model optimization relies on voxel or regional punishment and lacks global awareness and interpretation from the texture and edge. To overcome the limitations of the existing methods, we propose a new cerebrovascular segmentation method to obtain more refined structures.

Methods: In this paper, we propose a new adversarial model that achieves segmentation using segmentation model and filters the results using discriminator. Considering the sample imbalance in cerebrovascular imaging, we separated the TOF-MRA images and utilized high- and low-frequency images to enhance the texture and edge representation. The encoder weight sharing from the segmentation model not only saves the model parameters, but also strengthens the integration and separation correlation. Diversified discrimination enhances the robustness and regularization of the model.

Results: The adversarial model was tested using two cerebrovascular datasets. It scored 82.26% and 73.38%, respectively, ranking first on both datasets. The results show that our method not only outperforms the recent cerebrovascular segmentation model, but also surpasses the common adversarial models.

Conclusion: Our adversarial model focuses on improving the extraction ability of the model on texture and edge, thereby achieving awareness of the global cerebrovascular topology. Therefore, we obtained an accurate and robust cerebrovascular segmentation. This framework has potential applications in many imaging fields, particularly in the application of sample imbalance. Our code is available at the website https://github.com/MontaEllis/ISA-model.

Keywords: Adversarial model; Cerebrovascular segmentation; Deep learning; TOF-MRA.

MeSH terms

  • Algorithms*
  • Computer Simulation
  • Image Processing, Computer-Assisted / methods
  • Magnetic Resonance Angiography* / methods
  • Magnetic Resonance Imaging