TT-PINN: A Tensor-Compressed Neural PDE Solver for Edge Computing
Not on the map. Why?
Abstract (arXiv)
Physics-informed neural networks (PINNs) have been increasingly employed due to their capability of modeling complex physics systems. To achieve better expressiveness, increasingly larger network sizes are required in many problems. This has caused challenges when we need to train PINNs on edge devices with limited memory, computing and energy resources. To enable training PINNs on edge devices, this paper proposes an end-to-end compressed PINN based on Tensor-Train decomposition. In solving a Helmholtz equation, our proposed model significantly outperforms the original PINNs with few parameters and achieves satisfactory prediction with up to 15× overall parameter reduction.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 8 of its 16 references that are in the map. First posted on arXiv 4 Jul 2022.
Builds on
8 of 16 references.
References · 8
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015Automatic differentiation in machine learning: a surveycited by 846
- 2021Physics-informed neural networks with hard constraints for inverse designcited by 277
- 2015Tensorizing Neural Networkscited by 246
- 2020DeepReach: A Deep Learning Approach to High-Dimensional Reachabilitycited by 81
- 2019Bayesian Tensorized Neural Networks with Automatic Rank Selectioncited by 24
- 2020Towards Compact Neural Networks via End-to-End Training: A Bayesian Tensor Approach with Automatic Rank Determinationcited by 17
- 2020A Neural Network Approach Applied to Multi-Agent Optimal Controlcited by 14
Built on by
Citing papers per year: 2023: 3; 2024: 1; 2025: 2.
6 papers.
- 2023PIFON-EPT: MR-Based Electrical Property Tomography Using Physics-Informed Fourier Networkscited by 4
- 2023Real-Time FJ/MAC PDE Solvers via Tensorized, Back-Propagation-Free Optical PINN Trainingcited by 4
- 2024Separable Operator Networkscited by 4
- 2023Quantization-Aware and Tensor-Compressed Training of Transformers for Natural Language Understandingcited by 2
- 2025Tensor-Compressed and Fully-Quantized Training of Neural PDE Solverscited by 1
- 2025Experimental Demonstration of an Optical Neural PDE Solver via On-Chip PINN Trainingcited by 0
- 2025Tensor-Compressed and Fully-Quantized Training of Neural PDE Solvers10 Dec 2025
- 2025Experimental Demonstration of an Optical Neural PDE Solver via On-Chip PINN Training1 Jan 2025
- 2024Separable Operator Networks15 Jul 2024
- 2023Real-Time FJ/MAC PDE Solvers via Tensorized, Back-Propagation-Free Optical PINN Training31 Dec 2023
- 2023Quantization-Aware and Tensor-Compressed Training of Transformers for Natural Language Understanding1 Jun 2023
- 2023PIFON-EPT: MR-Based Electrical Property Tomography Using Physics-Informed Fourier Networks23 Feb 2023