No Cover Image

Journal article 765 views 135 downloads

Sampling strategies for learning-based 3D medical image compression

Omniah Nagoor, Joss Whittle Orcid Logo, Jingjing Deng, Benjamin Mora Orcid Logo, Mark Jones Orcid Logo

Machine Learning with Applications, Volume: 8, Start page: 100273

Swansea University Authors: Omniah Nagoor, Jingjing Deng, Benjamin Mora Orcid Logo, Mark Jones Orcid Logo

  • 1-s2.0-S2666827022000135-main.pdf

    PDF | Version of Record

    © 2022 The Author(s). Published by Elsevier Ltd. This is an open access article under the CC BY license

    Download (4.31MB)

Abstract

Recent achievements of sequence prediction models in numerous domains, including compression, provide great potential for novel learning-based codecs. In such models, the input sequence’s shape and size play a crucial role in learning the mapping function of the data distribution to the target outpu...

Full description

Published in: Machine Learning with Applications
ISSN: 2666-8270
Published: Elsevier BV 2022
Online Access: Check full text

URI: https://cronfa.swan.ac.uk/Record/cronfa59383
Tags: Add Tag
No Tags, Be the first to tag this record!
Abstract: Recent achievements of sequence prediction models in numerous domains, including compression, provide great potential for novel learning-based codecs. In such models, the input sequence’s shape and size play a crucial role in learning the mapping function of the data distribution to the target output. This work examines numerous input configurations and sampling schemes for a many-to-one sequence prediction model, specifically for compressing 3D medical images (16-bit depth) losslessly. The main objective is to determine the optimal practice for enabling the proposed Long Short-Term Memory (LSTM) model to achieve high compression ratio and fast encoding-decoding performance.Our LSTM models are trained with 4-fold cross-validation on 12 high-resolution CT dataset while measuring model’s compression ratios and execution time. Several configurations of sequences have been evaluated, and our results demonstrate that pyramid-shaped sampling represents the best trade-off between performance and compression ratio (up to 3x). We solve a problem of non-deterministic environments that allow our models to run in parallel without much compression performance drop.Experimental evaluation was carried out on datasets acquired by different hospitals, representing different body segments, and distinct scanning modalities (CT and MRI). Our new methodology allows straightforward parallelisation that speeds-up the decoder by up to 37x compared to previous methods. Overall, the trained models demonstrate efficiency and generalisability for compressing 3D medical images losslessly while still outperforming well-known lossless methods by approximately 17% and 12%. To the best of our knowledge, this is the first study that focuses on voxel-wise predictions of volumetric medical imaging for lossless compression.
Keywords: 3D predictors; Deep learning; Lossless compression; Medical image compression; Sequence prediction model; LSTM
College: Faculty of Science and Engineering
Start Page: 100273