MRIComp4Flow: Compression of 3D Brain MRI for Training Multi-Modal Generative Models

Lisa K. Fischer1,*, Mykhailo Riabets1,*, Daniel Rueckert4,5, Benedikt Wiestler2,3, Anke Meyer-Baese6,7, Sandeep Nagar2,3,7,*
1Technical University of Munich 2Munich Center for Machine Learning 3AI for Image-Guided Diagnosis and Therapy, TUM
4Chair for AI in Healthcare and Medicine, TUM University Hospital 5Imperial College London 6Florida State University 7Institute for Advanced Study, TUM
*Equal contribution
SASHIMI: Simulation and Synthesis in Medical Imaging
MICCAI 2026 Workshop

Abstract

Large-scale multi-modal MRI datasets impose substantial storage and I/O costs, limiting the training of 3D generative models on commodity infrastructure. While lossy compression is known to preserve accuracy for discriminative segmentation networks, its effect on generative models, which must learn the full data distribution rather than a decision boundary, is unexplored. We study whether standard image codecs can effectively compress semantically rich brain tumor MRI while preserving the fidelity required to train and deploy a 3D MRI generative model. Each 3D volume is compressed with JPEG2000 or a near-lossless JPEG-LS pipeline. Next, a Wavelet Flow Matching model, conditioned on BraTS image sequences (T1n, T1c, T2, T2f), is trained on compressed data. At a 20:1 compression ratio, synthesis quality is statistically equivalent to a model trained on uncompressed data (mean PSNR 27.3 dB vs. 27.0 dB and mean SSIM 0.95 vs. 0.96 across modalities). JPEG2000 compression is a practical step toward scalable 3D MRI generative modeling without degrading synthesis quality.

Pipeline

MRIComp4Flow pipeline: BraTS volumes are compressed offline with JPEG2000 or JPEG-LS, decoded on the fly, and synthesized with a unified Wavelet Flow Matching model.

BraTS volumes are compressed offline, decoded on the fly, and used to train a unified Wavelet Flow Matching synthesizer.

Qualitative results

Poster

AI-IDT: AI for Image-Guided Diagnosis and Therapy TUM Institute for Advanced Study Florida State University

BibTeX

@inproceedings{fischer2026mricomp4flow,
  title={{MRIComp4Flow}: Compression of {3D} Brain {MRI} for Training Multi-Modal Generative Models},
  author={Fischer, Lisa K. and Riabets, Mykhailo and Rueckert, Daniel and Wiestler, Benedikt and Meyer-Baese, Anke and Nagar, Sandeep},
  booktitle={Simulation and Synthesis in Medical Imaging ({SASHIMI}), {MICCAI} 2026 Workshop},
  year={2026},
  url={https://arxiv.org/abs/2608.10291}
}