Foundation Model for Lossy Compression of Spatiotemporal Scientific Data

TL;DR

This paper introduces a foundation model combining variational autoencoders and super-resolution for efficient lossy compression of scientific spatiotemporal data, achieving higher compression ratios and better quality.

Contribution

It presents a novel foundation model that integrates hyper-prior VAE and super-resolution modules for improved scientific data compression.

Findings

Achieves up to 4x higher compression ratios than existing methods.

Super-resolution module increases compression efficiency by 30%.

Model generalizes well across unseen domains and data shapes.

Abstract

We present a foundation model (FM) for lossy scientific data compression, combining a variational autoencoder (VAE) with a hyper-prior structure and a super-resolution (SR) module. The VAE framework uses hyper-priors to model latent space dependencies, enhancing compression efficiency. The SR module refines low-resolution representations into high-resolution outputs, improving reconstruction quality. By alternating between 2D and 3D convolutions, the model efficiently captures spatiotemporal correlations in scientific data while maintaining low computational cost. Experimental results demonstrate that the FM generalizes well to unseen domains and varying data shapes, achieving up to 4 times higher compression ratios than state-of-the-art methods after domain-specific fine-tuning. The SR module improves compression ratio by 30 percent compared to simple upsampling techniques. This approach significantly reduces storage and transmission costs for large-scale scientific simulations while preserving data integrity and fidelity.