CoLoRA: Continuous low-rank adaptation for reduced implicit neural modeling of parameterized partial differential equations
Jules Berman, Benjamin Peherstorfer
Abstract
This work introduces reduced models based on Continuous Low Rank Adaptation (CoLoRA) that pre-train neural networks for a given partial differential equation and then continuously adapt low-rank weights in time to rapidly predict the evolution of solution fields at new physics parameters and new initial conditions. The adaptation can be either purely data-driven or via an equation-driven variational approach that provides Galerkin-optimal approximations. Because CoLoRA approximates solution fields locally in time, the rank of the weights can be kept small, which means that only few training trajectories are required offline so that CoLoRA is well suited for data-scarce regimes. Predictions with CoL-oRA are orders of magnitude faster than with classical methods and their accuracy and parameter efficiency is higher compared to other neural network 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.
Cited by top-tier papers5
- Parametric model reduction of mean-field and stochastic systems via higher-order action matchingJules Berman, Tobias Blickhan, Benjamin PeherstorferNeurIPS 2024 · 10 citations
- Learning of Population Dynamics: Inverse Optimization Meets JKO SchemeMikhail Persiianov, Jiawei Chen, Petr Mokrov, Alexander Tyurin et al.ICLR 2026 · 7 citations
- LoRA-EnVar: Parameter-Efficient Hybrid Ensemble Variational Assimilation for Weather ForecastingYi Xiao, Hang Fan, Kun Chen, Ye Cao et al.NeurIPS 2025 · 3 citations
- A Dirac-Frenkel-Onsager principle: Instantaneous residual minimization with gauge momentum for nonlinear parametrizations of PDE solutionsMatteo Raviola, Benjamin PeherstorferICML 2026 · 1 citation
- ELoRA: Low-Rank Adaptation for Equivariant GNNsChen Wang, Siyu Hu, Guangming Tan, Weile JiaICML 2025
Builds on15
- Implicit Neural Representations with Periodic Activation FunctionsVincent Sitzmann, Julien N. P. Martel, Alexander W. Bergman, David B. Lindell et al.NeurIPS 2020 · 4,008 citations
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Message Passing Neural PDE SolversJohannes Brandstetter, Daniel E. Worrall, Max WellingICLR 2022 · 410 citations
- Operator Learning with Neural Fields: Tackling PDEs on General GeometriesLouis Serrano, Lise Le Boudec, Armand Kassaï Koupaï, Thomas X. Wang et al.NeurIPS 2023 · 114 citations
- Initialization and Regularization of Factorized Neural LayersMikhail Khodak, Neil A. Tenenholtz, Lester Mackey, Nicolò FusiICLR 2021 · 74 citations
Related papers
- Randomized Sparse Neural Galerkin Schemes for Solving Evolution Equations with Deep NetworksJules Berman, Benjamin PeherstorferNeurIPS 2023 · 42 citations
- Flora: Low-Rank Adapters Are Secretly Gradient CompressorsYongchang Hao, Yanshuai Cao, Lili MouICML 2024 · 113 citations
- CROM: Continuous Reduced-Order Modeling of PDEs Using Implicit Neural RepresentationsPeter Yichen Chen, Jinxu Xiang, Dong Heon Cho, Yue Chang et al.ICLR 2023 · 11 citations
- GeoLoRA: Geometric integration for parameter efficient fine-tuningSteffen Schotthöfer, Emanuele Zangrando, Gianluca Ceruti, Francesco Tudisco et al.ICLR 2025
- Learning continuous-time PDEs from sparse data with graph neural networksValerii Iakovlev, Markus Heinonen, Harri LähdesmäkiICLR 2021 · 81 citations
