Skip to content

The MultiScale Network for hierarchical regression (MS-Net) performs 3D regression based on a hierarchical principle: coarse inputs provide broad information about the data, and progressively finer-scale inputs can be used to refine this information.

License

Notifications You must be signed in to change notification settings

je-santos/ms_net

Repository files navigation

MS-Net: Computationally Efficient Multiscale Neural Network

Implementation of The MS-Net in Pytorch. This model provides an easy and effcient way to train neural networks with computationally large 3D arrays. The application shown in the paper considered simulations of flow through porous materials, but the method is general, and should be applicable to any other application involving large 2D/3D arrays.

Small example

An example (for electrical conductivity) can be found in this Google Colab

Usage

import torch

from network import MS_Net
from pore_utils import rnd_array
from network_tools import get_masks

net = MS_Net( 
              num_scales   := 4,   # num of trainable convNets
              num_features  = 1,   # input features (Euclidean distance, etc)
              num_filters   = 2,   # num of kernels on each layer of the finest model (most expensive)
              summary       = True # print the model summary
)

x     = rnd_array( size=128, scales = num_scales )
masks = get_masks( x[-1],    scales = num_scales )
y     = net( x, masks )[-1] # final prediction

Data

Download the desired data from the digital rocks portal (or create your own via your preferred simulation method).

Workflow

The following illustration shows how the information flows through the individual networks.

Video

A high-level talk describing the MS-Net highlights is posted on youtube

Citation

If you use our code for your own research, we would be grateful if you cite our publication TiPM

@article{MSNET,
title = "Computationally Efficient Multiscale Neural Networks Applied to Fluid Flow in Complex 3D Porous Media",
journal = " Transp Porous Med",
year = "2021",
issn = "1573-1634",
doi = "https://doi.org/10.1007/s11242-021-01617-y",
url = "https://link.springer.com/article/10.1007/s11242-021-01617-y"
}

About

The MultiScale Network for hierarchical regression (MS-Net) performs 3D regression based on a hierarchical principle: coarse inputs provide broad information about the data, and progressively finer-scale inputs can be used to refine this information.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published