Generalization analysis of deep CNNs under maximum correntropy criterion

Yingqiao Zhang, Zhiying Fang*, Jun Fan

*Corresponding author for this work

Research output: Contribution to journalJournal articlepeer-review

1 Citation (Scopus)

Abstract

Convolutional neural networks (CNNs) have gained immense popularity in recent years, finding their utility in diverse fields such as image recognition, natural language processing, and bio-informatics. Despite the remarkable progress made in deep learning theory, most studies on CNNs, especially in regression tasks, tend to heavily rely on the least squares loss function. However, there are situations where such learning algorithms may not suffice, particularly in the presence of heavy-tailed noises or outliers. This predicament emphasizes the necessity of exploring alternative loss functions that can handle such scenarios more effectively, thereby unleashing the true potential of CNNs. In this paper, we investigate the generalization error of deep CNNs with the rectified linear unit (ReLU) activation function for robust regression problems within an information-theoretic learning framework. Our study demonstrates that when the regression function exhibits an additive ridge structure and the noise possesses a finite pth moment, the empirical risk minimization scheme, generated by the maximum correntropy criterion and deep CNNs, achieves fast convergence rates. Notably, these rates align with the mini-max optimal convergence rates attained by fully connected neural network model with the Huber loss function up to a logarithmic factor. Additionally, we further establish the convergence rates of deep CNNs under the maximum correntropy criterion when the regression function resides in a Sobolev space on the sphere.

Original languageEnglish
Article number106226
JournalNeural Networks
Volume174
Early online date5 Mar 2024
DOIs
Publication statusPublished - Jun 2024

Scopus Subject Areas

  • Cognitive Neuroscience
  • Artificial Intelligence

User-Defined Keywords

  • Convergence rates
  • Convolutional neural networks
  • Heavy-tailed noises
  • Information-theoretic learning
  • Learning theory
  • Maximum correntropy criterion

Fingerprint

Dive into the research topics of 'Generalization analysis of deep CNNs under maximum correntropy criterion'. Together they form a unique fingerprint.

Cite this