Demystify the Gravity Well in the Optimization Landscape (Student Abstract)

Authors

  • Jason Xiaotian Dou University of Pittsburgh
  • Runxue Bao University of Pittsburgh
  • Susan Song Carnegie Mellon University
  • Shuran Yang University of California, Berkeley
  • Yanfu Zhang University of Pittsburgh
  • Paul Pu Liang Carnegie Mellon University
  • Haiyi Harry Mao University of Pittsburgh

DOI:

https://doi.org/10.1609/aaai.v37i13.26961

Keywords:

Reinforcement Learning, Optimization, Representation Learning

Abstract

We provide both empirical and theoretical insights to demystify the gravity well phenomenon in the optimization landscape. We start from describe the problem setup and theoretical results (an escape time lower bound) of the Softmax Gravity Well (SGW) in the literature. Then we move toward the understanding of a recent observation called ASR gravity well. We provide an explanation of why normal distribution with high variance can lead to suboptimal plateaus from an energy function point of view. We also contribute to the empirical insights of curriculum learning by comparison of policy initialization by different normal distributions. Furthermore, we provide the ASR escape time lower bound to understand the ASR gravity well theoretically. Future work includes more specific modeling of the reward as a function of time and quantitative evaluation of normal distribution’s influence on policy initialization.

Downloads

Published

2024-07-15

How to Cite

Dou, J. X., Bao, R., Song, S., Yang, S., Zhang, Y., Liang, P. P., & Mao, H. H. (2024). Demystify the Gravity Well in the Optimization Landscape (Student Abstract). Proceedings of the AAAI Conference on Artificial Intelligence, 37(13), 16202-16203. https://doi.org/10.1609/aaai.v37i13.26961