Meta-INR: Efficient Encoding of Volumetric Data via Meta-Learning Implicit Neural Representation

TL;DR

Meta-INR introduces a meta-learning approach to initialize implicit neural representations for volumetric data, enabling faster adaptation and improved generalization across similar datasets, with applications in simulation analysis.

Contribution

It proposes a novel meta-learning based pretraining strategy for INR that accelerates training and enhances interpretability for volumetric data.

Findings

Faster convergence with few gradient updates.

Improved generalization to unseen volumetric data.

Effective in simulation parameter analysis.

Abstract

Implicit neural representation (INR) has emerged as a promising solution for encoding volumetric data, offering continuous representations and seamless compatibility with the volume rendering pipeline. However, optimizing an INR network from randomly initialized parameters for each new volume is computationally inefficient, especially for large-scale time-varying or ensemble volumetric datasets where volumes share similar structural patterns but require independent training. To close this gap, we propose Meta-INR, a pretraining strategy adapted from meta-learning algorithms to learn initial INR parameters from partial observation of a volumetric dataset. Compared to training an INR from scratch, the learned initial parameters provide a strong prior that enhances INR generalizability, allowing significantly faster convergence with just a few gradient updates when adapting to a new volume and better interpretability when analyzing the parameters of the adapted INRs. We demonstrate that Meta-INR can effectively extract high-quality generalizable features that help encode unseen similar volume data across diverse datasets. Furthermore, we highlight its utility in tasks such as simulation parameter analysis and representative timestep selection. The code is available at https://github.com/spacefarers/MetaINR.