We study here a fixed mini-batch gradient decent (FMGD) algorithm to solve optimization problems with massive datasets. In FMGD, the whole sample is split into multiple non-overlapping partitions. Once the partitions are formed, they are then fixed throughout the rest of the algorithm. For convenience, we refer to the fixed partitions as fixed mini-batches. Then for each computation iteration, the gradients are sequentially calculated on each fixed mini-batch. Because the size of fixed mini-batches is typically much smaller than the whole sample size, it can be easily computed. This leads to much reduced computation cost for each computational iteration. It makes FMGD computationally efficient and practically more feasible. To demonstrate the theoretical properties of FMGD, we start with a linear regression model with a constant learning rate. We study its numerical convergence and statistical efficiency properties. We find that sufficiently small learning rates are necessarily required for both numerical convergence and statistical efficiency. Nevertheless, an extremely small learning rate might lead to painfully slow numerical convergence. To solve the problem, a diminishing learning rate scheduling strategy can be used. This leads to the FMGD estimator with faster numerical convergence and better statistical efficiency. Finally, the FMGD algorithms with random shuffling and a general loss function are also studied.
翻译:本文研究了用于解决大规模数据集优化问题的固定小批量梯度下降(FMGD)算法。在FMGD中,整个样本被划分为多个互不重叠的分区。这些分区一旦形成,则在算法后续过程中保持不变。为方便起见,我们将这些固定分区称为固定小批量。在每个计算迭代中,梯度依次在每个固定小批量上计算。由于固定小批量的大小通常远小于整个样本量,计算过程较为简便。这使得每次计算迭代的计算成本大幅降低,从而赋予FMGD高效的计算性能与更强的实际可行性。为论证FMGD的理论性质,我们首先考虑采用恒定学习率的线性回归模型,研究其数值收敛性与统计效率特性。研究发现,要实现数值收敛与统计效率,必须采用充分小的学习率。然而,极小的学习率可能导致数值收敛速度极其缓慢。为解决该问题,可采用递减学习率调度策略,由此获得的FMGD估计量具有更快的数值收敛速度与更优的统计效率。最后,本文还研究了带随机洗牌与一般损失函数的FMGD算法。