Improving Probabilistic Models in Text Classification Via Active Learning

  • Mitchell Bosley
  • , Saki Kuzushima
  • , Ted Enamorado
  • , Yuki Shiraito

    Research output: Contribution to journalArticlepeer-review

    1 Scopus citations

    Abstract

    Social scientists often classify text documents to use the resulting labels as an outcome or a predictor in empirical research. Automated text classification has become a standard tool since it requires less human coding. However, scholars still need many human-labeled documents for training. To reduce labeling costs, we propose a new algorithm for text classification that combines a probabilistic model with active learning. The probabilistic model uses both labeled and unlabeled data, and active learning concentrates labeling efforts on difficult documents to classify. Our validation study shows that with few labeled data, the classification performance of our algorithm is comparable to state-of-the-art methods at a fraction of the computational cost. We replicate the results of two published articles with only a small fraction of the original labeled data used in those studies and provide open-source software to implement our method.

    Original languageEnglish
    Pages (from-to)985-1002
    Number of pages18
    JournalAmerican Political Science Review
    Volume119
    Issue number2
    DOIs
    StatePublished - May 1 2025

    Fingerprint

    Dive into the research topics of 'Improving Probabilistic Models in Text Classification Via Active Learning'. Together they form a unique fingerprint.

    Cite this