A segmented topic model based on the two-parameter Poisson-Dirichlet process

  • Authors:
  • Lan Du;Wray Buntine;Huidong Jin

  • Affiliations:
  • Research School of Information Sciences and Engineering, The Australian National University, Canberra, Australia and NICTA, Canberra, Australia;Research School of Information Sciences and Engineering, The Australian National University, Canberra, Australia and NICTA, Canberra, Australia;Research School of Information Sciences and Engineering, The Australian National University, Canberra, Australia and CSIRO Mathematics, Informatics and Statistics, Canberra, Australia

  • Venue:
  • Machine Learning
  • Year:
  • 2010

Quantified Score

Hi-index 0.00

Visualization

Abstract

Documents come naturally with structure: a section contains paragraphs which itself contains sentences; a blog page contains a sequence of comments and links to related blogs. Structure, of course, implies something about shared topics. In this paper we take the simplest form of structure, a document consisting of multiple segments, as the basis for a new form of topic model. To make this computationally feasible, and to allow the form of collapsed Gibbs sampling that has worked well to date with topic models, we use the marginalized posterior of a two-parameter Poisson-Dirichlet process (or Pitman-Yor process) to handle the hierarchical modelling. Experiments using either paragraphs or sentences as segments show the method significantly outperforms standard topic models on either whole document or segment, and previous segmented models, based on the held-out perplexity measure.