ajsbsd/navier-stokes-2d-dataset

Fourier Neural Operator for Navier-Stokes 2D

This model is a Fourier Neural Operator (FNO) fine-tuned on the Navier-Stokes 2D dataset for solving partial differential equations in fluid dynamics. It serves as a fast surrogate model for traditional computational fluid dynamics simulations.

Model Description

The Fourier Neural Operator is a neural network architecture designed to learn mappings between function spaces, making it particularly effective for solving partial differential equations. This specific model has been trained to predict fluid dynamics governed by the Navier-Stokes equations in two dimensions.

Fine-Tuning Process

What is Fine-Tuning?

Fine-tuning is like teaching a skilled expert to specialize in a particular domain. In this case, we started with an FNO model that had general knowledge of physical systems and specialized it for 2D fluid dynamics using the Navier-Stokes dataset.

Training Details

• Repository: NNs-to-NOs
• Training Script: python train_single_res.py fno.yaml
• Epochs: 10
• Framework: PyTorch

Key Hyperparameters

• Learning Rate: 0.001 (careful, gradual learning)
• Optimizer: Adam (efficient optimization strategy)
• Batch Size: 32 (examples processed simultaneously)
• Resolution: [64, 64] (grid size for fluid state predictions)
• Modes: 12 (frequency modes captured by the FNO)
• Width: 20 (model complexity parameter)

Applications

This model enables fast approximations of fluid dynamics simulations, useful for:

• Engineering design and optimization
• Weather and climate modeling research
• Scientific computing acceleration
• Real-time fluid simulation applications

Usage

The model can be used as a surrogate for traditional computational fluid dynamics simulations, providing significant speedup while maintaining reasonable accuracy for problems within the training distribution.

Performance

The model achieves an L2 error of 0.0 on the validation set (please replace with actual performance metrics from your training).

Limitations

• Performance may degrade on flow regimes not represented in the training data
• Generalization depends on the diversity of the Navier-Stokes 2D dataset
• Scalability to higher dimensions or more complex physics requires further evaluation

Ethical Considerations

This model should be deployed responsibly, especially in critical applications where simulation accuracy is paramount. Users should understand the model's limitations and validate outputs against known benchmarks when possible.