• Laser & Optoelectronics Progress
  • Vol. 61, Issue 18, 1837005 (2024)
Wenyue Hao1, Huaiyu Cai1,*, Tingtao Zuo2, Zhongwei Jia3..., Yi Wang1 and Xiaodong Chen1|Show fewer author(s)
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
    DOI: 10.3788/LOP232774 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    θ||ϵθ(x0+σϵ)-x0||2
    xt=1-βtxt-1+βtϵ, ϵ~N(0,1)
    xt=αt¯x0+1-αt¯ϵ,  ϵ~N(0,1)
    xt-1=11-βtxt-βt1-αt¯ϵθ(xt,t)+βt1-αt-1¯1-αt¯ϵ,  ϵ~N(0,1)
    θ||ϵθ(αt¯x0+1-αt¯ϵ)-ϵ||2
    =MSE(ϵθ,ϵ)+γSSIM(ϵθ,ϵ)
    DDice =2×|RpredRtrue||Rpred|+|Rtrue|
    JJM =RpredRtrueRpredRtrue
    HHD =max{d(Cseg,Cgt),d(Cgt,Cseg)}
    PPAD =|Apred-Atrue|Atrue
    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