Divide2Conquer (D2C): a decentralized approach towards overfitting remediation in deep learning

Loading...
Thumbnail Image

Publisher

Institute of Electrical and Electronics Engineers Inc.

Citation

M. S. Bari Siddiqui, M. Mohaiminul Islam and M. G. Rabiul Alam, "Divide2Conquer (D2C): A Decentralized Approach Towards Overfitting Remediation in Deep Learning," 2024 IEEE International Conference on Big Data (BigData), Washington, DC, USA, 2024, pp. 1458-1463, doi: 10.1109/BigData62323.2024.10826082.

Abstract

Overfitting remains a persistent challenge in deep learning. It is primarily attributed to data outliers, noise, and limited training set sizes. This paper presents Divide2Conquer (D2C), a novel technique designed to address this issue. D2C proposes partitioning the training data into multiple subsets and training separate identical models on them. To avoid overfitting on any specific subset, the trained parameters from these models are aggregated and averaged periodically throughout the training phase, enabling the model to learn from the entire dataset while mitigating the impact of individual outliers or noise. Empirical evaluations on multiple benchmark datasets across various deep learning tasks demonstrate that D2C effectively improves generalization performance, particularly for larger datasets. This study verifies D2C's ability to achieve significant performance gains both as a standalone technique and when used in conjunction with other overfitting reduction methods through a series of experiments, including analysis of decision boundaries, loss curves, and other performance metrics. It also provides valuable insights into the implementation and hyperparameter tuning of D2C. Our codes are publicly available at: https://github.com/Saiful185/Divide2Conquer.

Description

Type

Conference Proceedings