FastParam

Machine Learning Strategies for Accelerating Parametric Models for Gravitational Wave Detection in Pulsar Signals
FastParam is a machine learning-enhanced framework for pulsar timing array analysis that accelerates the computational pipeline for detecting gravitational waves from pulsars. The project addresses the computational bottleneck in traditional Bayesian inference methods used in PTA analysis by implementing advanced machine learning strategies to optimise parametric models, significantly reducing the time required to analyse pulsar timing residuals and extract gravitational wave signals. By leveraging neural network architectures and efficient sampling techniques, FastParam enables faster parameter estimation and model comparison, facilitating the detection of nanohertz-frequency gravitational waves in the stochastic background whilst maintaining the statistical rigour required for astrophysical inference. This approach is particularly crucial as PTA datasets grow larger with ongoing observations from collaborations like EPTA, NANOGrav, and PPTA, where traditional analysis methods become prohibitively time-consuming.
FastParam has been selected and funded under the ICSC – Centro Nazionale di Ricerca in HPC, Big Data e Quantum Computing, as part of the PNRR, and supported by the European Union – NextGenerationEU.
Access Resources Learn More

Welcome to the FastParam Website

FastParam is an innovative machine-learning project for speeding up astrophysical inference in the context of Gravitational Wave discovery. Here you will get to know our work, the pipeline we built and you will be able to browse the results we obtained. We aim to present the scientific impact of our work as well as the open-source data released from the community and the software we released.

Everything is open and accessible. See the Outputs section below to learn more, or open the FastParam Resource Hub to access the project resources. If you’d like to download datasets or code, we’ll just ask you to fill out a short form with your affiliation and intended use.

Discover the Project

Learn the challenges posed by high-dimensional inference in Gravitational Wave discovery. Explore the solutions developed within the FastParam Project to address them and the integrated pipeline for dimentionality reduction and fast sampling techniques.

Learn More

Access Resources

Browse the EPTA-DR2 dataset our work has been based on and the output of the different approaches from the state-of-the-art to the final stage of development of FastParam.

Explore

Check Usage Terms

Find clear and transparent information about licensing conditions, recommended citation formats, and guidelines for the responsible reuse of materials. These terms help ensure that FastParam's outputs are properly credited and can be applied ethically in future research.

View Terms

About the FastParam Project

FastParam is a research project funded by the ICSC – Centro Nazionale di Ricerca in HPC, Big Data e Quantum Computing, as part of the PNRR, and supported by the European Union – NextGenerationEU. It aims at reducing inference times in the context of Pulsar Timing Arrays, galactic-scale observational instruments consisting of sets of pulsars, whose beams' time of arrival are registered over a time window of 15–20 years. Many of such datasets exist ranging from EPTA to PPTA, accounting for a growing community of researchers. The state-of-the-art software enables complex modelling and inference of astrophysical and cosmological sources of delay in the time of arrival of the pulsar signals, but faces enormous challenges in terms of computational time.

We have explored and implemented dimensionality reduction and machine learning-driven sampling solutions to reduce such time, allowing rapid testing of physical hypotheses pushing innovation and theoretical development in the field. We used Hierarchical Bayesian Models with normalizing flow boosted sampling for fast execution and hyperparameter reparametrisation for better accuracy. We also developed a novel Empirical Bayes approach for efficient application of nested sampling to the Pulsar Timing Array problem. The method has shown promising results in terms of both efficiency and speed as well as accuracy with respect to the injected target values from synthetic data. The project delivers openly accessible tools: frameworks, PTA analysis algorithms, and source code repositories, all provided under transparent licensing terms. In this way, FastParam contributes to reproducible science and accelerates discovery for both the astrophysics and AI communities.

🧠 Parameter space transformation

Normalizing flows for invertibility of quasi-linear tranformations learned from prior samples. Same models are used to simplify the multimodal parameter distributions.

🔬 Physics‑Informed Empirical Modeling

From previous studies and theory, estimated values of physical parameters are available, which can be incorporated in our Empirical Bayes approach via advanced regularisation techniques.

🧪 High quality inference

Prior-likelihood overlap ensured by our approaches lead to comporable-to-better accuracy of inference with respect to the literature.

Project outputs

  • Project outputs: A software package for rapid inference on Pulsar Timing Arrays. On input an enterprise PTA object, our pipeline can follow two possible roads stemming from the hierarchical Bayesian modeling of the problem. An accurate inference reparametrisation of prior hyperparameter space is available for posterior independence from possibily misspecified priors. An empirical Bayes approach has been developed for non-hierarchical modeling with minimal loss of accuracy but enormous gain in speed.
  • Synthetic datasets: Have been produced for software testing, which are modeled over real pulsars according to their physical description. Synthetic data is obtained using widely acknowledge instruments like the tempo2 package. Injecting noise and Gravitational Wave parameters, we have been able to recover such values in record time.
  • Inference: A wide demo set of posterior and plots which showcases the output of our new inference pipeline. One can browse by prior choice, number of pulsars and physical model assumptions: we have pushed the boundaries of our research by stress testing on the widest possible scenarios.

Validation & benchmarking

Consistency checks with injected values has been the core metric for the validation of our inference pipeline. Also, the growing literature provides benchmark results for accuracy, which we have hustled to meet whilst focusing on fast sampling and inference.

Methodology at a glance

The core innovations in the two branches of the inference pipeline can be summarised as:

  • flow-guided transformation of the parameter space for posterior independence of prior assumptions and genAI enhanced nested sampling for accurate inference of noise and Gravitational wave parameters;
  • evolutionary computation in empirical prior selection for induced prior-likelihood overlap, subsequent hyperparameters discard and minimal sample rejection during nested sampling.

Ownership, collaboration & funding

FastParam is conceived and developed under the leadership of Koexai S.r.l., which coordinates all project activities and ensures their delivery. The valuable support of Massimo Meneghetti who acted as scientific referee, nominated by INAF (Spoke 3 leader). The project benefits from the valuable scientific input of Eleonora Villa who has shared her expetise in physical modelling.

FastParam has been selected and funded under the ICSC – Centro Nazionale di Ricerca in HPC, Big Data e Quantum Computing, as part of the PNRR, and supported by the European Union – NextGenerationEU.

  • European Union — NextGenerationEU
  • PNRR — Piano Nazionale di Ripresa e Resilienza
  • ICSC — National Centre for HPC, Big Data and Quantum Computing

Outputs

Explore FastParam's outputs: datasets, source code, publications, diagrams, and findings. For downloads and full details, open the Resource Hub.

FastParam Core Repository

Browse the source code, training scripts, model architectures, and reproducible configurations. Clone the repository to run your own experiments.

  • Type: Source code
  • Version: v1.0
  • Released: 30/11/2025
  • Files: Python
  • Full size: TBC
  • Licence: MIT
Last updated: 30/11/2025

Technical Diagrams & Notes

View network architectures and training workflows. Includes visual diagrams and technical specifications.

  • Type: Documentation
  • Version: v1.0
  • Released: 30/11/2025
  • Files:PDF
  • Full size: Various
  • Licence: CC BY 4.0
Last updated: 30/11/2025

Requesting the full datasets

Large files & licensing: full releases are provided for non-commercial research on a short request.

  1. Tell us your name, affiliation, email, and intended use.
  2. We'll reply with access instructions and a download link.
info@koexai.com

Contact & Acknowledgements

Project Contact

For resource access requests, collaboration inquiries, or technical questions:

info@koexai.com

Ownership, Collaboration & Funding

Koexai S.r.l., — Project lead and coordinating body.

Massimo Meneghetti — Scientific referee, nominated by INAF (Spoke 3 leader).

Eleonora Villa — Researcher in Univeristy of Milan (Bicocca) with great experience and expertise in physical modelling.