Lightweight posterior construction for gravitational-wave catalogs with the Kolmogorov-Arnold network

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liu, Wenshuai, Dong, Yiming, Wang, Ziming, Shao, Lijing
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908692575682560
author Liu, Wenshuai
Dong, Yiming
Wang, Ziming
Shao, Lijing
author_facet Liu, Wenshuai
Dong, Yiming
Wang, Ziming
Shao, Lijing
contents Neural density estimation has seen widespread applications in the gravitational-wave (GW) data analysis, which enables real-time parameter estimation for compact binary coalescences and enhances rapid inference for subsequent analysis such as population inference. In this work, we explore the application of using the Kolmogorov-Arnold network (KAN) to construct efficient and interpretable neural density estimators for lightweight posterior construction of GW catalogs. By replacing conventional activation functions with learnable splines, KAN achieves superior interpretability, higher accuracy, and greater parameter efficiency on related scientific tasks. Leveraging this feature, we propose a KAN-based neural density estimator, which ingests megabyte-scale GW posterior samples and compresses them into model weights of tens of kilobytes. Subsequently, analytic expressions requiring only several kilobytes can be further distilled from these neural network weights with minimal accuracy trade-off. In practice, GW posterior samples with fidelity can be regenerated rapidly using the model weights or analytic expressions for subsequent analysis. Our lightweight posterior construction strategy is expected to facilitate user-level data storage and transmission, paving a path for efficient analysis of numerous GW events in the next-generation GW detectors.
format Preprint
id arxiv_https___arxiv_org_abs_2508_18698
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Lightweight posterior construction for gravitational-wave catalogs with the Kolmogorov-Arnold network
Liu, Wenshuai
Dong, Yiming
Wang, Ziming
Shao, Lijing
General Relativity and Quantum Cosmology
High Energy Astrophysical Phenomena
Applications
Machine Learning
Neural density estimation has seen widespread applications in the gravitational-wave (GW) data analysis, which enables real-time parameter estimation for compact binary coalescences and enhances rapid inference for subsequent analysis such as population inference. In this work, we explore the application of using the Kolmogorov-Arnold network (KAN) to construct efficient and interpretable neural density estimators for lightweight posterior construction of GW catalogs. By replacing conventional activation functions with learnable splines, KAN achieves superior interpretability, higher accuracy, and greater parameter efficiency on related scientific tasks. Leveraging this feature, we propose a KAN-based neural density estimator, which ingests megabyte-scale GW posterior samples and compresses them into model weights of tens of kilobytes. Subsequently, analytic expressions requiring only several kilobytes can be further distilled from these neural network weights with minimal accuracy trade-off. In practice, GW posterior samples with fidelity can be regenerated rapidly using the model weights or analytic expressions for subsequent analysis. Our lightweight posterior construction strategy is expected to facilitate user-level data storage and transmission, paving a path for efficient analysis of numerous GW events in the next-generation GW detectors.
title Lightweight posterior construction for gravitational-wave catalogs with the Kolmogorov-Arnold network
topic General Relativity and Quantum Cosmology
High Energy Astrophysical Phenomena
Applications
Machine Learning
url https://arxiv.org/abs/2508.18698