swMATH ID: 37026
Software Authors: Park, J. W.
Description: baobab: Training data generator for hierarchically modeling strong lenses with Bayesian neural networks. The baobab package can generate images of strongly-lensed systems, given some configurable prior distributions over the parameters of the lens and light profiles as well as configurable assumptions about the instrument and observation conditions. It supports prior distributions ranging from artificially simple to empirical. A major use case for baobab is the generation of training and test sets for hierarchical inference using Bayesian neural networks (BNNs). The idea is that Baobab will generate the training and test sets using different priors. A BNN trained on the training dataset learns not only the parameters of individual lens systems but also, implicitly, the hyperparameters describing the training set population (the training prior). Such hierarchical inference is crucial in scenarios where the training and test priors are different, so that techniques such as importance weighting can be employed to bridge the gap in the BNN response.
Homepage: https://doi.org/10.5281/zenodo.4476822
Source Code:  https://github.com/jiwoncpark/baobab
Dependencies: Python
Related Software: SLSprinkler; HDF5; PyYAML; h5py; Astropy; Matplotlib; SciPy; pandas; NumPy; lenstronomy; Python; deeplenstronomy
Cited in: 0 Publications