Affiliations: [a] Departament of Computer Science, University of Brasilia, Brasilia, Brazil | [b] Institute of Mathematical and Computer Sciences, University of São Paulo, São Carlos, Brazil
Correspondence:
[*]
Corresponding author: Thiago de Paulo Faleiros, Departament of Computer Science, University of Brasilia, P.O. Box 4466, P.C. 70910-900, Brasilia, DF, Brazil. E-mail: thiagodepaulo@unb.br.
Abstract: Graph-based algorithms have aroused considerable interests in recent years by facilitating pattern recognition and learning via information propagation process through the graph. Here, we propose an unsupervised learning algorithm based on propagation on bipartite graph, referred to as Propagation in Bipartite Graph (PBG) algorithm. The contributions of this approach are threefold: 1) we present an iterative graph-based algorithm and a straight-forward bipartite representation for textual data, in which vertices represent documents and words, and edges between documents and words represent the occurrences of the words in the documents. Additionally, 2) we show that PBG is more flexible and easier to be adapted for different applications than the mathematical formalism of the generative models, and 3) we present a comprehensive evaluation and comparison of PBG to other topic extraction techniques. Here, we describe the strategy employed in PBG algorithm as a problem of maximization of similarity between latent vectors assigned to vertices and edges and demonstrate that the proposed strategy can be improved by assigning good initial values for the vectors. We notice that PBG can be parallelized by a simple adjustment in the algorithm. We also show that the proposed algorithm is competitive with LDA and NMF in the task of textual collection modelling, returning coherent topics, and in the dimensionality reduction task.
