Physics-Informed Spiking Neural Networks via Conservative Flux Quantization

TL;DR

This paper introduces PISNN, a physics-informed spiking neural network framework that ensures strict physical conservation laws and robust temporal generalization, enabling accurate, energy-efficient real-time physics simulations on edge devices.

Contribution

The paper proposes a novel PISNN framework with C-LIF neurons and CFQ strategy, ensuring local mass conservation and general-purpose physics solving capabilities.

Findings

Accurately simulates 1D heat equation dynamics with mass conservation.

Effectively models 2D Laplace's Equation with high fidelity.

Outperforms conventional PINNs in energy efficiency and long-term stability.

Abstract

Real-time, physically-consistent predictions on low-power edge devices is critical for the next generation embodied AI systems, yet it remains a major challenge. Physics-Informed Neural Networks (PINNs) combine data-driven learning with physics-based constraints to ensure the model's predictions are with underlying physical principles.However, PINNs are energy-intensive and struggle to strictly enforce physical conservation laws. Brain-inspired spiking neural networks (SNNs) have emerged as a promising solution for edge computing and real-time processing. However, naively converting PINNs to SNNs degrades physical fidelity and fails to address long-term generalization issues. To this end, this paper introduce a novel Physics-Informed Spiking Neural Network (PISNN) framework. Importantly, to ensure strict physical conservation, we design the Conservative Leaky Integrate-and-Fire (C-LIF) neuron, whose dynamics structurally guarantee local mass preservation. To achieve robust temporal generalization, we introduce a novel Conservative Flux Quantization (CFQ) strategy, which redefines neural spikes as discrete packets of physical flux. Our CFQ learns a time-invariant physical evolution operator, enabling the PISNN to become a general-purpose solver -- conservative-by-construction. Extensive experiments show that our PISNN excels on diverse benchmarks. For both the canonical 1D heat equation and the more challenging 2D Laplace's Equation, it accurately simulates the system dynamics while maintaining perfect mass conservation by design -- a feat that is challenging for conventional PINNs. This work establishes a robust framework for fusing the rigor of scientific computing with the efficiency of neuromorphic engineering, paving the way for complex, long-term, and energy-efficient physics predictions for intelligent systems.