Enhancing Deep Neural Network Classification Performance Through Novel Weight Initialization: t-SNE Supported Walsh Matrix Approach
Deep Neural Networks, as a subset of AI, outper-form in understanding complex relationships. The key to this success lies in the network's ability to adapt to problem-specific nuances. During model training, the network dynamically optimizes its weights by updating them during backpropagation while trying to minimize the value of the loss function. Throughout this process, the shaping of model weights is crucially linked to how they were initialized. In this study, we introduce the auxiliary network model, called Sup-Walsh (Support Walsh), which reorganizes weights to enhance class boundaries. We tested our approach on three publicly available datasets using popular classification models. For instance, when using AlexNet [1] on the MNIST dataset [2], integrating Sup-Walsh led to a significant increase in accuracy after first epoch from 14.61% to 78.99%. Similarly, GoogleNet [3] on the FashionMNIST dataset [4] showed a notable 31.61% accuracy difference between configurations without and with Sup-Walsh after first epoch. Across nearly all experiments, our proposed method consistently outperformed existing approaches, demonstrating its potential to improve classification accuracy. Code availability: Code is available at Efficient-Weight-Initializer.
Authors
- Ankit Agrawal (ORCID: https://orcid.org/0000-0002-5519-0302)
- Vishu Gupta (ORCID: https://orcid.org/0000-0002-4931-7194)
- Wei‐keng Liao (ORCID: https://orcid.org/0009-0008-9411-2543)
- Yuwei Mao (ORCID: https://orcid.org/0000-0003-2438-1998)
- Alok Choudhary (ORCID: https://orcid.org/0000-0001-8152-6319)
- Alec Peltekian
- Muhammed Nur Talha Kilic (ORCID: https://orcid.org/0000-0001-5879-8279)
- Kewei Wang
Institutions
- Northwestern University (US)
Publication Details
- Published
- 2024-12-18
- DOI
- https://doi.org/10.1109/icmla61862.2024.00153
- Primary Topic
- Face and Expression Recognition
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Science Foundation
- U.S. Department of Energy
- U.S. Department of Commerce