Package: tmm-fast
tmm_fasttmm_fast or transfer-matrix-method_fast is a lightweight package to speed up optical planar multilayer thin-film device computation.
Typically, one is interested in computing the Reflection and/or Transmission of light through a multilayer thin-film depending on the
wavelength and angle of incidence of the incoming light. The package is build on theoriginal tmm packagefrom sjbyrnes but quite a lot faster. Depending on the number of layers, wavelength range and angular range speed-ups of ~100x are possible. The physics behind the transfer matrix method can be studied in any textbook on optical devices or inMultilayer optical calculationsfrom Steven J. Byrnes.
More detailed information about the package and its applications can be found inTMM-Fast: A Transfer Matrix Computation Package for Multilayer Thin-Film Optimization.Figure 1: Overview about what you can do with this package.Getting StartedTo get a local copy up and running follow these steps:Installationtmm-fastpackage requirespytorch, therefore it is better to install it first followingofficial instructions.Stable version frompipy.org:pipinstalltmm-fastLatest version fromgithub.com:pipinstallgit+https://github.com/MLResearchAtOSRAM/tmm_fastUnified functionality of tmm_fast: Sponge PyTorch functionality for freeParallelized computation of reflection and transmission for coherent light spectra that traverse
a bunch of multilayer thin-films with dispersive materials.
This package is essentially build on Pytorch and its related functionalities such as GPU accelerations and Autograd.
It naturally allows for:GPU accelerated computationsTo compute gradients regarding the multilayer thin-film (i.e. N, T) thanks to Pytorch AutogradVectorized computations, i.e. using Einstein summation conventionIn general, tmm_fast is a lightweight package to speed up the computations of reflection and transmission of optical planar multilayer thin-films by vectorization regardinga set of multilayer thin-films consisting of various layers of particular material and thickness,a set of wavelengths, anda set of incident angles.For old guards: Numpy is fully supportedAll of the inputs can also be a numpy array format instead of torch tensors.
However, all internal computations are processed via PyTorch and thus the output data is converted back to numpy arrays again.
Hence, the use of numpy input may increase computation time due to data type conversions.Benefits and conducted sanity checks, backgroundsDepending on the number of thin films an their number of layers as well as the considered wavelengths that irradiate the thin film under particular angles of incident, the computation time can be decreased by 2-3 orders of magnitude.
This claim is supported by several cross-checks (https://arxiv.org/abs/2111.13667), conducted with the code provided by Steven J. Byrnes (https://arxiv.org/abs/1603.02720). Of course, the checks covered both, computational time and physical outputs.The physics behind the transfer matrix method can be studied in any textbook on optical devices or related papers, e.g.Chapter on thin films in Microphotonics (http://www.photonics.intec.ugent.be/download/ocs129.pdf) by Dries Van Thourhout, Roel Baets (UGent) and Heidi Ottevaere (VUB)Multilayer optical computations (https://arxiv.org/abs/1603.02720) by Steven J. ByrnesThe Fresnel Coefficient of Thin Film Multilayer Using Transfer Matrix Method (https://iopscience.iop.org/article/10.1088/1757-899X/518/3/032026) by Zahraa Hummam Mohammedgym-multilayerthinfilmThe proposed OpenAI/Farama-Foundation gymnasium environment utilizes the parallelized transfer-matrix method (TMM-Fast) to implement the optimization of multi-layer thin films as parameterized Markov decision processes. A very intuitive example is provided in example.py.
Whereas the contained physical methods are well-studied and known since decades, the contribution of this code lies the transfer to an OpenAI/Farama-Foundation gymnasium environment. The intention is to enable AI researchers without optical expertise to solve the corresponding parameterized Markov decision processes. Due to their structure, the solution of such problems is still an active field of research in the AI community.The publicationParameterized Reinforcement learning for Optical System Optimizationused a related environment.Getting startedTo get started you can do the example py-files for tmm (example_tmm.py) or the gymnasium environment (example_gym.py)!Multi-layer thin films meet parameterized reinforcement learningReinforcement learning is an area of machine learning concerned with how intelligent agents ought to take actions in an environment in order to maximize the notion of reward. The code to be published implements such an environment for the optimization of multi-layer thin films.
In principle, the proposed code allows to execute actions taken by an agent. These actions determine which material and with which thickness to stack next, thereby consecutively forming a multi-layer thin film as illustrated in Figure 1. Such a multilayer thin-film exhibits optical characteristics. By comparison between the actual and user-defined desired characteristics, a notion of numeric reward is computed based on which the agent learns to distinguish between good and bad design choices. Due to its physical and mathematical structure, the optimization of multi-layer thin film remains a challenging and thus still active field of research in the scientific community. As such it gained recent attention in many publications. Therefore, naturally the need for a standardized environment arises to make the corresponding research more trustful, comparable and consistent.Figure 2: Principal idea of an OpenAI/Farama-Foundation gymnasium environment. The agent takes an action that specifies the material and thickness of the layer to stack next. The environment implements the multi-layer thin film generation as consecutive conduction of actions and assigns a reward to a proposed multi-layer thin film based on how close the actual (solid orange line) fulfils a desired (dashed orange line) characteristic. The made experience is used to adapt the taken actions made in order to increase the reward and thus generate more and more sophisticated multi-layer thin films.Description of key featuresThe environment can include•	cladding of the multi-layer thin film (e.g. substrate and ambient materials),•	dispersive and dissipative materials,•	spectral and angular optical behavior of multi-layer thin films (See figure 3),•	… and many more.The environment class allows to•	conduct so-called parameterized actions (See publication) that define a multi-layer thin film,•	evaluate the generated thin film given a desired optical response, and•	render the results (See figure 3).In general, the comprehensive optimization of multi-layer thin films in regards of optical response encompasses•	the number of layers (integer),•	the thickness of each layer (float),•	the material of each layer (categorical, integer).Figure 3: Rendered output of the environment. Reflectivity (left) over angle of incidence and spectrum of a multi-layer thin film (right). Here, the stack features four layers and each layer’s material was chosen from a set of eight alternatives. The reward is computed based on a desired reflectivity, which is one for each angle and wavelength, but not displayed in this figure.CitingIf you use the code from this repository for your projects, please citeTMM-Fast: A Transfer Matrix Computation Package for Multilayer Thin-Film Optimizationor/andParameterized reinforcement learning for optical system optimizationin your publications.