Training-Free Condition Video Diffusion Models for single frame Spatial-Semantic Echocardiogram Synthesis

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Nguyen, Van Phi, Ha, Tri Nhan Luong, Pham, Huy Hieu, Tran, Quoc Long
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914939517534208
author Nguyen, Van Phi
Ha, Tri Nhan Luong
Pham, Huy Hieu
Tran, Quoc Long
author_facet Nguyen, Van Phi
Ha, Tri Nhan Luong
Pham, Huy Hieu
Tran, Quoc Long
contents Conditional video diffusion models (CDM) have shown promising results for video synthesis, potentially enabling the generation of realistic echocardiograms to address the problem of data scarcity. However, current CDMs require a paired segmentation map and echocardiogram dataset. We present a new method called Free-Echo for generating realistic echocardiograms from a single end-diastolic segmentation map without additional training data. Our method is based on the 3D-Unet with Temporal Attention Layers model and is conditioned on the segmentation map using a training-free conditioning method based on SDEdit. We evaluate our model on two public echocardiogram datasets, CAMUS and EchoNet-Dynamic. We show that our model can generate plausible echocardiograms that are spatially aligned with the input segmentation map, achieving performance comparable to training-based CDMs. Our work opens up new possibilities for generating echocardiograms from a single segmentation map, which can be used for data augmentation, domain adaptation, and other applications in medical imaging. Our code is available at \url{https://github.com/gungui98/echo-free}
format Preprint
id arxiv_https___arxiv_org_abs_2408_03035
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Training-Free Condition Video Diffusion Models for single frame Spatial-Semantic Echocardiogram Synthesis
Nguyen, Van Phi
Ha, Tri Nhan Luong
Pham, Huy Hieu
Tran, Quoc Long
Image and Video Processing
Computer Vision and Pattern Recognition
Conditional video diffusion models (CDM) have shown promising results for video synthesis, potentially enabling the generation of realistic echocardiograms to address the problem of data scarcity. However, current CDMs require a paired segmentation map and echocardiogram dataset. We present a new method called Free-Echo for generating realistic echocardiograms from a single end-diastolic segmentation map without additional training data. Our method is based on the 3D-Unet with Temporal Attention Layers model and is conditioned on the segmentation map using a training-free conditioning method based on SDEdit. We evaluate our model on two public echocardiogram datasets, CAMUS and EchoNet-Dynamic. We show that our model can generate plausible echocardiograms that are spatially aligned with the input segmentation map, achieving performance comparable to training-based CDMs. Our work opens up new possibilities for generating echocardiograms from a single segmentation map, which can be used for data augmentation, domain adaptation, and other applications in medical imaging. Our code is available at \url{https://github.com/gungui98/echo-free}
title Training-Free Condition Video Diffusion Models for single frame Spatial-Semantic Echocardiogram Synthesis
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.03035