FA-INR: Adaptive Implicit Neural Representations for Interpretable Exploration of Simulation Ensembles

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Ziwei, Duan, Yuhan, Xiong, Tianyu, Chen, Yi-Tang, Chao, Wei-Lun, Shen, Han-Wei
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910086429933568
author Li, Ziwei
Duan, Yuhan
Xiong, Tianyu
Chen, Yi-Tang
Chao, Wei-Lun
Shen, Han-Wei
author_facet Li, Ziwei
Duan, Yuhan
Xiong, Tianyu
Chen, Yi-Tang
Chao, Wei-Lun
Shen, Han-Wei
contents Surrogate models are essential for efficient exploration of large-scale ensemble simulations. Implicit neural representations (INRs) provide a compact and continuous framework for modeling spatially structured data, but they often struggle with learning complex localized structures within the scientific fields. Recent INR-based surrogates address this by augmenting INRs with explicit feature structures, but at the cost of flexibility and substantial memory overhead. In this paper, we present Feature-Adaptive INR (FA-INR), an adaptive INR-based surrogate model for high-fidelity and interpretable exploration of ensemble simulations. Instead of relying on structured feature representations, FA-INR leverages cross-attention over a learnable key-value memory bank to allocate model capacity adaptively based on the data characteristics. To further improve scalability, we introduce a coordinate-guided mixture of experts (MoE) framework that enhances both efficiency and specialization of feature representations. More importantly, the learned experts produce an interpretable partition over the simulation domain, enabling scientists to identify complex structures and perform localized parameter-space exploration. Beyond quantitative and qualitative evaluations, we also demonstrate that our learned expert specialization can reveal meaningful scientific insights and support localized sensitivity analysis.
format Preprint
id arxiv_https___arxiv_org_abs_2506_06858
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle FA-INR: Adaptive Implicit Neural Representations for Interpretable Exploration of Simulation Ensembles
Li, Ziwei
Duan, Yuhan
Xiong, Tianyu
Chen, Yi-Tang
Chao, Wei-Lun
Shen, Han-Wei
Machine Learning
Artificial Intelligence
Surrogate models are essential for efficient exploration of large-scale ensemble simulations. Implicit neural representations (INRs) provide a compact and continuous framework for modeling spatially structured data, but they often struggle with learning complex localized structures within the scientific fields. Recent INR-based surrogates address this by augmenting INRs with explicit feature structures, but at the cost of flexibility and substantial memory overhead. In this paper, we present Feature-Adaptive INR (FA-INR), an adaptive INR-based surrogate model for high-fidelity and interpretable exploration of ensemble simulations. Instead of relying on structured feature representations, FA-INR leverages cross-attention over a learnable key-value memory bank to allocate model capacity adaptively based on the data characteristics. To further improve scalability, we introduce a coordinate-guided mixture of experts (MoE) framework that enhances both efficiency and specialization of feature representations. More importantly, the learned experts produce an interpretable partition over the simulation domain, enabling scientists to identify complex structures and perform localized parameter-space exploration. Beyond quantitative and qualitative evaluations, we also demonstrate that our learned expert specialization can reveal meaningful scientific insights and support localized sensitivity analysis.
title FA-INR: Adaptive Implicit Neural Representations for Interpretable Exploration of Simulation Ensembles
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2506.06858