CoNeS: Conditional neural fields with shift modulation for multi-sequence MRI translation

Yunjie Chen10000-0001-9478-6953, Marius Staring10000-0003-2885-5812, Olaf M. Neve20000-0002-5104-8448, Stephan R. Romeijn10000-0002-4634-447X, Erik F. Hensen20000-0002-4393-7421, Berit M. Verbist10000-0002-1010-2583, Jelmer M. Wolterink30000-0001-5505-475X, Qian Tao40000-0001-7480-0703
1: Department of Radiology, Leiden University Medical Center, 2: Department of Otorhinolaryngology and Head & Neck Surgery, Leiden University Medical Center, 3: Department of Applied Mathematics, Technical Medical Center, University of Twente, 4: Department of Imaging Physics, Delft University of Technology
Publication date: 2024/02/09
https://doi.org/10.59275/j.melba.2024-d61g
PDF · Code · arXiv

Abstract

Multi-sequence magnetic resonance imaging (MRI) has found wide applications in both modern clinical studies and deep learning research. However, in clinical practice, it frequently occurs that one or more of the MRI sequences are missing due to different image acquisition protocols or contrast agent contraindications of patients, limiting the utilization of deep learning models trained on multi-sequence data. One promising approach is to leverage generative models to synthesize the missing sequences, which can serve as a surrogate acquisition. State-of-the-art methods tackling this problem are based on convolutional neural networks (CNN) which usually suffer from spectral biases, resulting in poor reconstruction of high-frequency fine details. In this paper, we propose Conditional Neural fields with Shift modulation (CoNeS), a model that takes voxel coordinates as input and learns a representation of the target images for multi-sequence MRI translation. The proposed model uses a multi-layer perceptron (MLP) instead of a CNN as the decoder for pixel-to-pixel mapping. Hence, each target image is represented as a neural field that is conditioned on the source image via shift modulation with a learned latent code. Experiments on BraTS 2018 and an in-house clinical dataset of vestibular schwannoma patients showed that the proposed method outperformed state-of-the-art methods for multi-sequence MRI translation both visually and quantitatively. Moreover, we conducted spectral analysis, showing that CoNeS was able to overcome the spectral bias issue common in conventional CNN models. To further evaluate the usage of synthesized images in clinical downstream tasks, we tested a segmentation network using the synthesized images at inference. The results showed that CoNeS improved the segmentation performance when some MRI sequences were missing and outperformed other synthesis models. We concluded that neural fields are a promising technique for multi-sequence MRI translation.

Keywords

Neural fields · Magnetic Resonance Imaging · generative models · image-to-image translation · segmentation

Bibtex @article{melba:2024:004:chen, title = "CoNeS: Conditional neural fields with shift modulation for multi-sequence MRI translation", author = "Chen, Yunjie and Staring, Marius and Neve, Olaf M. and Romeijn, Stephan R. and Hensen, Erik F. and Verbist, Berit M. and Wolterink, Jelmer M. and Tao, Qian", journal = "Machine Learning for Biomedical Imaging", volume = "2", issue = "Special Issue for Generative Models", year = "2024", pages = "657--685", issn = "2766-905X", doi = "https://doi.org/10.59275/j.melba.2024-d61g", url = "https://melba-journal.org/2024:004" }
RISTY - JOUR AU - Chen, Yunjie AU - Staring, Marius AU - Neve, Olaf M. AU - Romeijn, Stephan R. AU - Hensen, Erik F. AU - Verbist, Berit M. AU - Wolterink, Jelmer M. AU - Tao, Qian PY - 2024 TI - CoNeS: Conditional neural fields with shift modulation for multi-sequence MRI translation T2 - Machine Learning for Biomedical Imaging VL - 2 IS - Special Issue for Generative Models SP - 657 EP - 685 SN - 2766-905X DO - https://doi.org/10.59275/j.melba.2024-d61g UR - https://melba-journal.org/2024:004 ER -

2024:004 cover