Deep-MacroFin: Informed Equilibrium Neural Network for Continuous Time Economic Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wu, Yuntao, Guo, Jiayuan, Gopalakrishna, Goutham, Poulos, Zissis
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909609407545344
author Wu, Yuntao
Guo, Jiayuan
Gopalakrishna, Goutham
Poulos, Zissis
author_facet Wu, Yuntao
Guo, Jiayuan
Gopalakrishna, Goutham
Poulos, Zissis
contents In this paper, we present Deep-MacroFin, a comprehensive framework designed to solve partial differential equations, with a particular focus on models in continuous time economics. This framework leverages deep learning methodologies, including Multi-Layer Perceptrons and the newly developed Kolmogorov-Arnold Networks. It is optimized using economic information encapsulated by Hamilton-Jacobi-Bellman (HJB) equations and coupled algebraic equations. The application of neural networks holds the promise of accurately resolving high-dimensional problems with fewer computational demands and limitations compared to other numerical methods. This framework can be readily adapted for systems of partial differential equations in high dimensions. Importantly, it offers a more efficient (5$\times$ less CUDA memory and 40$\times$ fewer FLOPs in 100D problems) and user-friendly implementation than existing libraries. We also incorporate a time-stepping scheme to enhance training stability for nonlinear HJB equations, enabling the solution of 50D economic models.
format Preprint
id arxiv_https___arxiv_org_abs_2408_10368
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Deep-MacroFin: Informed Equilibrium Neural Network for Continuous Time Economic Models
Wu, Yuntao
Guo, Jiayuan
Gopalakrishna, Goutham
Poulos, Zissis
Machine Learning
Computational Engineering, Finance, and Science
Computational Finance
I.0; J.4
In this paper, we present Deep-MacroFin, a comprehensive framework designed to solve partial differential equations, with a particular focus on models in continuous time economics. This framework leverages deep learning methodologies, including Multi-Layer Perceptrons and the newly developed Kolmogorov-Arnold Networks. It is optimized using economic information encapsulated by Hamilton-Jacobi-Bellman (HJB) equations and coupled algebraic equations. The application of neural networks holds the promise of accurately resolving high-dimensional problems with fewer computational demands and limitations compared to other numerical methods. This framework can be readily adapted for systems of partial differential equations in high dimensions. Importantly, it offers a more efficient (5$\times$ less CUDA memory and 40$\times$ fewer FLOPs in 100D problems) and user-friendly implementation than existing libraries. We also incorporate a time-stepping scheme to enhance training stability for nonlinear HJB equations, enabling the solution of 50D economic models.
title Deep-MacroFin: Informed Equilibrium Neural Network for Continuous Time Economic Models
topic Machine Learning
Computational Engineering, Finance, and Science
Computational Finance
I.0; J.4
url https://arxiv.org/abs/2408.10368