GaMPEN: A Machine Learning Framework for Estimating Bayesian Posteriors of Galaxy Morphological Parameters
Open accessAbstract
We introduce a novel machine learning framework for estimating the Bayesian posteriors of morphological parameters for arbitrarily large numbers of galaxies. The Galaxy Morphology Posterior Estimation Network (GaMPEN) estimates values and uncertainties for a galaxy's bulge-to-total light ratio ($L_B/L_T$), effective radius ($R_e$), and flux ($F$). To estimate posteriors, GaMPEN uses the Monte Carlo Dropout technique and incorporates the full covariance matrix between the output parameters in its loss function. GaMPEN also uses a Spatial Transformer Network (STN) to automatically crop input galaxy frames to an optimal size before determining their morphology. This will allow it to be applied to new data without prior knowledge of galaxy size. Training and testing GaMPEN on galaxies simulated to match $z < 0.25$ galaxies in Hyper Suprime-Cam Wide $g$-band images, we demonstrate that GaMPEN achieves typical errors of $0.1$ in $L_B/L_T$, $0.17$ arcsec ($\sim 7\%$) in $R_e$, and $6.3\times10^4$ nJy ($\sim 1\%$) in $F$. GaMPEN's predicted uncertainties are well-calibrated and accurate ($<5\%$ deviation) -- for regions of the parameter space with high residuals, GaMPEN correctly predicts correspondingly large uncertainties. We also demonstrate that we can apply categorical labels (i.e., classifications such as "highly bulge-dominated") to predictions in regions with high residuals and verify that those labels are $\gtrsim 97\%$ accurate. To the best of our knowledge, GaMPEN is the first machine learning framework for determining joint posterior distributions of multiple morphological parameters and is also the first application of an STN to optical imaging in astronomy.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
Ghosh et al. (2022) showed that a combination of MDNs and Monte Carlo Dropout can provide well calibrated uncertainties of galaxy structural parameters.
“This has been recently done by Ghosh et al. (2022) who showed that a combination of MDNs and Monte Carlo Dropout can provide well calibrated uncertainties of galaxy structural parameters.”
From The Dawes Review 10: The impact of deep learning for the analysis of galaxy surveys (2023) · cites this paper as Ghosh et al. (2022)
Sections 4.1 and 4.4 describe using Monte Carlo dropout to estimate posteriors for galaxy morphological parameters, and Section 5 reports accurately calibrated posteriors. But Sections 3.2 and 4.2 describe a network predicting a Gaussian mean and covariance, not an MDN; our copy does not support the claim that the demonstrated combination included MDNs. Page numbers are unavailable in our copy. (LLM, conf=0.86)
Experimental — accuracy not yet measured
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
No references could be resolved yet.
Research works printed but not linked: 83
- Aihara, H., Alsayyad, Y., Ando, M., et al. 2019, PASJ, 71, 114 · DOI · Paper page
- Aihara, H., Armstrong, R., Bickerton, S., et al. 2018, PASJ, 70, S8 · DOI
- Bender, R., Burstein, D., & Faber, S. M. 1992, ApJ, 399, 462 · DOI
- Bernardi, M., Meert, A., Sheth, R. K., et al. 2013, MNRAS, 436, 697 · DOI
- Binney, J., & Merrifield, M. 1998, Galactic Astronomy (Princeton, NJ: Princeton Univ. Press), 796
- Chen, A., Chow, A., Davidson, A., et al. 2020, in Proc. Fourth Int. Workshop on Data Management for End-to-End Machine Learning (New York: ACM), 1
- Cheng, T.-Y., Conselice, C. J., Aragón-Salamanca, A., et al. 2021, MNRAS, 507, 4425 · DOI
- Cholesky, A.-L. 1924, BGeod, 2, 67 · DOI
- Chollet, F. 2021, Deep Learning with Python (2nd edn.; New York: Simon & Schuster), https://www.simonandschuster.com/books/Deep-Learning- with-Python-Second-Edition/Francois-Chollet/9781617296864
- Cranmer, M., Tamayo, D., Rein, H., et al. 2021, PNAS, 118, 2026053118 · DOI
- Denker, J. S., & Lecun, Y. 1990, in Advances in Neural Information Processing Systems, 3, ed.
- R. P. Lippmann, J. Moody, & D. Touretzky (San Francisco, CA: Morgan Kaufmann), https://proceedings.neurips.cc/ paper/1990/file/7eacb532570ff6858afd2723755ff790-Paper.pdf
- Dieleman, S., Willett, K. W., & Dambre, J. 2015, MNRAS, 450, 1441 · DOI · Paper page
- Dimauro, P., Daddi, E., Shankar, F., et al. 2022, MNRAS, 513, 256 · DOI
- Dodge, Y. 2008, The Concise Encyclopedia of Statistics (New York: Springer), 502, http://www.stewartschultz.com/statistics/books/The% 20Concise%20Encyclopedia%20of%20Statistics.pdf
- Foreman-Mackey, D. 2016, JOSS, 1, 24 · DOI
- Fukushima, K. 1980, Biol. Cybern., 36, 193 · DOI · Paper page
- Gal, Y., & Ghahramani, Z. 2016, in 33rd Int. Conf. Machine Learning, 48, ICML 2016 (PMLR) (Cambridge, MA: MIT Press), 1050, https:// proceedings.mlr.press/v48/gal16.html
- Ghosh, A., Urry, C. M., Wang, Z., et al. 2020, ApJ, 895, 112 · DOI · Paper page
- Goodfellow, I., Bengio, Y., & Courville, A. 2016, Deep Learning (Cambridge, MA: MIT Press), https://www.deeplearningbook.org/
- Grogin, N. A., Kocevski, D. D., Faber, S. M., et al. 2011, ApJS, 197, 35 · DOI
- Harris, C. R., Millman, K. J., van der Walt, S. J., et al. 2020, Natur, 585, 357 · DOI · Paper page
- Hausen, R., & Robertson, B. E. 2020, ApJS, 248, 20 · DOI · Paper page
- Haussler, B., McIntosh, D. H., Barden, M., et al. 2007, ApJS, 172, 615 · DOI
- Hoyle, B. 2016, A&C, 16, 34 · DOI
- Hubble, E. P. 1926, ApJ, 64, 321 · DOI
- Huertas-Company, M., Bernardi, M., Pérez-González, P. G., et al. 2016, MNRAS, 462, 4495 · DOI
- Huertas-Company, M., Gravet, R., Cabrera-Vives, G., et al. 2015, ApJS, 221, 8 · DOI · Paper page
- Hunter, J. D. 2007, CSE, 9, 90 · DOI · Paper page
- Ivezić, Z., Kahn, S. M., Tyson, J. A., et al. 2019, ApJ, 873, 111 · DOI · Paper page
- Jaderberg, M., Simonyan, K., Zisserman, A., & Kavukcuoglu, K. 2015, in Proc. 28th Int. Conf. on Neural Information Processing Systems, Vol. 1, ed.
- C. Cortes et al. (Cambridge, MA: MIT Press), http://proceedings.neurips. cc/paper/2015/file/33ceb07bf4eeb3da587e268d663aba1a-Paper.pdf
- Jordan, M. I., Ghahramani, Z., Jaakkola, T. S., & Saul, L. K. 1999, Mach. Learn., 37, 183 · DOI
- Kendall, A., & Gal, Y. 2017, arXiv:1703.04977 · arXiv
- Kim, E. J., & Brunner, R. J. 2017, MNRAS, 464, 4463 · DOI · Paper page
- Kloek, T., & van Dijk, H. K. 1978, Econometrica, 46, 1 · DOI
- Lecun, Y., Bengio, Y., & Hinton, G. 2015, Natur, 521, 436 · DOI · Paper page
- LeCun, Y., Bottou, L., Bengio, Y., & Haffner, P. 1998, IEEEP, 86, 2278 · DOI · Paper page
- Lintott, C. J., Schawinski, K., Slosar, A., et al. 2008, MNRAS, 389, 1179 · DOI · Paper page
- McKinney, W. 2010, in Proc. of the 9th Python in Science Conf., ed.
- Meert, A., Vikram, V., & Bernardi, M. 2013, MNRAS, 433, 1344 · DOI
- Natarajan, P., Tang, K. S., Khochfar, S., et al. 2021, arXiv:2103.13932 · arXiv
- Nielsen, M. A. 2015, Neural Networks and Deep Learning (Determination Press), http://neuralnetworksanddeeplearning.com/
- Paszke, A., Gross, S., Massa, F., et al. 2019, in Advances in Neural Information Processing Systems 32 (NeurIPS 2019), ed.
- H. Wallach et al. (Red Hook, NY: Curran Associates Inc.), 8024, https://proceedings.neurips.cc/paper/ 2019/hash/bdbca288fee7f92f2bfa9f7012727740-Abstract.html
- Pawlowski, N., Brock, A., Lee, M. C. H., Rajchl, M., & Glocker, B. 2017, arXiv:1711.01297 · arXiv
- Pedregosa, F., Varoquaux, G., Gramfort, A., et al. 2011, J. Mach. Learn. Res., 12, 2825 · DOI
- Peng, C. Y., Ho, L. C., Impey, C. D., & Rix, H.-W. 2002, AJ, 124, 266 · DOI
- Perreault Levasseur, L., Hezaveh, Y. D., & Wechsler, R. H. 2017, ApJL, 850, L7 · DOI · Paper page
- Planck Collaboration, Aghanim, N., Akrami, Y., et al. 2018, A&A, 641, A6 · DOI · Paper page
- Powell, M. C., Urry, C. M., Cardamone, C. N., et al. 2017, ApJ, 835, 22 · DOI
- Pozzetti, L., Bolzonella, M., Zucca, E., et al. 2010, A&A, 523, A13 · DOI
- Price-Whelan, A. M., Sipőcz, B. M., Günther, H. M., et al. 2018, AJ, 156, 123 · DOI
- Racca, G. D., Laureijs, R., Stagnaro, L., et al. 2016, Proc. SPIE, 9904, 99040O · DOI
- Rix, H., Barden, M., Beckwith, S. V. W., et al. 2004, ApJS, 152, 163 · DOI
- Robitaille, T. P., Tollerud, E. J., Greenfield, P., et al. 2013, A&A, 558, A33 · DOI · Paper page
- Rowe, B. T., Jarvis, M., Mandelbaum, R., et al. 2015, A&C, 10, 121 · DOI
- Rumelhart, D. E., Hinton, G. E., & Williams, R. J. 1986, Natur, 323, 533 · DOI · Paper page
- Schawinski, K., Urry, C. M., Simmons, B. D., et al. 2014, MNRAS, 440, 889 · DOI
- Schmidhuber, J. 2015, NN, 61, 85 · DOI · Paper page
- Shallue, C. J., & Vanderburg, A. 2018, AJ, 155, 94 · DOI
- Sharma, K., Kembhavi, A., Kembhavi, A., et al. 2020, MNRAS, 491, 2280 · DOI
- Shimakawa, R., Tanaka, T. S., Toshikage, S., & Tanaka, M. 2021, PASJ, 73, 1575 · DOI
- Simard, L., Willmer, C. N. A., Vogt, N. P., et al. 2002, ApJS, 142, 1 · DOI
- Simmons, B. D., Lintott, C., Willett, K. W., et al. 2017, MNRAS, 464, 4420 · DOI
- Simonyan, K., & Zisserman, A. 2014, arXiv:1409.1556 · arXiv
- Spergel, D., Gehrels, N., Breckinridge, J., et al. 2013, arXiv:1305.5422 · arXiv
- Srivastava, N., Hinton, G., Krizhevsky, A., Sutskever, I., & Salakhutdinov, R. 2014, J. Mach. Learn. Res., 15, 1929, https://jmlr.org/papers/v15/ srivastava14a.html
- Tanaka, T. S., Shimakawa, R., Shimasaku, K., et al. 2022, PASJ, 74, 1 · DOI
- Tarsitano, F., Bruderer, C., Schawinski, K., & Hartley, W. G. 2022, MNRAS, 511, 3330 · DOI
- Tremaine, S., Gebhardt, K., Bender, R., et al. 2002, ApJ, 574, 740 · DOI
- Tuccillo, D., Huertas-Company, M., Decencière, E., et al. 2018, MNRAS, 475, 894 · DOI · Paper page
- van der Wel, A., Franx, M., van Dokkum, P. G., et al. 2014, ApJ, 788, 28 · DOI
- Vega-Ferrero, J., Domínguez Sánchez, H., Bernardi, M., et al. 2021, MNRAS, 506, 1927 · DOI
- Wagner-Carena, S., Park, J. W., Birrer, S., et al. 2021, ApJ, 909, 187 · DOI
- Walmsley, M., Lintott, C., Géron, T., et al. 2021, MNRAS, 509, 3966 · DOI
- Walmsley, M., Smith, L., Lintott, C., et al. 2020, MNRAS, 491, 1554 · DOI · Paper page
- Wilson, A. G. 2020, arXiv:2001.10995 · arXiv
- Wilson, A. G., & Izmailov, P. 2020, arXiv:2002.08791 · arXiv
- Wu, C., Wong, O. I., Rudnick, L., et al. 2019, MNRAS, 482, 1211 · DOI
- Wuyts, S., Förster Schreiber, N. M., van der Wel, A., et al. 2011, ApJ, 742, 96 · DOI
- York, D. G., Adelman, J., Anderson, J. E., Jr+, et al. 2000, AJ, 120, 1579 · DOI
- Zanisi, L., Huertas-Company, M., Lanusse, F., et al. 2021, MNRAS, 501, 4359 · DOI · Paper page
Cited by
Sorted by citations
The Dawes Review 10: The impact of deep learning for the analysis of galaxy surveys
Huertas-Company, M., & Lanusse, F. (2023) · Publications of the Astronomical Society of Australia · doi:10.1017/pasa.2022.55 · Cited by 102 · see citing works
Auge, C., Sanders, D. B., Treister, E., Urry, C. M., Kirkpatrick, A., Cappelluti, N., Ananna, T. T., Boquien, M., Baloković, M., Civano, F., et al. (2023) · The Astrophysical Journal · doi:10.3847/1538-4357/acf21a · Cited by 12 · see citing works
“A detailed analysis of the host galaxies ’ morphology for each one of these sources is ongoing in the AHA collaboration ( Schawinski et al. 2014; Powell et al. 2017; Ghosh et al. 2022; Tian et al. 2023 ) .”
BackgroundSelf-citationGhosh, A., Urry, C., Powell, M., Shimakawa, R., van den Bosch, F. C., Nagai, D., Mitra, K., & Connolly, A. J. (2024) · Astrophysical Journal · doi:10.3847/1538-4357/ad596f · Cited by 8 · see citing works
“GaMPEN is an ML framework that can estimate full Bayesian posteriors for different structural parameters of galaxies and has been extensively tested on both simulations and HSC-Wide data (Ghosh et al. 2022, 2023).”
BackgroundMethodSelf-citationGhosh, A., Urry, C., Mishra, A., Perreault-Levasseur, L., Natarajan, P., Sanders, D., Nagai, D., Tian, C., Cappelluti, N., Kartaltepe, J., et al. (2022) · Astrophysical Journal · doi:10.3847/1538-4357/acd546 · Cited by 8 · see citing works
“In Ghosh et al. (2022), we extensively tested and reported on GaMPEN’s performance on simulated HSC z ≤ 0.25 g-band galaxies.”
BackgroundMethodResultInfluentialSelf-citationThe Spectral Energy Distributions of Galaxies
Iyer, K. G., Pacifici, C., Calistro-Rivera, G., & Lovell, C. C. (2025) · doi:10.48550/arxiv.2502.17680 · Cited by 6
“The second source of uncertainties (also called epistemic uncertainties; Ghosh et al. 2022) come from modeling limitations and choices.”
LaMassa, S. M., Peca, A., Urry, C. M., Glikman, E., Ananna, T. T., Auge, C., Civano, F., Ghosh, A., Kirkpatrick, A., Koss, M., et al. (2024) · The Astrophysical Journal · doi:10.3847/1538-4357/ad6e7d · Cited by 5 · see citing works
“We are currently in the process of re-leasing a morphological catalog for AGN host galaxies in Stripe 82X using the Galaxy Morphology Posterior Estimation Network (GaMPEN; Ghosh et al. 2022) and PSFGAN (Tian et al. 2023).”
MethodSelf-citationHow to set up your first machine learning project in astronomy
Buchner, J., & Fotopoulou, S. (2024) · Nature Reviews Physics · doi:10.1038/s42254-024-00743-y · Cited by 3 · see citing works
“37 for an example from galaxy morphology).”
Tian, C., Urry, C. M., Ghosh, A., Ofman, R., Ananna, T. T., Auge, C., Cappelluti, N., Powell, M. C., Sanders, D. B., Schawinski, K., et al. (2023) · The Astrophysical Journal · doi:10.3847/1538-4357/acad79 · Cited by 4 · see citing works
“…2015; Huertas-Company et al. 2015; Tuccillo et al. 2018; Dimauro et al. 2018; Domı́nguez Sánchez et al. 2018; Barchi et al. 2020; Walmsley et al. 2020; Vega-Ferrero et al. 2021; Cheng et al. 2021a,b), which can be done both qualitatively (Ghosh et al. 2020) and quantitatively (Ghosh et al. 2022).”
BackgroundSelf-citationA probabilistic deep learning model to distinguish cusps and cores in dwarf galaxies
Expósito-Márquez, J., Brook, C. B., Cintio, A. D., Macciò, A. V., Grand, R. J. J., Battaglia, G., & Arjona-Gálvez, E. (2022) · Monthly Notices of the Royal Astronomical Society · doi:10.1093/mnras/stac3799 · Cited by 8 · see citing works
Igel, C., Ghosh, A., Connolly, A., Robertson, B., Urry, C., Edwards, L., & Shimakawa, R. (2026) · Astrophysical Journal · doi:10.3847/1538-4357/ae771d · Cited by 1 · see citing works
“GaMPEN has been extensively tested on both simulated and real HSC data ( A. Ghosh et al. 2022; A. Ghosh et al. 2023 ) and has been shown to produce accurate estimates with robust uncertainties.”
Self-citationStars and stellar populations in Milky Way and the nearby galaxies
Sivarani, T., Subramanian, S., Bandyopadhyay, A., Banerjee, P., Bhattacharya, S., Choudhury, S., Ghosh, S., Hema, B. P., Jog, C., Hota, A., et al. (2025) · Journal of astrophysics and astronomy · doi:10.1007/s12036-024-10030-y · Cited by 1 · see citing works
Tian, C., Urry, C., Ghosh, A., Nagai, D., Ananna, T., Powell, M., Auge, C., Mishra, A., Sanders, D., Cappelluti, N., et al. (2025) · Astrophysical Journal · doi:10.3847/1538-4357/adaec0 · Cited by 1 · see citing works
Self-citationZheng, W.-Y., Wang, L.-L., Yin, Y.-P., Zhao, L.-M., Wang, W.-B., & Li, Z.-J. (2026) · Universe · doi:10.3390/universe12100297
“In particular, Ghosh et al. [28] quantified aleatoric and epistemic uncertainties for galaxy morphological parameters, demonstrating the value of posterior-aware deep learning for galaxy-property inference.”
张业旺, Yi, Z., Qin, Z., Du, W., Li, R., Liu, M., Bu, Y., & Li, D. (2026) · The Astrophysical Journal Supplement Series · doi:10.3847/1538-4365/ae7b29
“…lensing parameters, D. Tanoglidis et al. ( 2022 ) applied Bayesian neural networks to infer structural parameters of simulated LSBGs, and A. Ghosh et al. ( 2022 ) developed the GaMPEN framework, trained on simulated Hyper Suprime-Cam g -band data, which predicts galaxy parameters along…”
- Authors
- Ghosh, A., Urry, C. M., Rau, A., Perreault-Levasseur, L., Cranmer, M., Schawinski, K., Stark, D., Tian, C., Ofman, R., Ananna, T. T., Auge, C., Cappelluti, N., Sanders, D. B., & Treister, E.
- Affiliations
- Yale University (US)
- Mila - Quebec Artificial Intelligence Institute (CA)
- Princeton University (US)
- Dartmouth College (US)
- University of Miami (US)
- University of Hawaiʻi at Mānoa (US)
- Pontificia Universidad Católica de Chile (CL)
- ORCID
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2022
- Volume
- 935
- Issue
- 2
- Pages
- 138-138
- ISSN
- 0004-637X (ISSN-L)
- arXiv
- 2207.05107
- OpenAlex
- W4293239474
- Cited by
- 16
- References
- 63
- Citations by year
- 2026: 3 · 2025: 3 · 2024: 3 · 2023: 5 · 2022: 2
- FWCI
- 1.30
- Type
- journal article
- Language
- en
- Open access
- gold
- doi.org
- Licence
- Locations
- The Astrophysical Journal (publishedVersion, cc-by)
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Funders
- National Science Foundation (1715512, ACE210002, FB210003, AFB-170002, 1238877)
- Smithsonian Institution
- National Aeronautics and Space Administration (80NSSC18K0418, FB210003, NNX08AR22G, AST-1238877)
- Yale University
- Princeton University
- Johns Hopkins University
- Space Telescope Science Institute (NNX08AR22G)
- Durham University
- Ministry of Education, Culture, Sports, Science and Technology
- Academia Sinica
- Cabinet Office, Government of Japan
- National Central University
- National Astronomical Observatory of Japan
- High Energy Accelerator Research Organization
- Eötvös Loránd Tudományegyetem
- Agencia Nacional de Investigación y Desarrollo (NCN19-058, 1200495, ACE210002, 1190818, FB210003)
- Max-Planck-Institut für Astronomie
- Toray Science Foundation
- Smithsonian Astrophysical Observatory
- Science Mission Directorate (NNX08AR22G)
- Japan Society for the Promotion of Science
- Japan Science and Technology Agency
- Planetary Science Division (NNX08AR22G)
- Los Alamos National Laboratory
- Topics
- Physical Sciences › Physics and Astronomy › Astronomy and Astrophysics › Galaxies: Formation, Evolution, Phenomena
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
- Physical Sciences › Physics and Astronomy › Astronomy and Astrophysics › Stellar, planetary, and galactic studies
- Keywords
- galaxy morphology, Bayesian posterior estimation, effective radius, Monte Carlo dropout, spatial transformer network, uncertainty calibration, joint posterior distribution, Hyper Suprime-Cam, optical imaging
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review