Login / Signup

A law of data separation in deep learning.

Hangfeng HeWeijie J Su
Published in: Proceedings of the National Academy of Sciences of the United States of America (2023)
While deep learning has enabled significant advances in many areas of science, its black-box nature hinders architecture design for future artificial intelligence applications and interpretation for high-stakes decision-makings. We addressed this issue by studying the fundamental question of how deep neural networks process data in the intermediate layers. Our finding is a simple and quantitative law that governs how deep neural networks separate data according to class membership throughout all layers for classification. This law shows that each layer improves data separation at a constant geometric rate, and its emergence is observed in a collection of network architectures and datasets during training. This law offers practical guidelines for designing architectures, improving model robustness and out-of-sample performance, as well as interpreting the predictions.
Keyphrases
  • deep learning
  • artificial intelligence
  • neural network
  • big data
  • machine learning
  • electronic health record
  • convolutional neural network
  • public health
  • data analysis
  • mass spectrometry