Display options
Cite
Zhao W, Chen JJ, Perkins R, et al. A heuristic approach to determine an appropriate number of topics in topic modeling. BMC Bioinformatics. 2015;16:S8doi: 10.1186/1471-2105-16-S13-S8.
Zhao, W., Chen, J. J., Perkins, R., Liu, Z., Ge, W., Ding, Y., & Zou, W. (2015). A heuristic approach to determine an appropriate number of topics in topic modeling. BMC bioinformatics, 16S8. https://doi.org/10.1186/1471-2105-16-S13-S8
Zhao, Weizhong, et al. "A heuristic approach to determine an appropriate number of topics in topic modeling." BMC bioinformatics vol. 16 (2015): S8. doi: https://doi.org/10.1186/1471-2105-16-S13-S8
Zhao W, Chen JJ, Perkins R, Liu Z, Ge W, Ding Y, Zou W. A heuristic approach to determine an appropriate number of topics in topic modeling. BMC Bioinformatics. 2015;16:S8. doi: 10.1186/1471-2105-16-S13-S8. Epub 2015 Sep 25. PMID: 26424364; PMCID: PMC4597325.
Copy Download .nbib Format: NLM
Share it on

BMC Bioinformatics. 2015;16:S8. doi: 10.1186/1471-2105-16-S13-S8. Epub 2015 Sep 25.

A heuristic approach to determine an appropriate number of topics in topic modeling.

BMC bioinformatics

Weizhong Zhao, James J Chen, Roger Perkins, Zhichao Liu, Weigong Ge, Yijun Ding, Wen Zou

PMID: 26424364 PMCID: PMC4597325 DOI: 10.1186/1471-2105-16-S13-S8

Abstract

BACKGROUND: Topic modelling is an active research field in machine learning. While mainly used to build models from unstructured textual data, it offers an effective means of data mining where samples represent documents, and different biological endpoints or omics data represent words. Latent Dirichlet Allocation (LDA) is the most commonly used topic modelling method across a wide number of technical fields. However, model development can be arduous and tedious, and requires burdensome and systematic sensitivity studies in order to find the best set of model parameters. Often, time-consuming subjective evaluations are needed to compare models. Currently, research has yielded no easy way to choose the proper number of topics in a model beyond a major iterative approach.

METHODS AND RESULTS: Based on analysis of variation of statistical perplexity during topic modelling, a heuristic approach is proposed in this study to estimate the most appropriate number of topics. Specifically, the rate of perplexity change (RPC) as a function of numbers of topics is proposed as a suitable selector. We test the stability and effectiveness of the proposed method for three markedly different types of grounded-truth datasets: Salmonella next generation sequencing, pharmacological side effects, and textual abstracts on computational biology and bioinformatics (TCBB) from PubMed.

CONCLUSION: The proposed RPC-based method is demonstrated to choose the best number of topics in three numerical experiments of widely different data types, and for databases of very different sizes. The work required was markedly less arduous than if full systematic sensitivity studies had been carried out with number of topics as a parameter. We understand that additional investigation is needed to substantiate the method's theoretical basis, and to establish its generalizability in terms of dataset characteristics.

References

  1. IEEE/ACM Trans Comput Biol Bioinform. 2005 Apr-Jun;2(2):143-56 - PubMed
  2. Mol Syst Biol. 2010;6:343 - PubMed
  3. J Mach Learn Res. 2008 Sep;9:1981-2014 - PubMed
  4. BMC Bioinformatics. 2014;15 Suppl 11:S11 - PubMed
  5. Proc Natl Acad Sci U S A. 2004 Apr 6;101 Suppl 1:5228-35 - PubMed
  6. Bioinformatics. 2011 Jul 1;27(13):i61-8 - PubMed
  7. Bioinformatics. 2010 Jun 15;26(12):i7-12 - PubMed
  8. Nucleic Acids Res. 2004;32(5):1792-7 - PubMed

MeSH terms

Publication Types