Foundation Model for Lossy Compression of Spatiotemporal Scientific Data

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Xiao, Lee, Jaemoon, Rangarajan, Anand, Ranka, Sanjay
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913623007297536
author Li, Xiao
Lee, Jaemoon
Rangarajan, Anand
Ranka, Sanjay
author_facet Li, Xiao
Lee, Jaemoon
Rangarajan, Anand
Ranka, Sanjay
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2412_17184
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Foundation Model for Lossy Compression of Spatiotemporal Scientific Data
Li, Xiao
Lee, Jaemoon
Rangarajan, Anand
Ranka, Sanjay
Machine Learning
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.
title Foundation Model for Lossy Compression of Spatiotemporal Scientific Data
topic Machine Learning
url https://arxiv.org/abs/2412.17184