Scalable Stellar Parameter Inference Using Python-based LASP: From CPU Optimization to GPU Acceleration

doi:10.3847/1538-4357/ae1446

Return

Scalable Stellar Parameter Inference Using Python-based LASP: From CPU Optimization to GPU Acceleration

2025-12-31

DOI：10.3847/1538-4357/ae1446

Original

Save

View PDF

Abstract

En 中文

To enhance the efficiency, scalability, and cross-survey applicability of stellar parameter inference in large spectroscopic datasets, we present a modular, parallelized Python framework with automated error estimation, built on the LAMOST Atmospheric Parameter Pipeline (LASP) originally implemented in IDL. Rather than a direct code translation, this framework refactors LASP with two complementary modules: LASP-CurveFit, a new implementation of the LASP fitting procedure that runs on a CPU, preserving legacy logic while improving data I/O and multithreaded execution efficiency; and LASP-Adam-GPU, a GPU-accelerated method that introduces grouped optimization by constructing a joint residual function over multiple observed and model spectra, enabling high-throughput parameter inference across tens of millions of spectra. Applied to 10 million LAMOST spectra, the framework reduces runtime from 84 to 48 hr on the same CPU platform and to 7 hr on an NVIDIA A100 GPU, while producing results consistent with those from the original pipeline. The inferred errors agree well with the parameter variations from repeat observations of the same target (excluding radial velocities), while the official empirical errors used in LASP are more conservative. When applied to DESI DR1, our effective temperatures and surface gravities agree better with APOGEE than those from the DESI pipeline, particularly for cool giants, while the latter performs slightly better in radial velocity and metallicity. These results suggest that the framework delivers reliable accuracy, efficiency, and transferability, offering a practical approach to parameter inference in large spectroscopic surveys. The code and DESI-based catalog are available via DOI: 10.12149/101679 and DOI: 10.12149/101675, respectively.

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

The Astrophysical Journal

IF:

Papers:

1.5K

Citations:

Organization

University of Tokyo

Scholars:

7.1W

Papers: 6.5W

Citations: 2.2K

Dezhou University

Scholars:

1.9K

Papers: 1.1K

Citations: 1.4K

university of hertfordshire

Scholars:

669

Papers: 432

Citations: 0

Chinese Academy of Sciences

Scholars:

3.9W

Papers: 1.5W

Citations: 58.4W

View more organizations

Cited Papers

Array programming with NumPy

Nature

IF0

2020-09-16

OAAI

Charles R. Harris; K. Jarrod Millman; Stéfan J. van der Walt; Ralf Gommers; Pauli Virtanen; David Cournapeau; Eric Wieser; Julian Taylor; Sebastian Berg; Nathaniel J. Smith; Robert Kern; Matti Picus; Stephan Hoyer; Marten H. van Kerkwijk; Matthew Brett; Allan Haldane; Jaime Fernández del Río; Mark Wiebe; Pearu Peterson; Pierre Gérard-Marchant; Kevin Sheppard; Tyler Reddy; Warren Weckesser; Hameer Abbasi; Christoph Gohlke; Travis E. Oliphant

Save

The Payne: Self-consistent ab initio Fitting of Stellar Spectra

ASTROPHYSICAL JOURNAL

IF5.4

2019-07-08

153

OAAI

Ting, Yuan-Sen; Conroy, Charlie; Rix, Hans-Walter; Cargile, Phillip

Save

Estimation of Stellar Atmospheric Parameters with Light Gradient Boosting Machine Algorithm and Principal Component Analysis

ASTRONOMICAL JOURNAL

IF5.1

2022-03-08

OAAI

Liang, Junchao; Bu, Yude; Tan, Kefeng; Pan, Jingchang; Yi, Zhenping; Kong, Xiaoming; Fan, Zhou

Save

The Robotic Multiobject Focal Plane System of the Dark Energy Spectroscopic Instrument (DESI)

ASTRONOMICAL JOURNAL

IF5.1

2022-12-07

OAAI

Silber, Joseph Harry; Fagrelius, Parker; Fanning, Kevin; Schubnell, Michael; Aguilar, Jessica Nicole; Ahlen, Steven; Ameel, Jon; Ballester, Otger; Baltay, Charles; Bebek, Chris; Beard, Dominic Benton; Besuner, Robert; Cardiel-Sas, Laia; Casas, Ricard; Castander, Francisco Javier; Claybaugh, Todd; Dobson, Carl; Duan, Yutong; Dunlop, Patrick; Edelstein, Jerry; Emmet, William T.; Elliott, Ann; Evatt, Matthew; Gershkovich, Irena; Guy, Julien; Harris, Stu; Heetderks, Henry; Heetderks, Ian; Honscheid, Klaus; Maria Illa, Jose; Jelinsky, Patrick; Jelinsky, Sharon R.; Jimenez, Jorge; Karcher, Armin; Kent, Stephen; Kirkby, David; Kneib, Jean-Paul; Lambert, Andrew; Lampton, Mike; Leitner, Daniela; Levi, Michael; McCauley, Jeremy; Meisner, Aaron; Miller, Timothy N.; Miquel, Ramon; Mundet, Julia; Poppett, Claire; Rabinowitz, David; Reil, Kevin; Roman, David; Schlegel, David; Serrano, Santiago; Van Shourt, William; Sprayberry, David; Tarle, Gregory; Tie, Suk Sien; Weaverdyck, Curtis; Zhang, Kai; Azzaro, Marco; Bailey, Stephen; Becerril, Santiago; Blackwell, Tami; Bouri, Mohamed; Brooks, David; Buckley-Geer, Elizabeth; Penate Castro, Jose; Derwent, Mark; Dey, Arjun; Dhungana, Govinda; Doel, Peter; Eisenstein, Daniel J.; Fahim, Nasib; Garcia-Bellido, Juan; Gaztanaga, Enrique; Gontcho, Satya Gontcho A.; Gutierrez, Gaston; Horler, Philipp; Kehoe, Robert; Kisner, Theodore; Kremin, Anthony; Kronig, Luzius; Landriau, Martin; Le Guillou, Laurent; Martini, Paul; Moustakas, John; Palanque-Delabrouille, Nathalie; Peng, Xiyan; Percival, Will; Prada, Francisco; Allende Prieto, Carlos; Gonzalez de Rivera, Guillermo; Sanchez, Eusebio; Sanchez, Justo; Sharples, Ray; Soares-Santos, Marcelle; Schlafly, Edward; Weaver, Benjamin Alan; Zhou, Zhimin; Zhu, Yaling; Zou, Hu

Save

Numerical Optimization

Springer Series in Operations Research and Financial Engineering

IF0

2006-01-01

PREAI

Save

Gaia Data Release 3 Analysis of RVS spectra using the General Stellar Parametriser from spectroscopy

ASTRONOMY & ASTROPHYSICS

IF5.8

2023-06-16

121

OAAI

Recio-Blanco, A.; De laverny, P.; Palicio, P. A.; Kordopatis, G.; Alvarez, M. A.; Schultheis, M.; Contursi, G.; Zhao, H.; Elipe, G. Torralba; Ordenovic, C.; Manteiga, M.; Dafonte, C.; Oreshina-Slezak, I.; Bijaoui, A.; Fremat, Y.; Seabroke, G.; Pailler, F.; Spitoni, E.; Poggio, E.; Creevey, O. L.; Aramburu, A. Abreu; Accart, S.; Andrae, R.; Bailer-Jones, C. A. L.; Bellas-Velidis, I.; Brouillet, N.; Brugaletta, E.; Burlacu, A.; Carballo, R.; Casamiquela, L.; Chiavassa, A.; Cooper, W. J.; Dapergolas, A.; Delchambre, L.; Dharmawardena, T. E.; Drimmel, R.; Edvardsson, B.; Fouesneau, M.; Garabato, D.; Garcia-Lario, P.; Garcia-Torres, M.; Gavel, A.; Gomez, A.; Gonzalez-Santamaria, I.; Hatzidimitriou, D.; Heiter, U.; Piccolo, A. Jean-Antoine; Kontizas, M.; Korn, A. J.; Lanzafame, A. C.; Lebreton, Y.; Le Fustec, Y.; Licata, E. L.; Lindstrom, H. E. P.; Livanou, E.; Lobel, A.; Lorca, A.; Romeo, A. Magdaleno; Marocco, F.; Marshall, D. J.; Mary, N.; Nicolas, C.; Pallas-Quintela, L.; Panem, C.; Pichon, B.; Riclet, F.; Robin, C.; Rybizki, J.; Santovena, R.; Silvelo, A.; Smart, R. L.; Sarro, L. M.; Sordo, R.; Soubiran, C.; Suveges, M.; Ulla, A.; Vallenari, A.; Zorec, J.; Utrilla, E.; Bakker, J.

Save

Gaia Data Release 3 Summary of the content and survey properties

ASTRONOMY & ASTROPHYSICS

IF5.8

2023-06-16

227

OAAI

Vallenari, A.; Brown, A. G. A.; Prusti, T.; de Bruijne, J. H. J.; Arenou, F.; Babusiaux, C.; Biermann, M.; Creevey, O. L.; Ducourant, C.; Evans, D. W.; Eyer, L.; Guerra, R.; Hutton, A.; Jordi, C.; Klioner, S. A.; Lammers, U. L.; Lindegren, L.; Luri, X.; Mignard, F.; Panem, C.; Pourbaix, D.; Randich, S.; Sartoretti, P.; Soubiran, C.; Tanga, P.; Walton, N. A.; Bailer-Jones, C. A. L.; Bastian, U.; Drimmel, R.; Jansen, F.; Katz, D.; Lattanzi, M. G.; vanLeeuwen, F.; Bakker, J.; Cacciari, C.; Castaneda, J.; De Angeli, F.; Fabricius, C.; Fouesneau, M.; Fremat, Y.; Galluccio, L.; Guerrier, A.; Heiter, U.; Masana, E.; Messineo, R.; Mowlavi, N.; Nicolas, C.; Nienartowicz, K.; Pailler, F.; Panuzzo, P.; Riclet, F.; Roux, W.; Seabroke, G. M.; Sordo, R.; Thevenin, F.; Gracia-Abril, G.; Portell, J.; Teyssier, D.; Altmann, M.; Andrae, R.; Audard, M.; Bellas-Velidis, I.; Benson, K.; Berthier, J.; Blomme, R.; Burgess, P. W.; Busonero, D.; Busso, G.; Canovas, H.; Carry, B.; Cellino, A.; Cheek, N.; Clementini, G.; Damerdji, Y.; Davidson, M.; Deteodoro, P.; Campos, M. Nunez; Delchambre, L.; Dell'Oro, A.; Esquej, P.; Fernandez-Hernandez, J.; Fraile, E.; Garabato, D.; Garcia-Lario, P.; Gosset, E.; Haigron, R.; Halbwachs, J. -L.; Hambly, N. C.; Harrison, D. L.; Hernandez, J.; Hestroffer, D.; Hodgkin, S. T.; Holl, B.; Janssen, K.; deFombelle, G. Jevardat; Jordan, S.; Krone-Martins, A.; Lanzafame, A. C.; Loeffler, W.; Marchal, O.; Marrese, P. M.; Moitinho, A.; Muinonen, K.; Osborne, P.; Pancino, E.; Pauwels, T.; Recio-Blanco, A.; Reyle, C.; Riello, M.; Rimoldini, L.; Roegiers, T.; Rybizki, J.; Sarro, L. M.; Siopis, C.; Smith, M.; Sozzetti, A.; Utrilla, E.; vanLeeuwen, M.; Abbas, U.; Abraham, P.; Aramburu, A. Abreu; Aerts, C.; Aguado, J. J.; Ajaj, M.; Aldea-Montero, F.; Altavilla, G.; Alvarez, M. A.; Alves, J.; Anders, F.; Anderson, R. I.; Varela, E. Anglada; Antoja, T.; Baines, D.; Baker, S. G.; Balaguer-Nunez, L.; Balbinot, E.; Balog, Z.; Barache, C.; Barbato, D.; Barros, M.; Barstow, M. A.; Bartolome, S.; Bassilana, J. -L.; Bauchet, N.; Becciani, U.; Bellazzini, M.; Berihuete, A.; Bernet, M.; Bertone, S.; Bianchi, L.; Binnenfeld, A.; Blanco-Cuaresma, S.; Blazere, A.; Boch, T.; Bombrun, A.; Bossini, D.; Bouquillon, S.; Bragaglia, A.; Bramante, L.; Breedt, E.; Bressan, A.; Brouillet, N.; Brugaletta, E.; Bucciarelli, B.; Burlacu, A.; Butkevich, A. G.; Buzzi, R.; Caffau, E.; Cancelliere, R.; Cantat-Gaudin, T.; Carballo, R.; Carlucci, T.; Carnerero, M. I.; Carrasco, J. M.; Casamiquela, L.; Castellani, M.; Castro-Ginard, A.; Chaoul, L.; Charlot, P.; Chemin, L.; Chiaramida, V.; Chiavassa, A.; Chornay, N.; Comoretto, G.; Contursi, G.; Cooper, W. J.; Cornez, T.; Cowell, S.; Crifo, F.; Cropper, M.; Crosta, M.; Crowley, C.; Dafonte, C.; Dapergolas, A.; David, M.; David, P.; deLaverny, P.; DeLuise, F.; DeMarch, R.; DeRidder, J.; deSouza, R.; deTorres, A.; del Peloso, E. F.; delPozo, E.; Delbo, M.; Delgado, A.; Delisle, J. -B.; Demouchy, C.; Dharmawardena, T. E.; Di Matteo, P.; Diakite, S.; Diener, C.; Distefano, E.; Dolding, C.; Edvardsson, B.; Enke, H.; Fabre, C.; Fabrizio, M.; Faigler, S.; Fedorets, G.; Fernique, P.; Fienga, A.; Figueras, F.; Fournier, Y.; Fouron, C.; Fragkoudi, F.; Gai, M.; Garcia-Gutierrez, A.; Garcia-Reinaldos, M.; Garcia-Torres, M.; Garofalo, A.; Gavel, A.; Gavras, P.; Gerlach, E.; Geyer, R.; Giacobbe, P.; Gilmore, G.; Girona, S.; Rida, G. Giu Ff; Gomel, R.; Gomez, A.; Gonzalez-Nunez, J.; Gonzalez-Santamaria, I.; Gonzalez-Vidal, J. J.; Granvik, M.; Guillout, P.; Guiraud, J.; Gutierrez-Sanchez, R.; Guy, L. P.; Hatzidimitriou, D.; Hauser, M.; Haywood, M.; Helmer, A.; Helmi, A.; Sarmiento, M. H.; Hidalgo, S. L.; Hilger, T.; Hladczuk, N.; Hobbs, D.; Holland, G.; Huckle, H. E.; Jardine, K.; Jasniewicz, G.; Piccolo, A. Jean-Antoine; Jimenez-Arranz, O.; Jorissen, A.; Campillo, J. Juaristi; Julbe, F.; Karbevska, L.; Kervella, P.; Khanna, S.; Kontizas, M.; Kordopatis, G.; Korn, A. J.; Kospal, A.; Kostrzewa-Rutkowska, Z.; Kruszynska, K.; Kun, M.; Laizeau, P.; Lambert, S.; Lanza, A. F.; Lasne, Y.; LeCampion, J. -F.; Lebreton, Y.; Lebzelter, T.; Leccia, S.; Leclerc, N.; Lecoeur-Taibi, I.; Liao, S.; Licata, E. L.; Lindstrom, H. E. P.; Lister, T. A.; Livanou, E.; Lobel, A.; Lorca, A.; Loup, C.; MadreroPardo, P.; MagdalenoRomeo, A.; Managau, S.; Mann, R. G.; Manteiga, M.; Marchant, J. M.; Marconi, M.; Marcos, J.; MarcosSantos, M. M. S.; MarinPina, D.; Marinoni, S.; Marocco, F.; Marshall, D. J.; Polo, L. Martin; Martin-Fleitas, J. M.; Marton, G.; Mary, N.; Masip, A.; Massari, D.; Mastrobuono-Battisti, A.; Mazeh, T.; McMillan, P. J.; Messina, S.; Michalik, D.; Millar, N. R.; Mints, A.; Molina, D.; Molinaro, R.; Molnar, L.; Monari, G.; Monguio, M.; Montegriffo, P.; Montero, A.; Mor, R.; Mora, A.; Morbidelli, R.; Morel, T.; Morris, D.; Muraveva, T.; Murphy, C. P.; Musella, I.; Nagy, Z.; Noval, L.; Ocana, F.; Ogden, A.; Ordenovic, C.; Osinde, J. O.; Pagani, C.; Pagano, I.; Palaversa, L.; Palicio, P. A.; Pallas-Quintela, L.; Panahi, A.; Payne-Wardenaar, S.; Esteller, X. Penalosa; Penttila, A.; Pichon, B.; Piersimoni, A. M.; Pineau, F. -X.; Plachy, E.; Plum, G.; Poggio, E.; Prsa, A.; Pulone, L.; Racero, E.; Ragaini, S.; Rainer, M.; Raiteri, C. M.; Rambaux, N.; Ramos, P.; Ramos-Lerate, M.; ReFiorentin, P.; Regibo, S.; Richards, P. J.; Diaz, C. Rios; Ripepi, V.; Riva, A.; Rix, H. -W.; Rixon, G.; Robichon, N.; Robin, A. C.; Robin, C.; Roelens, M.; Rogues, H. R. O.; Rohrbasser, L.; Romero-Gomez, M.; Rowell, N.; Royer, F.; Mieres, D. Ruz; Rybicki, K. A.; Sadowski, G.; Nunez, A. Saez; Selles, A. Sagrista; Sahlmann, J.; Salguero, E.; Samaras, N.; Gimenez, V. Sanchez; Sanna, N.; Santovena, R.; Sarasso, M.; Schultheis, M.; Sciacca, E.; Segol, M.; Segovia, J. C.; Segransan, D.; Semeux, D.; Shahaf, S.; Siddiqui, H. I.; Siebert, A.; Siltala, L.; Silvelo, A.; Slezak, E.; Slezak, I.; Smart, R. L.; Snaith, O. N.; Solano, E.; Solitro, F.; Souami, D.; Souchay, J.; Spagna, A.; Spina, L.; Spoto, F.; Steele, I. A.; Steidelmueller, H.; Stephenson, C. A.; Suveges, M.; Surdej, J.; Szabados, L.; Szegedi-Elek, E.; Taris, F.; Taylor, M. B.; Teixeira, R.; Tolomei, L.; Tonello, N.; Torra, F.; Torra, J.; Elipe, G. Torralba; Trabucchi, M.; Tsounis, A. T.; Turon, C.; Ulla, A.; Unger, N.; Vaillant, M. V.; van Dillen, E.; van Reeven, W.; Vanel, O.; Vecchiato, A.; Viala, Y.; Vicente, D.; Voutsinas, S.; Weiler, M.; Wevers, T.; Wyrzykowski, L.; Yoldas, A.; Yvard, P.; Zhao, H.; Zorec, J.; Zucker, S.; Zwitter, T.

Save

Automatic stellar spectral parameterization pipeline for LAMOST survey

Proceedings of the International Astronomical Union

IF0

2015-07-01

OAAI

Yue Wu; Bing Du; Ali Luo; Yongheng Zhao; Hailong Yuan

Save

A unified architecture for natural language processing

Proceedings of the 25th international conference on Machine learning - ICML '08

IF0

2008-01-01