Non-adversarial Learning: Vector-Quantized Common Latent Space for Multi-sequence MRI

Luyi Han, Tao Tan*, Tianyu Zhang, Xin Wang, Yuan Gao, Chunyao Lu, Xinglong Liang, Haoran Dou, Yunzhi Huang, Ritse Mann, MG Linguraru, Q Dou, A Feragen, S Giannarou, B Glocker, K Lekadir, JA Schnabel

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference article in proceedingAcademicpeer-review

Abstract

Adversarial learning helps generative models translate MRI from source to target sequence when lacking paired samples. However, implementing MRI synthesis with adversarial learning in clinical settings is challenging due to training instability and mode collapse. To address this issue, we leverage intermediate sequences to estimate the common latent space among multi-sequence MRI, enabling the reconstruction of distinct sequences from the common latent space. We propose a generative model that compresses discrete representations of each sequence to estimate the Gaussian distribution of vector-quantized common (VQC) latent space between multiple sequences. Moreover, we improve the latent space consistency with contrastive learning and increase model stability by domain augmentation. Experiments using BraTS2021 dataset show that our non-adversarial model outperforms other GAN-based methods, and VQC latent space aids our model to achieve (1) anti-interference ability, which can eliminate the effects of noise, bias fields, and artifacts, and (2) solid semantic representation ability, with the potential of one-shot segmentation. Our code is publicly available (https://github.com/fiy2W/mri_seq2seq).
Original languageEnglish
Title of host publicationMEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION
Subtitle of host publicationMICCAI 2024, PT XI
EditorsMarius George Linguraru, Aasa Feragen, Ben Glocker, Julia A. Schnabel, Qi Dou, Stamatia Giannarou, Karim Lekadir
PublisherSpringer
Pages481-491
Number of pages11
Volume15011
ISBN (Electronic)9783031721205
ISBN (Print)9783031721199
DOIs
Publication statusPublished - 2024
Event27th International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI) - Hotel du Golf Rotana, Marrakesh, Morocco
Duration: 6 Oct 202410 Oct 2024
Conference number: 27th
https://conferences.miccai.org/2024/en/

Publication series

SeriesLecture Notes in Computer Science
Volume15011
ISSN0302-9743

Conference

Conference27th International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI)
Abbreviated titleMICCAI 2024
Country/TerritoryMorocco
CityMarrakesh
Period6/10/2410/10/24
Internet address

Keywords

  • Latent Space
  • MRI synthesis
  • Multi-Sequence MRI

Fingerprint

Dive into the research topics of 'Non-adversarial Learning: Vector-Quantized Common Latent Space for Multi-sequence MRI'. Together they form a unique fingerprint.

Cite this