On cheap entropy-sparsified regression learning

Illia Horenko*, Edoardo Vecchi, Juraj Kardoš, Andreas Wächter, Olaf Schenk, Terence O'Kane, Patrick Gagliardini, Susanne Gerber

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

8 Scopus citations

Abstract

Regression learning is one of the long-standing problems in statistics, machine learning, and deep learning (DL). We show that writing this problem as a probabilistic expectation over (unknown) feature probabilities, increasing the number of unknown parameters and seemingly making the problem more complex-actually leads to a simplification, allowing to incorporate the physical principle of entropy maximization. It helps decompose a very general setting of this learning problem (including discretization, feature selection, and learning multiple piece-wise linear regressions) into an iterative sequence of simple substeps, which are either analytically solvable or cheaply computable through an efficient second-order numerical solver with a sublinear cost scaling. This leads to the computationally cheap and robust non- DL second-order Sparse Probabilistic Approximation for Regression Task Analysis (SPARTAn) algorithm, that can be efficiently applied to problems with millions of feature dimensions on a commodity laptop, when the state-of-the-art learning tools would require supercomputers. SPARTAn is compared to a range of commonly used regression learning tools on synthetic problems and on the prediction of the El Niño Southern Oscillation, the dominant interannual mode of tropical climate variability. The obtained SPARTAn learners provide more predictive, sparse, and physically explainable data descriptions, clearly discerning the important role of ocean temperature variability at the thermocline in the equatorial Pacific. SPARTAn provides an easily interpretable description of the timescales by which these thermocline temperature features evolve and eventually express at the surface, thereby enabling enhanced predictability of the key drivers of the interannual climate.

Original languageEnglish (US)
Article numbere2214972120
JournalProceedings of the National Academy of Sciences of the United States of America
Volume120
Issue number1
DOIs
StatePublished - Jan 3 2023

Keywords

  • climate prediction
  • entropy
  • numerics
  • supervised learning

ASJC Scopus subject areas

  • General

Fingerprint

Dive into the research topics of 'On cheap entropy-sparsified regression learning'. Together they form a unique fingerprint.

Cite this