Back

Usefulness of pystan and numpyro in Bayesian item response theory

Nishio, M.; Ota, E.; Matsuo, H.; Matsunaga, T.; Miyazaki, A.; Murakami, T.

2023-03-29 health informatics
10.1101/2023.03.29.23287903 medRxiv
Show abstract

PurposeThe purpose of this study is to compare two libraries dedicated to Markov chain Monte Carlo method: pystan and numpyro. Materials and methodsBayesian item response theory (IRT), 1PL-IRT and 2PL-IRT, were implemented with pystan and numpyro. Then, the Bayesian 1PL-IRT and 2PL-IRT were applied to two types of medical data obtained from a published paper. The same prior distributions of latent parameters were used in both pystan and numpyro. Estimation results of latent parameters of 1PL-IRT and 2PL-IRT were compared between pystan and numpyro. Additionally, the computational cost of Markov chain Monte Carlo method was compared between the two libraries. To evaluate the computational cost of IRT models, simulation data were generated from the medical data and numpyro. ResultsFor all the combinations of IRT types (1PL-IRT or 2PL-IRT) and medical data types, the mean and standard deviation of the estimated latent parameters were in good agreement between pystan and numpyro. In most cases, the sampling time using Markov chain Monte Carlo method was shorter in numpyro than that in pystan. When the large-sized simulation data were used, numpyro with a graphics processing unit was useful for reducing the sampling time. ConclusionNumpyro and pystan were useful for applying the Bayesian 1PL-IRT and 2PL-IRT.

Matching journals

The top 9 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.