KL_divergence support in sim_top_word