The parameterized microphysics of a numerical weather or climate prediction model is one of the parts of the system that requires the greatest computational load. Furthermore, the understanding of physical laws is not fully represented in the parameterizations. In this work, some Deep Learning architectures and techniques were applied in order to emulate a microphysics parameterization of two-moment clouds in a Single Column Model (1D). The best solution found presented minor errors when comparing the prediction with the original microphysics.

This work created Artificial Neural Networks models for use in emulation of the Morrisson Cloud Microphysics implemented in BAM1D. BAM1D is a Single Column Model which implements the same Microphysics implemented in BAM3D. It was written in Fortran code.

Some efforts where made in a Python Jupyter Notebook, in advance of finding the best Neural Networks arhitectures, as MLP, CNN1D, CNN2D, LSTM to emulate the microphysics. The data used was generated from input and output of BAM1D microphysics routine, and then was used to train the Neural Networks. The best models where chosen to emulate the microphysics in BAM1D.

One of the BAM1D versions implements the emulation in Fortran code by using the Fortran Keras Bridge(FKB) framework, that implements Keras MLP in Fortran. This version aims to be fast, for production level, but work need to be done to accelerate the code using OpenMP, for example.

There is another version of BAM1D, for research, which communicates to a Python code (bam1d_physics_emulation.py) through txt files. This version is more flexible, because Python code receives the inputs from Fortran, calculates the results using any of the models (MLP, CNN, LSTM) and then generates the emulations as an output file, which is read from BAM1D code in turn.

*** ToDo ***

This repository is organized as follows:

bam1d folder structure:

bam1d                                              -  main bam1d folder                        
L ORIGINALS                                        -  original bam1d code                      
L model                                                                                       
  L datain                                         -  input  data folder for running model     
  L dataout                                        -  output data folder for model output      
  L exec                                                                                      
    L ParModel_MPI                                 -  executable                               
    L PARMODEL                                     -  model configuration                      
  L sources                                                                                   
    L Physics                                                                                 
      L MicroPhysics.f90                           -  Main microphisics model                  
        L Convection/MicroPhysics/MORRISON_AERO                                               
          L Micro_HugMorr_NN.f90                   -  emulates original code                   
          L Micro_HugMorr.f90                      -  original code with csv gen. for training  
L run                                              -  scripts for compiling and running bam1d  

TODO:

• Upload paper schematic
• Upload BAM1D Fortran emulation version
• Change codes training and running to run easily in Github