Skip to content

esa/NIDN

Repository files navigation

Neural Inverse Design of Nanostructures (NIDN)

Documentation Status Tests GitHub last commit GitHub PyPI - Python Version GitHub contributors GitHub issues GitHub pull requests

Table of Contents
  1. About The Project
  2. Getting Started
  3. Usage
  4. Contributing
  5. License
  6. Contact

About The Project

Neural Inverse Design of Nanostructures (NIDN) is a Python project by the Advanced Concepts Team of ESA. The goal of the project is to enable inverse design of stacks of nanostructures, metamaterials, photonic crystals, etc., using neural networks in PyTorch. As forward models, it supports rigorous coupled-wave analysis and a finite-difference time-domain solver. There is an accompanying paper about to be published.

Scheme

Neural Inverse Design of Nanostructures with PyTorch
Explore the docs »

View Example notebook · Report Bug · Request Feature

Built With

This project is based on:

  • grcwa which was modified to include PyTorch support to allow neural network training.
  • fdtd which was modified to allow training neural networks with it and match the setup of NIDN

For more details than provided with NIDN on the forward models please refer to their docs. The adaptations of there code are in the folders nidn/trcwa/ and nidn/fdtd/.

Below you can see results of an exemplary optimization of a three-uniform-layer material to design a 1150nm filter.

Exemplary optimization of a three-uniform-layer material to design a 1150nm filter. | width=300 Exemplary optimization of a three-uniform-layer material to design a 1150nm filter. | width=300

Getting Started

This is a brief guide how to set up NIDN.

Installation

The easiest way is to pip install NIDN via

pip install nidn.

Alternatively, to use the latest code from this repo git clone the repository and make sure you have all the requirements installed.

To set up a conda environment for the project run conda env create -f environment.yml.

This will create a conda environment called nidn for it.

If you just want to install the module in your current environment you can run

pip install . in the root folder where the setup.py is located.

While NIDN does support GPU utilization there are only modest performance benefits to it at time of writing.

Test

After cloning the repository, developers can check the functionality of NIDN by running the following command in the nidn/tests directory:

pytest

Usage

Config

NIDN uses a central config file which is passed through the entire program. The default config parameters can be seen here. Practical usage of the config files is demonstrated in the included Jupyter notebook.

Use Case 1: Forward Model Simulation

This serves to compute the spectral characteristics of a material. The Jupyter notebooks Forward_Model_Simulation_with_FDTD.ipynb and Forward_Model_Simulation_with_RCWA.ipynb demonstrate this use case.

Use Case 2: Inverse Design of Nanostructures

This is the case you aim to design a material matching some target spectral characteristics. A thorough explanation is given in the Jupyter notebooks Inverse_Material_Design_with_FDTD.ipynb and Inverse_Material_Design_with_RCWA.ipynb.

Logging & Docs

To change the logging verbosity call nidn.setLogLevel(level) where level is one of TRACE,DEBUG,INFO,WARN and ERROR.

Detailed docs of NIDN are online can be built with sphinx. To do so, make sure sphinx is installed and run make html in the docs folder to get a local html version of the docs. readthedocs support may follow.

Supported Materials

If you try to design a material with the classification approach (see mentioned notebooks for more details), all materials in the materials folder will be utilized. You can manually add other materials there using data from, e.g., refractiveindex.info.

Roadmap

See the open issues for a list of proposed features (and known issues).

Contributing

The project is open to community contributions. Feel free to open an issue or write us an email if you would like to discuss a problem or idea first.

If you want to contribute, please

  1. Fork the project on GitHub.
  2. Get the most up-to-date code by following this quick guide for installing nidn from source:
    1. Get miniconda or similar
    2. Clone the repo
    git clone https://github.com/esa/nidn.git
    1. With the default configuration PyTorch with CUDA support is installed. If this should not happen, comment out cudatoolkit in the environment.yml.
    2. Set up the environment. This creates a conda environment called nidn and installs the required dependencies.
    conda env create -f environment.yml
    conda activate nidn

Once the installation is done, you are ready to contribute. Please note that PRs should be created from and into the main branch.

  1. Create your Feature Branch (git checkout -b feature/AmazingFeature)
  2. Commit your Changes (git commit -m 'Add some AmazingFeature')
  3. Push to the Branch (git push origin feature/AmazingFeature)
  4. Open a Pull Request on the main branch.

and we will have a look at your contribution as soon as we can.

Furthermore, please make sure that your PR passes all automated tests. Review will only happen after that. Only PRs created on the main branch with all tests passing will be considered.

License

Distributed under the GPL-3.0 License. See LICENSE for more information.

Contact

Created by ESA's Advanced Concepts Team

  • Pablo Gómez - pablo.gomez at esa.int

Project Link: https://github.com/esa/nidn