TAMI¶

Module Contents¶

InfoCTM: A Mutual Information Maximization Perspective of Cross-lingual Topic Modeling. AAAI 2023

class TAMI(temperature, weight_MI, pos_threshold, trans_e2c, pretrain_word_embeddings_en, pretrain_word_embeddings_cn)¶

Bases: torch.nn.Module

InfoCTM: A Mutual Information Maximization Perspective of Cross-lingual Topic Modeling. AAAI 2023

Xiaobao Wu, Xinshuai Dong, Thong Nguyen, Chaoqun Liu, Liangming Pan, Anh Tuan Luu

temperature¶

weight_MI¶

pos_threshold¶

pretrain_word_embeddings_en¶

pretrain_word_embeddings_cn¶

trans_e2c¶

trans_c2e¶

pos_trans_mask_en¶

pos_trans_mask_cn¶

neg_trans_mask_en¶

neg_trans_mask_cn¶

build_CVL_mask(embeddings)¶

translation_mask(mask, trans_dict_matrix)¶

compute_pos_neg(pretrain_word_embeddings_en, pretrain_word_embeddings_cn, trans_e2c, trans_c2e)¶

MutualInfo(anchor_feature, contrast_feature, mask, neg_mask)¶

forward(fea_en, fea_cn)¶