Physics-guided spatiotemporal neural models for fuel density prediction

# PHYSICS-GUIDED SPATIOTEMPORAL NEURAL MODELS FOR FUEL DENSITY PREDICTION
Source: [https://arxiv.org/html/2607.06999](https://arxiv.org/html/2607.06999)
Tolga Caglar, Jaynil Jaiswal, Saqib Azim, Yudhir Gala, Mai H\. Nguyen, Ilkay Altintas San Diego Supercomputer Center University of California, San Diego \{tcaglar, mhnguyen\}@ucsd\.edu

###### Abstract

This paper presents a physics\-guided machine learning \(PGML\) framework for fuel density prediction, integrating physics constraints and domain knowledge into deep learning models to enhance model accuracy and stability\. We explore three deep learning architectures — ConvLSTM, Adaptive Fourier Neural Operator \(AFNONet\), and Video Vision Transformer \(ViViT\) — to model the spatiotemporal evolution of fuel density\. Our approach incorporates differentiable physics\-informed terms in the loss function, including a mass\-conserving fuel transport term and a rate\-of\-spread estimation\. Experimental results, averaged across multiple independent trials, demonstrate that the proposed PGML framework outperforms purely data\-driven baselines without physics constraints in both accuracy and stability\. This framework enables computationally efficient, physically plausible fire forecasting to support adaptive prescribed burn management\.

## IIntroduction

Wildfires in the United States consume millions of acres each year, posing significant threats to ecosystems, human life, and property\[[9](https://arxiv.org/html/2607.06999#bib.bib12)\]\. To mitigate these risks, fire managers employ prescribed burns — intentional, controlled fires in designated forest areas designed to reduce fuel loads that can cause dangerous wildfires, enhance forest health, and help manage pests\[[4](https://arxiv.org/html/2607.06999#bib.bib1)\]\. However, these burns are only initiated under specific conditions to ensure safety and prevent unintended consequences\[[16](https://arxiv.org/html/2607.06999#bib.bib15)\]\. Fire managers rely on process\-based models such as FARSITE\[[6](https://arxiv.org/html/2607.06999#bib.bib7)\]and QUIC\-Fire\[[10](https://arxiv.org/html/2607.06999#bib.bib17)\]to predict the behavior and trajectory of prescribed fires under varying weather conditions \(primarily wind speed and wind direction\), fuel density in the area, and pre\-defined ignition conditions\. These process\-based models simulate physical fire dynamics to assess burn risks, but their computational demands often prevent real\-time decision\-making\[[1](https://arxiv.org/html/2607.06999#bib.bib9),[11](https://arxiv.org/html/2607.06999#bib.bib8)\]\.

While purely data\-driven deep learning architectures, such as CNNs\[[13](https://arxiv.org/html/2607.06999#bib.bib19),[5](https://arxiv.org/html/2607.06999#bib.bib31)\]and U\-Nets\[[18](https://arxiv.org/html/2607.06999#bib.bib13)\], offer computational speedup, they are ill\-suited for complex multi\-point ignitions and frequently predict non\-physical behaviors like spontaneous fuel regeneration\. To counteract this, existing physics\-guided ML approaches attempt to constrain outputs using partial differential equations \(PDEs\)\[[3](https://arxiv.org/html/2607.06999#bib.bib10),[1](https://arxiv.org/html/2607.06999#bib.bib9)\]\. However, these rigid mathematical formulations are often highly unstable, fail to converge, or suffer from well\-documented gradient flow pathologies\[[17](https://arxiv.org/html/2607.06999#bib.bib26),[14](https://arxiv.org/html/2607.06999#bib.bib20)\]\. Our proposed surrogate framework addresses these gaps by embedding domain\-specific physical constraints as ”soft” penalties, providing the guidance of physics without the numerical fragility of strict equation solving\.

In this paper, we present a framework to incorporate physics components into deep learning models that can work across different architectures\. Specifically, we present a comparative analysis with and without physics\-guided loss for three deep learning models, namely Convolutional LSTM \(ConvLSTM\)\[[15](https://arxiv.org/html/2607.06999#bib.bib21)\], Adaptive Fourier Neural Operator \(AFNONet\)\[[7](https://arxiv.org/html/2607.06999#bib.bib2)\], and Video Vision Transformer \(ViViT\)\[[2](https://arxiv.org/html/2607.06999#bib.bib3)\], for the task of spatiotemporal fuel density change\.

## IIDataset and Problem Formulation

We utilize an ensemble of QUIC\-Fire simulations of fire spread, measuring fuel density \(mass of fire fuel per unit volume\) over time in an area\. Each simulation runs for 50 seconds on a 300×\\times300 spatial grid with flat grassland\. The dataset comprises all possible combinations of 7 wind speed values \(ranging from 1 m/s to 15 m/s\), 11 wind direction values \(from 230 to 330 degrees\), 4 distinct ignition pattern types \(aerial, inward, outward, strip north\)\. The wind speed and wind direction remain constant across the entire spatial grid and all timesteps\. The data contains fuel densities in the range \[0,0\.7​k​g/m20,0\.7kg/m^\{2\}\]\. To ensure effective learning, preprocessing steps include normalization of wind speed and wind direction using min\-max scaling, and incorporation of a source map\. The source map is a 2d array that represents fuel density evolution over time at the lowest wind speed \(1 m/s\), and the same wind direction and ignition pattern as the input data, allowing the model to learn fuel density propagation patterns\. The final dataset consists of four feature channels: source map, wind direction, wind speed, and ignition pattern\.

Our objective is to model the evolution of fuel density as a prescribed fire propagates under different environmental conditions\. Specifically, we aim to develop a data\-driven emulator of QUIC\-Fire, using simulated runs as training data\. Each training example, indexed byii, is represented by an input tensorXi∈RT×H×W×CX\_\{i\}\\in R^\{T\\times H\\times W\\times C\}where T denotes the number of timesteps in the sequence, H and W define the spatial grid dimensions, and C represents the number of input channels\. The corresponding output tensor for fuel density predictions is given byYi∈RT×H×W×1Y\_\{i\}\\in R^\{T\\times H\\times W\\times 1\}\. In our setup, wind speed and wind direction remain static across time\. Ignition patterns and source fuel density maps, on the other hand, evolve dynamically over time\. By structuring the dataset in this manner, our approach allows the model to learn complex spatiotemporal interactions between fire behavior and environmental conditions, enabling improved predictions of fuel density evolution in prescribed fire scenarios\.

## IIIMethods

We evaluated three deep learning architectures to represent distinct spatiotemporal paradigms:ConvLSTMuses a convolutional neural network \(CNN\) for spatial structure and a long short\-term memory \(LSTM\) network for temporal dynamics;ViViTleverages self\-attention to model long\-range temporal dependencies; andAFNONetutilizes Fourier domain mixing to efficiently capture global dependencies\.

### III\-AConvLSTM

Our ConvLSTM model adapts the architecture proposed by Chatterjee et al\.\[[4](https://arxiv.org/html/2607.06999#bib.bib1)\]\. While the references work utilizes a four\-layer stacked ConvLSTM with Batch Normalization, we simplify this design to atwo\-layer structure\(64 filters,3×33\\times 3kernels\) to prevent overfitting on limited simulation data\.

We introduce two critical architectural enhancements to the standard ConvLSTM to support fuel density modeling:

1. 1\.Residual Connections:We attach an auxiliary 3D convolutional head to the first LSTM layer to allow injection of physics\-based gradients directly into the intermediate layers via residual connections\[[8](https://arxiv.org/html/2607.06999#bib.bib22)\], preventing gradient vanishing during backpropagation through time\.
2. 2\.Spatiotemporal Smoothing:We use a 3D kernel of \(3×3×33\\times 3\\times 3\) which integrates the information across adjacent time steps and pixels, ensuring temporal continuity in the predicted fire front\.

To maintain numerical stability, we useLeakyReLU\(α=0\.1\\alpha=0\.1\) to prevent the ”dead neuron” collapse often observed in sparse data like fire maps\.

### III\-BVideo Vision Transformers \(ViViT\)

We utilize ViViT\[[2](https://arxiv.org/html/2607.06999#bib.bib3)\], a transformer\-based architecture originally designed for video classification, to model the spatiotemporal evolution of fuel density\. ViViT processes its input by tokenizing the video sequence using tubelet embeddings, which serve as the spatiotemporal representation of the fuel grid in our use case\.

To reconstruct the fuel density map, we added additional processing layers to our ViViT\. Token embeddings are first mapped back to their spatial regions through a token\-to\-tubelet reconstruction process: A Multi\-Layer Perceptron \(MLP\) projects each token into its respective tubelet dimensions\. The reconstructed tubelets are then reshaped into the predicted tensor, followed by a Conv3D layer, which refines the temporal dependencies, ultimately generating the final fuel density prediction for the next timestep\. This structured approach allows ViViT to effectively model the spatiotemporal dynamics of fire spread, leveraging transformer\-based self\-attention for accurate fuel density forecasting\.

### III\-CAdaptive Fourier Neural Operator Network \(AFNONet\)

The Adaptive Fourier Neural Operator Network \(AFNONet\)\[[7](https://arxiv.org/html/2607.06999#bib.bib2)\], also a transformer\-based deep learning model, replaces the self\-attention mechanism, which has complexity ofO​\(N2\)O\(N^\{2\}\), with Fourier\-domain token mixing, reducing complexity toO​\(N​log⁡N\)O\(N\\log N\)while effectively capturing the fine\-scale differential physics of fluid dynamics\[[12](https://arxiv.org/html/2607.06999#bib.bib18)\]\.

We adapt the standard AFNO architecture for spatiotemporal forecasting by implementing anAFNONet\_Seq2Seqmodel\. While standard AFNO models typically function as autoregressive operators mappingXt→Xt\+1X\_\{t\}\\to X\_\{t\+1\}, our adaptation leverages 3D tokenization to map a context sequenceXi​nX\_\{in\}directly to a prediction sequenceYo​u​tY\_\{out\}in a single forward pass\. The input tensorX∈ℝB×C×Ti​n×H×WX\\in\\mathbb\{R\}^\{B\\times C\\times T\_\{in\}\\times H\\times W\}is processed by a 3D Patch Embedding layer\. The volume is partitioned into non\-overlapping ”tubelets” of size\(t​i​m​e,h​e​i​g​h​t,w​i​d​t​h\)\(time,height,width\)\. A 3D convolution projects each tubelet into a latent embedding dimensionDD, effectively compressing local spatiotemporal dynamics into a single token grid\. The token grid is reshaped to merge temporal and vertical spatial dimensions, allowing the AFNO blocks to process the sequence as a unified field\.

Unlike standard classification heads, our decoding head performs a high\-dimensional expansion to reconstruct the full video volume\. A linear projection maps each latent token back to the pixel space of its original tubelet:v^∈ℝCo​u​t×pt×ph×pw\\hat\{v\}\\in\\mathbb\{R\}^\{C\_\{out\}\\times p\_\{t\}\\times p\_\{h\}\\times p\_\{w\}\}\. Finally, a rearrangement operation \(un\-patching\) merges these local tubelet predictions back into the global grid structure, yielding the continuous output sequenceYo​u​t∈ℝB×To​u​t×Co​u​t×H×WY\_\{out\}\\in\\mathbb\{R\}^\{B\\times T\_\{out\}\\times C\_\{out\}\\times H\\times W\}\.

![Refer to caption](https://arxiv.org/html/2607.06999v1/figures/seq2seq_afnonet.png)Figure 1:Input frames are tokenized into 3D tubelets and processed via global Fourier\-domain mixing to capture long\-range dependencies\. The decoding head expands latent tokens into dense blocks, applying a pixel\-shuffle operation to reconstruct the complete future sequenceYo​u​tY\_\{out\}\.![Refer to caption](https://arxiv.org/html/2607.06999v1/figures/Fig2_Physics_Fixed.png)Figure 2:Results of training with physics\-guided loss – Fuel density prediction at timestep 42 for PG\-ConvLSTM, PG\-AFNONet, and PG\-ViViT are displayed, along with ground truth\.TABLE I:Results of ConvLSTM, AFNONet, and ViViT with and without Physics\-Guided Loss Components
\(Mean and Standard Deviation from 3 Runs\)

## IVIntegrating Physics Constraints

To enforce physical consistency, we define a composite loss function composed of the following loss components:

1\. Base MSE \(ℒMSE\\mathcal\{L\}\_\{\\text\{MSE\}\}\):We employ standard pixel\-wise Mean Squared Error\(MSE\) as an objective to minimize the discrepancy between the ground truth and prediction\.

2\. Fuel Transport \(ℒfuel\\mathcal\{L\}\_\{\\text\{fuel\}\}\):Since fuel consumption is irreversible, we penalize non\-physical fuel regeneration \(i\.e\., positive temporal gradients\) by taking MSE of pixels with increased fuel density over time\.

ℒfuel=1N​∑t𝟙\(F^t\>F^t−1\)⊙‖F^t−FtG​T‖2,\\mathcal\{L\}\_\{\\text\{fuel\}\}=\\frac\{1\}\{N\}\\sum\_\{t\}\\mathbbm\{1\}\_\{\(\\hat\{F\}\_\{t\}\>\\hat\{F\}\_\{t\-1\}\)\}\\odot\\left\\\|\\hat\{F\}\_\{t\}\-F^\{GT\}\_\{t\}\\right\\\|^\{2\},\(1\)whereNNis the number of timesteps,F^t\\hat\{F\}\_\{t\}is the 2d fuel density prediction map at timett, andFtG​TF^\{GT\}\_\{t\}is the 2d fuel density map atttobtained from the ground truth\.

3\. State\-Weighted Losses \(ℒburn,ℒunburn\\mathcal\{L\}\_\{\\text\{burn\}\},\\mathcal\{L\}\_\{\\text\{unburn\}\}\):To weight burned and unburned pixels independently, we separated loss components based on a differentiable mask calculated through the temperature\-scaled sigmoid function at timettas:

PtG​T=σ​\(\(F∗−FtG​T\)/𝒯\),P^\{GT\}\_\{t\}=\\sigma\\left\(\{\\left\(\{F^\{\*\}\}\-F^\{GT\}\_\{t\}\\right\)/\\mathcal\{T\}\}\\right\),\(2\)whereF∗=0\.665F^\{\*\}=0\.665is the heuristically determined fuel density threshold from the ground truth fuel densities, such thatFtG​T\>F∗F^\{GT\}\_\{t\}\>F^\{\*\}are the unburned pixels, andFtG​T≤F∗F^\{GT\}\_\{t\}\\leq F^\{\*\}are the burned pixels\. We scale the sigmoid function by𝒯=0\.02\\mathcal\{T\}=0\.02, narrowing the intermediate values between0and11to resemble a differentiable mask that can backpropagate during training\. UtilizingPG​TP^\{GT\}, we calculate the burned and unburned loss components through

ℒburn=⟨⟨PtG​T⊙LM​S​E⟩⟩ℒunburn=⟨⟨\(𝟙−PtG​T\)⊙LM​S​E⟩⟩\\displaystyle\\begin\{split\}\\mathcal\{L\_\{\\text\{burn\}\}\}&=\\mathopen\{\\hbox\{\\set@color$\{\\langle\}$\}\\kern\-1\.94444pt\\hbox\{\\set@color$\{\\langle\}$\}\}P^\{GT\}\_\{t\}\\odot L\_\{MSE\}\\mathclose\{\\hbox\{\\set@color$\{\\rangle\}$\}\\kern\-1\.94444pt\\hbox\{\\set@color$\{\\rangle\}$\}\}\\\\ \\mathcal\{L\_\{\\text\{unburn\}\}\}&=\\mathopen\{\\hbox\{\\set@color$\{\\langle\}$\}\\kern\-1\.94444pt\\hbox\{\\set@color$\{\\langle\}$\}\}\(\\mathbbm\{1\}\-P^\{GT\}\_\{t\}\)\\odot L\_\{MSE\}\\mathclose\{\\hbox\{\\set@color$\{\\rangle\}$\}\\kern\-1\.94444pt\\hbox\{\\set@color$\{\\rangle\}$\}\}\\end\{split\}\(3\)This formulation ensures thatℒburn\\mathcal\{L\}\_\{\\text\{burn\}\}dominates in active fire regions, whileℒunburn\\mathcal\{L\}\_\{\\text\{unburn\}\}stabilizes the background\.

4\. Rate of Spread \(ℒROS\\mathcal\{L\}\_\{\\text\{ROS\}\}\):For a fire propagating with eastward wind, the rate of spread \(ROS\) is the speed at which the leading edge in thex−x\-axis of a fire advances over the landscape\. We can calculate the leading edge of an eastward\-propagating fire at timettthrough

ν​\(t\)=Δ​x⋅𝚊𝚛𝚐𝚖𝚊𝚡j​\(Fi​j​\(t\)\),\\nu\(t\)=\\Delta x\\cdot\\verb\|argmax\|\_\{j\}\(F\_\{ij\}\(t\)\),\(4\)whereΔ​x=2​m\\Delta x=2mis the grid size and𝚊𝚛𝚐𝚖𝚊𝚡j​\(Fi​j​\(t\)\)\\verb\|argmax\|\_\{j\}\\left\(F\_\{ij\}\(t\)\\right\)is the index of the pixel with maximum fuel density at timett\. Using the Eq\. \([4](https://arxiv.org/html/2607.06999#S4.E4)\), we can calculate the ROS through

ROS​\(t\)=1t​\(ν​\(t\)−ν​\(0\)\)\.\\text\{ROS\}\(t\)=\\frac\{1\}\{t\}\\left\(\\nu\(t\)\-\\nu\(0\)\\right\)\.\(5\)However, the`argmax`function in Eq\. \([4](https://arxiv.org/html/2607.06999#S4.E4)\) is nondifferentiable, which prevents backpropagation during training\. Therefore, we approximate the leading edge using a series of differentiable equations as follows\.

We first project the 2D differentiable mask in Eq\. \([2](https://arxiv.org/html/2607.06999#S4.E2)\) to a 1D transverse profile throughpj​\(t\)=maxi⁡PtG​Tp\_\{j\}\(t\)=\\max\_\{i\}P\_\{t\}^\{GT\}and calculate a location score,Sj​\(t\)=j\+λ​\(pj​\(t\)−1\)S\_\{j\}\(t\)=j\+\\lambda\(p\_\{j\}\(t\)\-1\), to suppress noise from non\-burning columns\. Through the use of temperature\-scaled softmax of the location scores,𝚜𝚘𝚏𝚝𝚖𝚊𝚡​\(Sj​\(t\)/𝒯\)\\verb\|softmax\|\(S\_\{j\}\(t\)/\\mathcal\{T\}\), we calculate an approximate leading edge for the ground truth fuel map at timett,νG​T​\(t\)=∑jj⋅pj​\(t\)⋅fj​\(t\)\\nu^\{GT\}\(t\)=\\sum\_\{j\}j\\cdot p\_\{j\}\(t\)\\cdot f\_\{j\}\(t\)\. We then calculate the approximate ground\-truth rate of spread,ROSG​T​\(t\)\\text\{ROS\}^\{GT\}\(t\), through the approximate leading edge using Eq\. \([5](https://arxiv.org/html/2607.06999#S4.E5)\)\. For the predicted approximate rate of spread,ROS^​\(t\)\\widehat\{\\text\{ROS\}\}\(t\), we use the predicted fuel density map,F^t\\hat\{F\}\_\{t\}, instead of the ground truth,FtG​TF\_\{t\}^\{GT\}, while calculating the differentiable mask in Eq\. \([2](https://arxiv.org/html/2607.06999#S4.E2)\)\. We constrain the kinematic velocity of the fire front by minimizing the error between the predicted differentiable rate\-of\-spread \(ROS\) and the ground truth\. ROS is calculated through

ℒROS=‖ROS^​\(t\)−ROSG​T​\(t\)‖2\.\\mathcal\{L\}\_\{\\text\{ROS\}\}=\\left\\\|\\widehat\{\\text\{ROS\}\}\(t\)\-\\text\{ROS\}^\{GT\}\(t\)\\right\\\|^\{2\}\.\(6\)
5\. Total Loss \(ℒtotal\\mathcal\{L\}\_\{\\text\{total\}\}\):To enforce physical consistency, we proposeWiFireLoss, a composite objective function that aggregates fuel conservation, state consistency, and kinematic elements\. The total loss is given by:

ℒtotal=λMSE​ℒMSE\+λfuel​ℒfuel\+λburn​ℒburn\+λunburn​ℒunburn\+λROS​ℒROS,\\displaystyle\\begin\{split\}\\mathcal\{L\}\_\{\\text\{total\}\}&=\\lambda\_\{\\text\{MSE\}\}\\mathcal\{L\}\_\{\\text\{MSE\}\}\+\\lambda\_\{\\text\{fuel\}\}\\mathcal\{L\}\_\{\\text\{fuel\}\}\\\\ &\+\\lambda\_\{\\text\{burn\}\}\\mathcal\{L\}\_\{\\text\{burn\}\}\+\\lambda\_\{\\text\{unburn\}\}\\mathcal\{L\}\_\{\\text\{unburn\}\}\\\\ &\+\\lambda\_\{\\text\{ROS\}\}\\mathcal\{L\}\_\{\\text\{ROS\}\},\\end\{split\}\(7\)whereλM​S​E\\lambda\_\{MSE\},λf​u​e​l\\lambda\_\{fuel\},λb​u​r​n\\lambda\_\{burn\},λu​n​b​u​r​n\\lambda\_\{unburn\}, andλR​O​S\\lambda\_\{ROS\}are the weights associated with the corresponding loss components\.

## VExperimental Setup and Results

### V\-AExperimental Setup

All models were implemented in PyTorch usingbfloat16mixed precision on a single NVIDIA RTX 3090 Ti\. Training utilized the AdamW optimizer with a cosine annealing schedule \(PeakL​R=10−3→10−5LR=10^\{\-3\}\\to 10^\{\-5\}\)\. Models are trained using a context length of 6 frames \(Ti​n=6T\_\{in\}=6\) to predict a 3\-frame future window \(To​u​t=3T\_\{out\}=3\)\. For example, given an input sequence of timestepst1,…,t6t\_\{1\},\.\.\.,t\_\{6\}, the model predictst7,t8,t9t\_\{7\},t\_\{8\},t\_\{9\}\. Input and output images are296×296296\\times 296pixels\. The temporal stride is 1, so the next input sequence ist2,…,t7t\_\{2\},\.\.\.,t\_\{7\}\.

- •ConvLSTM:Trained over 15 epochs with a batch size of 4 and an early stopping patience of 5 epochs\. The hidden dimension size is 64\.
- •AFNONet:Configured with a 3D patch size of8×8×3​\(H×W×T\)8\\times 8\\times 3\\ \(H\\times W\\times T\), an embedding dimension of 256, and 4 layers with 4 spatial frequency blocks\. Trained for 50 epochs with a batch size of 8 and patience of 10 epochs\.
- •ViViT:Utilizes 3D tubelet tokens of shape\(t=3,h=4,w=4\)\(t=3,h=4,w=4\)\. The backbone consists of 4 layers, 4 attention heads, an embedding dimension of 256, and an MLP hidden dimension of 512\.

We use the following loss component weights \(λ\\lambda\):λM​S​E=1\\lambda\_\{MSE\}=1,λf​u​e​l=0\.01\\lambda\_\{fuel\}=0\.01,λR​O​S=0\.01\\lambda\_\{ROS\}=0\.01,λb​u​r​n=0\.1\\lambda\_\{burn\}=0\.1,λu​n​b​u​r​n=0\.1\\lambda\_\{unburn\}=0\.1\. Theseλ\\lambdavalues were chosen to capture the relative contribution of each component \(e\.g\., burned/unburned are less important than MSE, but more important than fuel transport and ROS\)\. After evaluating several sets of values, these provided the best balance between convergence and training stability\.

### V\-BExperimental Results

Table[I](https://arxiv.org/html/2607.06999#S3.T1)presents the performance evaluation across different metrics, including base MSE, state\-based errors \(burned/unburned\), fuel transport violation, and ROS error\. The results show that integrating the physics\-guidedWiFireLossimproves fuel density prediction performance across the tested architectures\. This demonstrates the generality of the approach for physics\-guided machine learning and its applicability to different deep learning models\. Importantly, the addition of physics loss terms reduces error means as well as standard deviations, indicating improved and more stable prediction performance\.

Fig[2](https://arxiv.org/html/2607.06999#S3.F2)illustrates the results of training withWiFireLoss\. Fuel density prediction at timestep 42 for all three models are displayed, along with the ground truth\. These images show that all models are able to predict fuel density with high fidelity\.

## VIConclusion

This work develops physics\-guided spatiotemporal ML approaches to predict and emulate fuel density evolution over time under different wind conditions and varying ignition patterns\. The objective of this project is to enable fire managers to adapt their plans to achieve a burn’s intended objective with changing environmental conditions\. Our approach uses a loss function with physical constraints to reduce physical inconsistencies in predictions, including fuel transport discrepancies, over\-burning in unburned areas, and over\-estimating fuel in burned areas\. Our framework can be extended to other spatiotemporal models for learning physically consistent fuel densities\. We demonstrate that integrating physics\-based constraints into ML models improves fuel density prediction for prescribed fire management\.

Future work includes evaluating the proposed method on extended\-length fire sequences and on diverse fuel types and topography\. We also plan to investigate incorporating other features and additional physical constraints\.

## Acknowledgment

The authors would like to thank the Societal Computing and Innovation Lab \(SCIL\) at the San Diego Supercomputer Center, University of California San Diego for their support of this study\. This work was funded in part by San Diego Gas and Electric \(SDG&E\) and NSF Award 2134904\.

## References

- \[1\]\(2021\)Emulation of wildland fire spread simulation using deep learning\.Neural Networks141,pp\. 184–198\.External Links:ISSN 0893\-6080,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.neunet.2021.04.006),[Link](https://www.sciencedirect.com/science/article/pii/S0893608021001337)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p1.1),[§I](https://arxiv.org/html/2607.06999#S1.p2.1)\.
- \[2\]A\. Arnab, M\. Dehghani, G\. Heigold, C\. Sun, M\. Lučić, and C\. Schmid\(2021\)ViViT: a video vision transformer\.External Links:2103\.15691,[Link](https://arxiv.org/abs/2103.15691)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p3.1),[§III\-B](https://arxiv.org/html/2607.06999#S3.SS2.p1.1)\.
- \[3\]L\. Bottero, F\. Calisto, G\. Graziano, V\. Pagliarino, M\. Scauda, S\. Tiengo, and S\. Azeglio\(2020\)Physics\-informed machine learning simulator for wildfire propagation\.External Links:2012\.06825,[Link](https://arxiv.org/abs/2012.06825)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p2.1)\.
- \[4\]S\. S\. Chatterjee, K\. Lindsay, N\. Chatterjee, R\. Patil, I\. A\. D\. Callafon, M\. Steinbach, D\. Giron, M\. H\. Nguyen, and V\. Kumar\(2023\)Prescribed fire modeling using knowledge\-guided machine learning for land management\.External Links:2310\.01593,[Link](https://arxiv.org/abs/2310.01593)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p1.1),[§III\-A](https://arxiv.org/html/2607.06999#S3.SS1.p1.1)\.
- \[5\]Z\. Cope\(2021\)Using convolutional neural networks to predict quic\-fire outputs\.InAGU Fall Meeting Abstracts,Vol\.2021,pp\. NH15A–0448\.Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p2.1)\.
- \[6\]M\. A\. Finney\(1998\)FARSITE: fire area simulator\-model development and evaluation\.U\.S\. Department of Agriculture, Forest Service, Rocky Mountain Research Station\.External Links:[Link](http://dx.doi.org/10.2737/RMRS-RP-4),[Document](https://dx.doi.org/10.2737/rmrs-rp-4)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p1.1)\.
- \[7\]J\. Guibas, M\. Mardani, Z\. Li, A\. Tao, A\. Anandkumar, and B\. Catanzaro\(2022\)Adaptive fourier neural operators: efficient token mixers for transformers\.External Links:2111\.13587,[Link](https://arxiv.org/abs/2111.13587)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p3.1),[§III\-C](https://arxiv.org/html/2607.06999#S3.SS3.p1.2)\.
- \[8\]K\. He, X\. Zhang, S\. Ren, and J\. Sun\(2015\)Deep residual learning for image recognition\.arXiv preprint arXiv:1512\.03385\.Cited by:[item 1](https://arxiv.org/html/2607.06999#S3.I1.i1.p1.1)\.
- \[9\]F\. Huot, R\. L\. Hu, M\. Ihme, Q\. Wang, J\. Burge, T\. Lu, J\. Hickey, Y\. Chen, and J\. Anderson\(2021\)Deep learning models for predicting wildfires from historical remote\-sensing data\.External Links:2010\.07445,[Link](https://arxiv.org/abs/2010.07445)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p1.1)\.
- \[10\]R\. R\. Linn, S\. L\. Goodrick, S\. Brambilla, M\. J\. Brown,et al\.\(2020\)QUIC\-fire: a fast\-running simulation tool for prescribed fire planning\.Environmental Modelling & Software125,pp\. 104616\.Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p1.1)\.
- \[11\]R\. R\. Linn, J\. M\. Reisner, J\. J\. Colman, and J\. L\. Winterkamp\(2002\)Studying wildfire behavior using firetec\.International Journal of Wildland Fire11\(3\-4\),pp\. 233–246\.External Links:[Document](https://dx.doi.org/10.1071/WF02007)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p1.1)\.
- \[12\]J\. Pathak, S\. Subramanian, P\. Harrington, S\. Raja, A\. Chattopadhyay, M\. Mardani, T\. Kurth, D\. Hall, Z\. Li, K\. Azizzadenesheli,et al\.\(2022\)FourCastNet: a global data\-driven high\-resolution weather model using adaptive fourier neural operators\.arXiv preprint arXiv:2202\.11214\.Cited by:[§III\-C](https://arxiv.org/html/2607.06999#S3.SS3.p1.2)\.
- \[13\]D\. Radke, A\. Hessler, and D\. Ellsworth\(2019\)FireCast: leveraging deep learning to predict wildfire spread\.InProceedings of the 28th International Joint Conference on Artificial Intelligence \(IJCAI\),pp\. 4575–4581\.Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p2.1)\.
- \[14\]M\. Raissi, P\. Perdikaris, and G\. E\. Karniadakis\(2019\)Physics\-informed neural networks: a deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations\.Journal of Computational Physics378,pp\. 686–707\.Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p2.1)\.
- \[15\]X\. Shi, Z\. Chen, H\. Wang, D\. Yeung, W\. Wong, and W\. Woo\(2015\)Convolutional lstm network: a machine learning approach for precipitation nowcasting\.InAdvances in Neural Information Processing Systems,Vol\.28\.Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p3.1)\.
- \[16\]USDA Forest Service\(2022\)Gallinas\-las dispensas prescribed fire declared wildfire review\.Technical reportU\.S\. Department of Agriculture, Forest Service\.Note:Accessed: 2026\-01\-17External Links:[Link](https://www.fs.usda.gov/sites/default/files/gallinas-las-dispensas-prescribed-fire-declared-wildfire-review.pdf)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p1.1)\.
- \[17\]S\. Wang, Y\. Teng, and P\. Perdikaris\(2021\)Understanding and mitigating gradient flow pathologies in physics\-informed neural networks\.SIAM Journal on Scientific Computing43\(5\),pp\. A3055–A3081\.External Links:[Document](https://dx.doi.org/10.1137/20M1318043)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p2.1)\.
- \[18\]P\. Zhang, Y\. Ban, and A\. Nascetti\(2021\)Learning u\-net without forgetting for near real\-time wildfire monitoring by the fusion of sar and optical time series\.Remote Sensing of Environment261,pp\. 112467\.External Links:ISSN 0034\-4257,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.rse.2021.112467),[Link](https://www.sciencedirect.com/science/article/pii/S0034425721001851)Cited by:[§I](https://arxiv.org/html/2607.06999#S1.p2.1)\.