Entropy-Guided Sampling of Flat Modes in Discrete Spaces

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Mohanty, Pinaki, Bhattacharya, Riddhiman, Zhang, Ruqi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910946808561664
author Mohanty, Pinaki
Bhattacharya, Riddhiman
Zhang, Ruqi
author_facet Mohanty, Pinaki
Bhattacharya, Riddhiman
Zhang, Ruqi
contents Sampling from flat modes in discrete spaces is a crucial yet underexplored problem. Flat modes represent robust solutions and have broad applications in combinatorial optimization and discrete generative modeling. However, existing sampling algorithms often overlook the mode volume and struggle to capture flat modes effectively. To address this limitation, we propose \emph{Entropic Discrete Langevin Proposal} (EDLP), which incorporates local entropy into the sampling process through a continuous auxiliary variable under a joint distribution. The local entropy term guides the discrete sampler toward flat modes with a small overhead. We provide non-asymptotic convergence guarantees for EDLP in locally log-concave discrete distributions. Empirically, our method consistently outperforms traditional approaches across tasks that require sampling from flat basins, including Bernoulli distribution, restricted Boltzmann machines, combinatorial optimization, and binary neural networks.
format Preprint
id arxiv_https___arxiv_org_abs_2505_02296
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Entropy-Guided Sampling of Flat Modes in Discrete Spaces
Mohanty, Pinaki
Bhattacharya, Riddhiman
Zhang, Ruqi
Machine Learning
Sampling from flat modes in discrete spaces is a crucial yet underexplored problem. Flat modes represent robust solutions and have broad applications in combinatorial optimization and discrete generative modeling. However, existing sampling algorithms often overlook the mode volume and struggle to capture flat modes effectively. To address this limitation, we propose \emph{Entropic Discrete Langevin Proposal} (EDLP), which incorporates local entropy into the sampling process through a continuous auxiliary variable under a joint distribution. The local entropy term guides the discrete sampler toward flat modes with a small overhead. We provide non-asymptotic convergence guarantees for EDLP in locally log-concave discrete distributions. Empirically, our method consistently outperforms traditional approaches across tasks that require sampling from flat basins, including Bernoulli distribution, restricted Boltzmann machines, combinatorial optimization, and binary neural networks.
title Entropy-Guided Sampling of Flat Modes in Discrete Spaces
topic Machine Learning
url https://arxiv.org/abs/2505.02296