adjidieng / DETM

MIT License
130 stars 39 forks source link

Loss decreases, KL_alpha and KL_theta increase #6

Closed mona-timmermann closed 3 years ago

mona-timmermann commented 3 years ago

On my own data and when trying to reproduce the results for the UN dataset, I observe rising KL_alpha and KL_theta using Adam optimizer.

I have tried different settings for Adam and other optimizers on my data, but have not found a solution. I always observe this issue with Adam (I have tried different parameter settings) and other optimizers like asgd don't seem to work at all (very bad topics, no improvements in loss reduction,...). Anybody else with that issue and an idea how to solve it?

Here is an example from training D-ETM on my own dataset:

Bildschirmfoto 2020-10-06 um 13 39 23