Laser & Optoelectronics Progress, Volume. 61, Issue 18, 1837005(2024)

Self-Supervised Pre-Training for Intravascular Ultrasound Image Segmentation Method Based on Diffusion Model

Wenyue Hao1, Huaiyu Cai1、*, Tingtao Zuo2, Zhongwei Jia3, Yi Wang1, and Xiaodong Chen1
Author Affiliations
  • 1Key Laboratory of Optoelectronics Information Technology, Ministry of Education, School of Precision Instrument and Opto-Electronics Engineering, Tianjin University, Tianjin 300072, China
  • 2Lepu Medical Technology (Beijing) Co., Ltd., Beijing 102200, China
  • 3Southwestern Lu Hospital, Liaocheng 252325, Shandong, China
  • show less

    To overcome the difficulty of obtaining large annotated datasets, a proxy task based on a diffusion model was introduced, allowing for self-supervised learning of a priori knowledge from unlabeled datasets, followed by fine-tuning on a small labeled dataset. Inspired by the diffusion model, different levels of noise are weighted with the original images as inputs to the model. By training the model to predict the input noise, a more robust learning of the representation of intravascular ultrasound (IVUS) images at the pixel level was achieved. Additionally, the combined loss function of mean square error (MSE) and structural similarity index (SSIM) was introduced to improve the performance of the model. The experimental results of this method on 20% dataset demonstrate that the Jaccard metric coefficients of the lumen and meida are increased by 0.044 and 0.101, respectively, compared with result of random initialization, and the Hausdorff distance coefficients are improved by 0.216 and 0.107, respectively, compared with result of random initialization, which is similar to the result of using 100% dataset for training. This framework applies to any structural image segmentation model and significantly reduces the reliance on ground truth while ensuring segmentation effectiveness.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Wenyue Hao, Huaiyu Cai, Tingtao Zuo, Zhongwei Jia, Yi Wang, Xiaodong Chen. Self-Supervised Pre-Training for Intravascular Ultrasound Image Segmentation Method Based on Diffusion Model[J]. Laser & Optoelectronics Progress, 2024, 61(18): 1837005

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Digital Image Processing

    Received: Dec. 28, 2023

    Accepted: Feb. 5, 2024

    Published Online: Sep. 14, 2024

    The Author Email: Huaiyu Cai (hycai@tju.edu.cn)

    DOI:10.3788/LOP232774

    CSTR:32186.14.LOP232774

    Topics