Solving high-dimensional parabolic PDEs using the tensor train format
Lorenz Richter, Leon Sallandt, Nikolas Nüsken
Abstract
High-dimensional partial differential equations (PDEs) are ubiquitous in economics, science and engineering. However, their numerical treatment poses formidable challenges since traditional grid-based methods tend to be frustrated by the curse of dimensionality. In this paper, we argue that tensor trains provide an appealing approximation framework for parabolic PDEs: the combination of reformulations in terms of backward stochastic differential equations and regression-type methods in the tensor format holds the promise of leveraging latent low-rank structures enabling both compression and efficient computation. Following this paradigm, we develop novel iterative schemes, involving either explicit and fast or implicit and accurate updates. We demonstrate in a number of examples that our methods achieve a favorable trade-off between accuracy and computational efficiency in comparison with state-of-the-art neural network based approaches.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 95eb7c8f-174b-4a67-97b7-1e9752ca7114Cited by top-tier papers8
- Machine Learning For Elliptic PDEs: Fast Rate Generalization Bound, Neural Scaling Law and Minimax OptimalityYiping Lu, Haoxuan Chen, Jianfeng Lu, Lexing Ying et al.ICLR 2022 · 54 citations
- Permutation Search of Tensor Network Structures via Local SamplingChao Li, Junhua Zeng, Zerui Tao, Qibin ZhaoICML 2022 · 31 citations
- Alternating Local Enumeration (TnALE): Solving Tensor Network Structure Search with Fewer EvaluationsChao Li, Junhua Zeng, Chunmei Li, Cesar F. Caiafa et al.ICML 2023 · 24 citations
- Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and InferenceDenis Blessing, Julius Berner, Lorenz Richter, Carles Domingo-Enrich et al.NeurIPS 2025 · 24 citations
- Cost-efficient Gaussian tensor network embeddings for tensor-structured inputsLinjian Ma, Edgar SolomonikNeurIPS 2022 · 18 citations
Related papers
- Tensor Train Diffusion: Leveraging Low-Rank Structures for High-Dimensional Score-Based SamplingRobert Gruhlke, Julius Berner, David Sommer, Lorenz RichterICML 2026
- Efficient Leverage Score Sampling for Tensor Train DecompositionVivek Bharadwaj, Beheshteh T. Rakhshan, Osman Asif Malik, Guillaume RabusseauNeurIPS 2024 · 7 citations
- Parametric Complexity Bounds for Approximating PDEs with Neural NetworksTanya Marwah, Zachary C. Lipton, Andrej RisteskiNeurIPS 2021 · 23 citations
- Fast Tensor Completion via Approximate Richardson IterationMehrdad Ghadiri, Matthew Fahrbach, Yunbum Kook, Ali JadbabaieICML 2025
- Solving Poisson Equations using Neural Walk-on-SpheresHong Chul Nam, Julius Berner, Anima AnandkumarICML 2024 · 12 citations
