Addressing Spectral Bias of Deep Neural Networks by Multi-Grade Deep Learning

TL;DR

This paper introduces Multi-Grade Deep Learning (MGDL), a method that composes shallow neural networks to effectively learn high-frequency features, addressing the spectral bias in deep neural networks and improving their ability to capture complex functions.

Contribution

The paper proposes MGDL, a novel incremental training approach that combines shallow networks to overcome spectral bias in DNNs, enabling better high-frequency feature learning.

Findings

MGDL effectively captures high-frequency information in various datasets.

Compositions of shallow networks can approximate high-frequency functions.

MGDL outperforms traditional DNNs in representing high-frequency features.

Abstract

Deep neural networks (DNNs) suffer from the spectral bias, wherein DNNs typically exhibit a tendency to prioritize the learning of lower-frequency components of a function, struggling to capture its high-frequency features. This paper is to address this issue. Notice that a function having only low frequency components may be well-represented by a shallow neural network (SNN), a network having only a few layers. By observing that composition of low frequency functions can effectively approximate a high-frequency function, we propose to learn a function containing high-frequency components by composing several SNNs, each of which learns certain low-frequency information from the given data. We implement the proposed idea by exploiting the multi-grade deep learning (MGDL) model, a recently introduced model that trains a DNN incrementally, grade by grade, a current grade learning from the residue of the previous grade only an SNN composed with the SNNs trained in the preceding grades as features. We apply MGDL to synthetic, manifold, colored images, and MNIST datasets, all characterized by presence of high-frequency features. Our study reveals that MGDL excels at representing functions containing high-frequency information. Specifically, the neural networks learned in each grade adeptly capture some low-frequency information, allowing their compositions with SNNs learned in the previous grades effectively representing the high-frequency features. Our experimental results underscore the efficacy of MGDL in addressing the spectral bias inherent in DNNs. By leveraging MGDL, we offer insights into overcoming spectral bias limitation of DNNs, thereby enhancing the performance and applicability of deep learning models in tasks requiring the representation of high-frequency information. This study confirms that the proposed method offers a promising solution to address the spectral bias of DNNs.