Back to Search Start Over

Entropic Fictitious Play for Mean Field Optimization Problem.

Authors :
Fan Chen
Zhenjie Ren
Songbo Wang
Source :
Journal of Machine Learning Research. 2023, Vol. 24, p1-36. 36p.
Publication Year :
2023

Abstract

We study two-layer neural networks in the mean field limit, where the number of neurons tends to infinity. In this regime, the optimization over the neuron parameters becomes the optimization over the probability measures, and by adding an entropic regularizer, the minimizer of the problem is identified as a fixed point. We propose a novel training algorithm named entropic fictitious play, inspired by the classical fictitious play in game theory for learning Nash equilibriums, to recover this fixed point, and the algorithm exhibits a two-loop iteration structure. Exponential convergence is proved in this paper and we also verify our theoretical results by simple numerical examples. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
15324435
Volume :
24
Database :
Academic Search Index
Journal :
Journal of Machine Learning Research
Publication Type :
Academic Journal
Accession number :
176355403