Skip to content

Javigsv/LDA_AdML

Repository files navigation

LDA_AdML

In this project we re-implemented the Latent Dirichlet Allocation (LDA) as described in the 2003 Blei et. al. paper. The model was originally created for topic modelling in large corpora in order to find low-dimensional representations of documents while retaining statistical relationships useful for tasks relating to information retrieval.

The model is trained with an Variational Inference - Expectation Maximization (VI-EM) alogrithm. Read more in the final report.

This was the final project of the course DD2434 Machine Learning, Advanced Course during the fall of 2020.

Results

Here is a small sample of the results are presented.

Topics over time

Using the model parameters one can approximate the "popularity" of topics over time. This can be seen for three topics in the graph below:

Topics over time

Is this graph accurate? Let's have a look at what happened during that time period.

Regarding climate, the peak is subjected around January 2017, which was a period when the Trump administration actively worked to erase ex-president Barack Obama’s climate initiatives. The graph of the Russian Election Allegations which gained large media traction after members of the United States Congress publicly disclosed a potential russian interference in the 2016 election. The media attention continued to increase as more US intelligence agencies confirmed the suspicions and the Office of the Director of National Intelligence published its detailed report in January 2017. Notably, the last peak from 2017-05 to 2017-07 coincides with the news surrounding Trump’s dismissal of James Comey, the FBI director, in may 9th, 2017.

An example article

Looking at topics with a high number of expected words belonging to a certain article and also making sure that the words have a high probability of belonging to the given topics (>0.9) one can find the "main" topics of articles. Below is an example of this.

Example article

About

LDA project (Large VI) for DD2434 Machine Learning Advanced Course

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •  

Languages