Research

Paper

TESTING March 18, 2026

State-dependent temperature control in Langevin diffusions using numerical exploratory Hamiltonian-Jacobi-Bellman equations

Authors

Taorui Wang, Xun Li, Gu Wang, Zhongqiang Zhang

Abstract

Choosing how much noise to add in Langevin dynamics is essential for making these algorithms effective in challenging optimization problems. One promising approach is to determine this noise by solving Hamilton-Jacobi-Bellman (HJB) equations and their exploratory variants. Though these ideas have been demonstrated to work well in one dimension, extension to high-dimensional minimization has been limited by two unresolved numerical challenges: setting reliable control bounds and stably computing the second-order information (Hessians) required by the equations. These issues and the broader impact of HJB parameters have not been systematically examined. This work provides the first such investigation. We introduce principled control bounds and develop a physics-informed neural network framework that embeds the structure of exploratory HJB equations directly into training, stabilizing computation, and enabling accurate estimation of state-dependent noise in high-dimensional problems. Numerical experiments demonstrate that the resulting method remains robust and effective well beyond low-dimensional test cases.

Metadata

arXiv ID: 2603.17934
Provider: ARXIV
Primary Category: math.NA
Published: 2026-03-18
Fetched: 2026-03-19 06:01

Related papers

Raw Data (Debug)
{
  "raw_xml": "<entry>\n    <id>http://arxiv.org/abs/2603.17934v1</id>\n    <title>State-dependent temperature control in Langevin diffusions using numerical exploratory Hamiltonian-Jacobi-Bellman equations</title>\n    <updated>2026-03-18T17:09:04Z</updated>\n    <link href='https://arxiv.org/abs/2603.17934v1' rel='alternate' type='text/html'/>\n    <link href='https://arxiv.org/pdf/2603.17934v1' rel='related' title='pdf' type='application/pdf'/>\n    <summary>Choosing how much noise to add in Langevin dynamics is essential for making these algorithms effective in challenging optimization problems. One promising approach is to determine this noise by solving Hamilton-Jacobi-Bellman (HJB) equations and their exploratory variants. Though these ideas have been demonstrated to work well in one dimension, extension to high-dimensional minimization has been limited by two unresolved numerical challenges: setting reliable control bounds and stably computing the second-order information (Hessians) required by the equations. These issues and the broader impact of HJB parameters have not been systematically examined. This work provides the first such investigation. We introduce principled control bounds and develop a physics-informed neural network framework that embeds the structure of exploratory HJB equations directly into training, stabilizing computation, and enabling accurate estimation of state-dependent noise in high-dimensional problems. Numerical experiments demonstrate that the resulting method remains robust and effective well beyond low-dimensional test cases.</summary>\n    <category scheme='http://arxiv.org/schemas/atom' term='math.NA'/>\n    <category scheme='http://arxiv.org/schemas/atom' term='math.OC'/>\n    <published>2026-03-18T17:09:04Z</published>\n    <arxiv:primary_category term='math.NA'/>\n    <author>\n      <name>Taorui Wang</name>\n    </author>\n    <author>\n      <name>Xun Li</name>\n    </author>\n    <author>\n      <name>Gu Wang</name>\n    </author>\n    <author>\n      <name>Zhongqiang Zhang</name>\n    </author>\n  </entry>"
}