LGAISep 14, 2024

Cross-Entropy Optimization for Hyperparameter Optimization in Stochastic Gradient-based Approaches to Train Deep Neural Networks

arXiv:2409.09240v12 citationsh-index: 1
Originality Synthesis-oriented
AI Analysis

This work addresses the challenge of tuning hyperparameters for deep neural networks, which can impact convergence and generalization, but it appears incremental as it builds on existing cross-entropy and EM methods without claiming major breakthroughs.

The paper tackles hyperparameter optimization in stochastic gradient-based deep learning by proposing a cross-entropy optimization method, analyzing it within an expectation maximization framework, and demonstrating its applicability to various optimization problems in machine learning.

In this paper, we present a cross-entropy optimization method for hyperparameter optimization in stochastic gradient-based approaches to train deep neural networks. The value of a hyperparameter of a learning algorithm often has great impact on the performance of a model such as the convergence speed, the generalization performance metrics, etc. While in some cases the hyperparameters of a learning algorithm can be part of learning parameters, in other scenarios the hyperparameters of a stochastic optimization algorithm such as Adam [5] and its variants are either fixed as a constant or are kept changing in a monotonic way over time. We give an in-depth analysis of the presented method in the framework of expectation maximization (EM). The presented algorithm of cross-entropy optimization for hyperparameter optimization of a learning algorithm (CEHPO) can be equally applicable to other areas of optimization problems in deep learning. We hope that the presented methods can provide different perspectives and offer some insights for optimization problems in different areas of machine learning and beyond.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes