Search papers, labs, and topics across Lattice.
The paper introduces P-PINN, a selective pruning framework to enhance the robustness of Physics-Informed Neural Networks (PINNs) in solving PDE inverse problems under noisy conditions. P-PINN identifies and prunes neurons that are overly influenced by corrupted data points by evaluating a joint residual-data fidelity indicator and a bias-based neuron importance measure. Experiments on PDE inverse problems demonstrate that P-PINN significantly improves accuracy and training stability, achieving up to a 96.6% reduction in relative error compared to baseline PINNs.
PINNs can be made dramatically more robust to noisy data (up to 96.6% error reduction) by selectively pruning neurons that are overly sensitive to corrupted samples.
Physics-informed neural networks (PINNs) provide a promising framework for solving inverse problems governed by partial differential equations (PDEs) by integrating observational data and physical constraints in a unified optimization objective. However, the ill-posed nature of PDE inverse problems makes them highly sensitive to noise. Even a small fraction of corrupted observations can distort internal neural representations, severely impairing accuracy and destabilizing training. Motivated by recent advances in machine unlearning and structured network pruning, we propose P-PINN, a selective pruning framework designed to unlearn the influence of corrupted data in a pretrained PINN. Specifically, starting from a PINN trained on the full dataset, P-PINN evaluates a joint residual--data fidelity indicator, a weighted combination of data misfit and PDE residuals, to partition the training set into reliable and corrupted subsets. Next, we introduce a bias-based neuron importance measure that quantifies directional activation discrepancies between the two subsets, identifying neurons whose representations are predominantly driven by corrupted samples. Building on this, an iterative pruning strategy then removes noise-sensitive neurons layer by layer. The resulting pruned network is fine-tuned on the reliable data subject to the original PDE constraints, acting as a lightweight post-processing stage rather than a complete retraining. Numerical experiments on extensive PDE inverse-problem benchmarks demonstrate that P-PINN substantially improves robustness, accuracy, and training stability under noisy conditions, achieving up to a 96.6\% reduction in relative error compared with baseline PINNs. These results indicate that activation-level post hoc pruning is a promising mechanism for enhancing the reliability of physics-informed learning in noise-contaminated settings.
