# Domain-Adaptive Neural Posterior Estimation for Strong Gravitational Lens Analysis

arXiv: https://arxiv.org/abs/2410.16347 · HTML: https://mlsanity.com/paper/2410.16347/

## Key facts

- Cited by 7 papers in the map (data through 14 Oct 2026).
- Builds on 34 of its 106 references that are in the map.
- First posted on arXiv 21 Oct 2024.

## Builds on (most cited first, 30 of 34)

- [PyTorch: An Imperative Style, High-Performance Deep Learning Library](https://mlsanity.com/paper/1912.01703/) (2019)
- [Domain-Adversarial Training of Neural Networks](https://mlsanity.com/paper/1505.07818/) (2015)
- [A Comprehensive Survey on Transfer Learning](https://mlsanity.com/paper/1911.02685/) (2019)
- [Masked Autoregressive Flow for Density Estimation](https://mlsanity.com/paper/1705.07057/) (2017)
- [The frontier of simulation-based inference](https://mlsanity.com/paper/1911.01429/) (2019)
- [A Survey of Unsupervised Deep Domain Adaptation](https://mlsanity.com/paper/1812.02849/) (2018)
- [A Brief Review of Domain Adaptation](https://mlsanity.com/paper/2010.03978/) (2020)
- [Automatic Posterior Transformation for Likelihood-Free Inference](https://mlsanity.com/paper/1905.07488/) (2019)
- [A review of domain adaptation without target labels](https://mlsanity.com/paper/1901.05335/) (2019)
- [An introduction to domain adaptation and transfer learning](https://mlsanity.com/paper/1812.11806/) (2018)
- [A Review of Single-Source Deep Unsupervised Visual Domain Adaptation](https://mlsanity.com/paper/2009.00155/) (2020)
- [All-in-one simulation-based inference](https://mlsanity.com/paper/2404.09636/) (2024)
- [Transfer Adaptation Learning: A Decade Survey](https://mlsanity.com/paper/1903.04687/) (2019)
- [Detecting Model Misspecification in Amortized Bayesian Inference with Neural Networks](https://mlsanity.com/paper/2112.08866/) (2021)
- [A Survey on Deep Domain Adaptation for LiDAR Perception](https://mlsanity.com/paper/2106.02377/) (2021)
- [Robust Neural Posterior Estimation and Statistical Model Criticism](https://mlsanity.com/paper/2210.06564/) (2022)
- [Investigating the Impact of Model Misspecification in Neural Simulation-based Inference](https://mlsanity.com/paper/2209.01845/) (2022)
- [Survey on Unsupervised Domain Adaptation for Semantic Segmentation for Visual Perception in Automated Driving](https://mlsanity.com/paper/2304.11928/) (2023)
- [A Closer Look at Domain Shift for Deep Learning in Histopathology](https://mlsanity.com/paper/1909.11575/) (2019)
- [Adversarial Weighting for Domain Adaptation in Regression](https://mlsanity.com/paper/2006.08251/) (2020)
- [Video Unsupervised Domain Adaptation with Deep Learning: A Comprehensive Survey](https://mlsanity.com/paper/2211.10412/) (2022)
- [DeepMerge II: Building Robust Deep Learning Algorithms for Merging Galaxy Identification Across Domains](https://mlsanity.com/paper/2103.01373/) (2021)
- [Neural Posterior Estimation with Differentiable Simulators](https://mlsanity.com/paper/2207.05636/) (2022)
- [Unsupervised Domain Adaptation for Constraining Star Formation Histories](https://mlsanity.com/paper/2112.14072/) (2021)
- [Domain adaptation techniques for improved cross-domain study of galaxy mergers](https://mlsanity.com/paper/2011.03591/) (2020)
- [Accelerated Bayesian SED Modeling using Amortized Neural Posterior Estimation](https://mlsanity.com/paper/2203.07391/) (2022)
- [Domain Adaptive Graph Neural Networks for Constraining Cosmological Parameters Across Multiple Data Sets](https://mlsanity.com/paper/2311.01588/) (2023)
- [Preconditioned Neural Posterior Estimation for Likelihood-free Inference](https://mlsanity.com/paper/2404.13557/) (2024)
- [Robustness of deep learning algorithms in astronomy -- galaxy morphology studies](https://mlsanity.com/paper/2111.00961/) (2021)
- [Domain Adaptation via Minimax Entropy for Real/Bogus Classification of Astronomical Alerts](https://mlsanity.com/paper/2308.07538/) (2023)

## Built on by (most cited first, 7 of 7)

- [Diffusion Models in Simulation-Based Inference: A Tutorial Review](https://mlsanity.com/paper/2512.20685/) (2025)
- [Does Unsupervised Domain Adaptation Improve the Robustness of Amortized Bayesian Inference? A Systematic Evaluation](https://mlsanity.com/paper/2502.04949/) (2025)
- [Tests for model misspecification in simulation-based inference: from local distortions to global model checks](https://mlsanity.com/paper/2412.15100/) (2024)
- [Inductive Domain Transfer In Misspecified Simulation-Based Inference](https://mlsanity.com/paper/2508.15593/) (2025)
- [Misspecification-robust amortised simulation-based inference using variational methods](https://mlsanity.com/paper/2509.05724/) (2025)
- [Safe Domain Adaptation for Physics: Overcoming Nuisances, Label Shifts, and Simulation Priors](https://mlsanity.com/paper/2608.18190/) (2026)
- [Mitigating Representation Gaps in Amortized Bayesian Inference with Auxiliary Supervision](https://mlsanity.com/paper/2609.39525/) (2026)

## Abstract (arXiv)

Modeling strong gravitational lenses is prohibitively expensive for modern and next-generation cosmic survey data. Neural posterior estimation (NPE), a simulation-based inference (SBI) approach, has been studied as an avenue for efficient analysis of strong lensing data. However, NPE has not been demonstrated to perform well on out-of-domain target data -- e.g., when trained on simulated data and then applied to real, observational data. In this work, we perform the first study of the efficacy of NPE in combination with unsupervised domain adaptation (UDA). The source domain is noiseless, and the target domain has noise mimicking modern cosmology surveys. We find that combining UDA and NPE improves the accuracy of the inference by 1-2 orders of magnitude and significantly improves the posterior coverage over an NPE model without UDA. We anticipate that this combination of approaches will help enable future applications of NPE models to real observational data.
