CAMMA-public / surgical-imagen

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Surgical Text-to-Image Generation


Abstract: Acquiring surgical data for research and development is significantly hindered by high annotation costs and practical and ethical constraints. Utilizing synthetically generated images could offer a valuable alternative. In this work, we conduct an in-depth analysis on adapting text-toimage generative models for the surgical domain, leveraging the CholecT50 dataset, which provides surgical images annotated with surgical action triplets (instrument, verb, target). We investigate various language models and find T5 to offer more distinct features for differentiating surgical actions based on triplet-based textual inputs. Our analysis demonstrates strong alignment between long and tripletbased captions, supporting the use of triplet-based labels. We address the challenges in training text-to-image models on triplet-based captions without additional input signals by uncovering that triplet text embeddings are instrumentcentric in the latent space and then, by designing an instrument-based class balancing technique to counteract the imbalance and skewness in the surgical data, improving training convergence. Extending Imagen, a diffusion-based generative model, we develop Surgical Imagen to generate photorealistic and activity-aligned surgical images from triplet-based textual prompts. We evaluate our model using diverse metrics, including human expert surveys and automated methods like FID and CLIP scores. We assess the model performance on key aspects: quality, alignment, reasoning, knowledge, and robustness, demonstrating the effectiveness of our approach in providing a realistic alternative to real data collection.



đź“° arXiv Preprint


CItation


@article{nwoye2024surgical,
    title={Surgical Text-to-Image Generation},
    author={Nwoye, Chinedu Innocent and Bose, Rupak and Elgohary, Kareem and Arboit, Lorenzo and Carlino, Giorgio and Lavanchy, Jo{\"e}l L and Mascagni, Pietro and Padoy, Nicolas},
    journal={arXiv preprint arXiv:2407.09230},
    year={2024}
}

Coming soon !

About


Languages

Language:HTML 100.0%