Saved in:
Bibliographic Details
Main Authors: Xu, Weiran, Hirami, Kengo, Eguchi, Koji
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2502.09944
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910845019095040
author Xu, Weiran
Hirami, Kengo
Eguchi, Koji
author_facet Xu, Weiran
Hirami, Kengo
Eguchi, Koji
contents In our study, we propose a self-supervised neural topic model (NTM) that combines the power of NTMs and regularized self-supervised learning methods to improve performance. NTMs use neural networks to learn latent topics hidden behind the words in documents, enabling greater flexibility and the ability to estimate more coherent topics compared to traditional topic models. On the other hand, some self-supervised learning methods use a joint embedding architecture with two identical networks that produce similar representations for two augmented versions of the same input. Regularizations are applied to these representations to prevent collapse, which would otherwise result in the networks outputting constant or redundant representations for all inputs. Our model enhances topic quality by explicitly regularizing latent topic representations of anchor and positive samples. We also introduced an adversarial data augmentation method to replace the heuristic sampling method. We further developed several variation models including those on the basis of an NTM that incorporates contrastive learning with both positive and negative samples. Experimental results on three datasets showed that our models outperformed baselines and state-of-the-art models both quantitatively and qualitatively.
format Preprint
id arxiv_https___arxiv_org_abs_2502_09944
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Self-Supervised Learning for Neural Topic Models with Variance-Invariance-Covariance Regularization
Xu, Weiran
Hirami, Kengo
Eguchi, Koji
Machine Learning
Computation and Language
In our study, we propose a self-supervised neural topic model (NTM) that combines the power of NTMs and regularized self-supervised learning methods to improve performance. NTMs use neural networks to learn latent topics hidden behind the words in documents, enabling greater flexibility and the ability to estimate more coherent topics compared to traditional topic models. On the other hand, some self-supervised learning methods use a joint embedding architecture with two identical networks that produce similar representations for two augmented versions of the same input. Regularizations are applied to these representations to prevent collapse, which would otherwise result in the networks outputting constant or redundant representations for all inputs. Our model enhances topic quality by explicitly regularizing latent topic representations of anchor and positive samples. We also introduced an adversarial data augmentation method to replace the heuristic sampling method. We further developed several variation models including those on the basis of an NTM that incorporates contrastive learning with both positive and negative samples. Experimental results on three datasets showed that our models outperformed baselines and state-of-the-art models both quantitatively and qualitatively.
title Self-Supervised Learning for Neural Topic Models with Variance-Invariance-Covariance Regularization
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2502.09944