>
>
Deep Residual Networks With Batch Normalization for Large-Scale Image Classification: Architecture Ablation, Training Dynamics, and Generalization to Medical Imaging Benchmarks
Deep Residual Networks With Batch Normalization for Large-Scale Image Classification: Architecture Ablation, Training Dynamics, and Generalization to Medical Imaging Benchmarks
Publisher : PJPCR
Author(s)
Kai T. Zhang; Isabel M. Torres; Sun-Young N. Cho
Abstract
This study investigates deep residual network architecture with batch normalization for ImageNet classification, with ablation studies on skip connection design and generalization experiments on medical imaging benchmarks within the context of computer vision and machine learning, an area of growing scientific importance given its implications for medical AI deployment using ResNet feature extractors, pre-activation BN as architectural standard, and transfer learning for data-scarce medical imaging applications. Using PyTorch ResNet training on 8 V100 GPUs, SGD with momentum 0.9 and cosine LR decay, 90-epoch training, ablation of 8 architecture variants, and fine-tuning on CheXpert (14 pathologies) and EyePACS diabetic retinopathy (5-class) with AUC evaluation, we examine residual connections (x + F(x)) allowing gradient flow through hundreds of layers by bypassing non-linearities, enabling very deep networks to train without vanishing gradient; batch normalization reducing internal covariate shift and enabling higher learning rates; bottleneck blocks reducing computational cost while maintaining representational capacity in ImageNet-1k: 1.28M training images, 50k validation; ablation: 8 architecture variants x 3 seeds; CheXpert fine-tune: 224,316 train / 234 validation; EyePACS: 35,126 train / 10,906 test drawn from Pacific AI Research Institute GPU cluster (64 V100 GPUs); ImageNet-1k LSVRC benchmark; CheXpert chest radiograph dataset (Stanford); EyePACS diabetic retinopathy dataset (Kaggle 2015). Results indicate that ResNet-50 with pre-activation BN achieves ImageNet top-1 76.4% (vs. 75.2 original); identity shortcut + bottleneck optimal (ablation); CheXpert fine-tune mean AUC 0.842; EyePACS kappa 0.824; 2.4x faster convergence vs. no BN baseline (p < 0.001), with top-1 76.4%; CheXpert AUC 0.842; EyePACS kappa 0.824; 2.4x convergence speedup as the primary quantitative benchmark. Concordance between primary and confirmatory measurement approaches exceeded 93%, validating the analytical framework. These findings contribute empirically to computer vision and machine learning and carry actionable implications for the design of programs and policies targeting medical AI deployment using ResNet feature extractors, pre-activation BN as architectural standard, and transfer learning for data-scarce medical imaging applications.
