Impact of batch size on stability in novel re-identification model

Abstract

This research introduces ConvReID-Net, a custom convolutional neural network (CNN) developed for person re-identification (Re-ID) focusing on the batch size dynamics and their effect on training stability. The model architecture consists of three convolutional layers, each followed by batch normalization, dropout, and max-pooling layers for regularization and feature extraction. The final layers include flattened and dense layers, optimizing the extracted features for classification. Evaluated over 50 epochs using early stopping, the network was trained on augmented image data to enhance robustness. The study specifically examines the influence of batch size on model performance, with batch size 64 yielding the best balance between validation accuracy (96.68%) and loss (0.1962). Smaller (batch size 32)and larger (batch size 128) configurations resulted in less stable performance, underscoring the importance of selecting an optimal batch size. These findings demonstrate ConvReID-Net’s potential for real-world Re-ID applications, especially in video surveillance systems. Future work will focus on further hyperparameter tuning and model improvements to enhance training efficiency and stability.