fabtwin is a Python package for designing multilayer optical
filters that still work after they are made. A filter of this kind
is a stack of thin transparent layers (tens of nanometres each) on a
glass substrate; the thickness and refractive index of each layer
decide which colours of light pass and which are blocked. A
deposition machine never builds exactly the stack you asked for. The
package learns how your machine tends to err from its past runs,
and then changes the design so that the worst few percent of the
devices it makes are as good as possible, because those decide how
many devices pass the specification.
It answers questions such as:
- What does this layer stack transmit and reflect, at each wavelength?
- Which layer thicknesses and indices give the best filter, if fabrication were perfect?
- Given logs of past runs (the recipe sent to the machine and what came out), what does a statistical model of the machine's errors look like, and how well does it match runs it has not seen?
- Which design keeps its worst 5 or 10 % of fabricated devices best?
- From one measured transmission spectrum of a finished stack, how far was each layer from its target thickness, and can that even be told apart?
- Which recipes should the calibration runs use, and how many runs are needed?
- What error band around a prediction holds for a stated fraction of new runs, without trusting any model, and how does it keep holding while the machine drifts?
- Has the machine changed since the twin was fitted, and is a new run unlike anything it has done before?
- From measurements at several angles, how far were both the thicknesses and the indices from the recipe?
- Halfway through a run, how should the remaining layers change to make up for the errors already made?
The optics are computed with the standard exact transfer-matrix method, and the design gradients with a derivative derived by hand (the "adjoint"), so the core runs on NumPy and SciPy alone. It is the generalized library form of the FabGAN-ID framework (Mahim, Islam, Rahman and Mohsin, IEEE Sensors Journal, 2026). An optional extra adds the paper's learned generative error model (a conditional WGAN-GP) in JAX. The guiding rule, from the paper:
Learn what cannot be simulated; differentiate what can.
The main results are checked by automated tests against closed-form answers, an independent optics code, or a second independent calculation (see How the results are checked). When an input is outside what a function can answer reliably, it raises an error that says why, instead of returning a number that looks fine but is not.
- A short guide to the words used here
- Install and requirements
- Examples (each with the output it prints)
- What is in the package
- When it refuses, and why
- How the results are checked
- Corrections in earlier versions
- Limits
- Where it comes from
- Citing, support and license
- Stack, layer, substrate -- the filter is
Nlayers on a substrate. Layerihas a thicknesst_i(micrometres) and a refractive indexn_i. Light arrives from the incidence medium (air, index 1, by default). - Transmittance
T, reflectanceR, absorptanceA-- the fraction of light power that passes through, bounces back, or is absorbed, at each wavelength;R + T + A = 1. For layers that absorb nothing,A = 0. An absorbing material has a complex indexn + i k; the extinction coefficientksets how strongly it absorbs. - Layer order -- layer 0 is the one next to the incidence medium,
layer
N - 1the one on the substrate. A coating is grown on the substrate, so in a real run layerN - 1is usually deposited first (the simulatedDepositionProcessinstead runs its correlated noise and intermixing in array order, from layer 0). - Normal and oblique incidence; s and p polarization -- light
arriving straight on (normal) or at an angle (oblique). At an angle,
the result depends on the direction in which the light's electric
field oscillates: across the plane of incidence (
"s") or within it ("p");"u"is unpolarized light, the average of the two. Beyond the critical angle (light going from a higher to a lower index at a steep angle) no light travels into the lower-index medium; the wave there only decays (it is evanescent). - Transfer-matrix method (TMM) -- the standard exact calculation
of
RandTfor a layer stack: one 2x2 matrix per layer, multiplied together (Macleod, Thin-Film Optical Filters). - Dispersion -- the index changes with wavelength. A Sellmeier
formula is a fitted equation for
n(wavelength), valid only over the range it was fitted on. The package describes a layer's index asn_i(lam) = n0_i * S(lam), wheren0_iis the index at a reference wavelength andSis the dispersion shape of the material (exactly 1 at the reference wavelength). - Merit
J-- one number scoring a spectrum, higher is better. The original design functions use a weighted sumJ = w . T + const; since 0.7.0 a merit can be any smooth function ofR,Tand the absorbed fractionA(seeOpticalModel). The built-in notch merit (block one laser line, pass everything else) isJ = 0.5 * (mean T in the pass bands) + 0.5 * (mean of 1 - T in the stop band): 1 for a perfect notch, 0 for the exact opposite. - Gradient and adjoint -- the gradient says how
Jchanges when each thickness or index changes a little. The adjoint is a way to get all of these at once, from two passes through the stack (one from the front, one from the back), however many layers there are. Here it is written out by hand, not produced by an automatic-differentiation library (software that derives gradients from code automatically). - Closed form, finite difference, standard error -- a closed-form answer is one given by an exact textbook formula. A (central) finite difference estimates a derivative by changing one input slightly up and down and comparing the two results. The standard error of an average is the typical size of its random error.
- Design box -- the allowed range of each thickness and index. A range with equal lower and upper limits freezes that parameter (for example, the index of a fixed deposited material).
- Recipe, run, trace -- a recipe is the stack you ask the machine for; a run is one deposition; a trace is the pair (recipe, what was actually made) for one run.
- Error vector
x-- one run's errors: the relative thickness errorst_made / t_recipe - 1of every layer, followed by the index errorsn_made - n_recipe. Length2N. - Process twin -- a statistical model that produces realistic
error vectors, fitted to traces.
GaussianTwinis a multivariate normal model; the optional learned twin is a conditional WGAN-GP (Wasserstein generative adversarial network with gradient penalty), a neural network trained to produce error vectors that look like the real ones, and whose errors may depend on the recipe. - Reference process --
PAPER_PROCESS, a simulated deposition machine with six kinds of error, used here to make example traces. It stands in for a real machine; it is not one. - CVaR (conditional value at risk) -- for a set of
Kmerit values from simulated fabricated devices,CVaR_alphais the mean of theceil(alpha * K)smallest. Withalpha = 0.1it is "the average of the worst 10 %". Robustifying a design means changing it to raise this number. - Yield -- the fraction of devices that meet a pass/fail specification.
- Conformal prediction -- a way to turn held-out errors into an error band that contains a stated fraction of new runs, without assuming any model is right. It needs the new runs to be exchangeable with the held-out ones (roughly: from the same, unchanged process). A Mondrian version does this separately for each group of runs (for example each recipe). A conformal p-value ranks a new run among held-out ones: a small value means the run is unlike them.
- Unbiased estimate, minibatch --
robustifyestimates the gradient from a small random batch of simulated devices (a minibatch) at each step; an estimate is unbiased when its average over many batches equals the exact value. - Permutation test, energy distance -- to ask whether two sets of runs come from the same machine, compute a distance between them (the energy distance, zero only for identical distributions), then see how often shuffling the runs between the two sets gives a distance at least as large; that fraction is the p-value.
pip install fabtwin # core: optics, adjoint, Gaussian twins, CVaR design
pip install 'fabtwin[twin]' # adds JAX and optax for the learned WGAN-GP twin
The core needs Python 3.10 or newer, NumPy 1.26 or newer and SciPy
1.11 or newer. The [twin] extra adds JAX 0.4.30 or newer and optax
0.2 or newer; nothing outside fabtwin.twin_jax imports JAX.
Units and conventions:
- Wavelengths and thicknesses are in micrometres (0.532 means 532 nm).
- Indices of design layers are given at a reference wavelength and
are real numbers. An absorbing material is written
n + i kwithk >= 0. The original gradient functions (transmittance_and_grads,merit_and_grad) are for non-absorbing stacks at normal incidence; the general ones added in 0.7.0 (stack_rta_and_grads, and every function that takesmodel=) also handle absorption and any angle. - Angles are in radians; polarization is
"s"or"p". - Error vectors are
(relative thickness errors, absolute index errors), as defined above. - Merits are higher-is-better, and CVaR is taken over the lower tail (the worst devices).
Each example below runs as written, and the output shown is what it
printed with fabtwin 0.7.0 (NumPy 2.4, SciPy 1.17, JAX 0.10 on
Linux). Wavelength ranges, stacks, error sizes and noise levels are
illustrative values chosen for the examples, not recommended designs.
The built-in materials and PAPER_PROCESS carry their own
literature references. All examples except example 8 need only the
core install; example 8 needs the [twin] extra. Each ran in under
15 seconds.
import numpy as np
import fabtwin as ft
lam0 = 0.550 # design wavelength, um
lam = np.array([lam0])
# One layer on a substrate of index 4.0. The ideal anti-reflection
# coating has index sqrt(1.0 * 4.0) = 2.0 and is a quarter wave thick.
n_f = 2.0
t = np.array([lam0 / (4 * n_f)]) # 0.06875 um
R, T = ft.stack_rt(lam, t, np.array([[n_f]]), n_inc=1.0,
n_sub=np.array([4.0]))
print(f"quarter-wave coating: R = {R[0]:.12f}, T = {T[0]:.12f}")
# Six alternating quarter-wave layers (HL)^3 make a mirror
# (H = high index, L = low index).
nH, nL, ns = 2.1, 1.5, 1.46
n = np.array([nH, nL] * 3)
R, T = ft.stack_rt(lam, lam0 / (4 * n), n[:, None], 1.0,
np.array([ns]))
Y = (nH / nL) ** 6 * ns # textbook closed form
print(f"(HL)^3 mirror: R = {R[0]:.6f}, closed form {((1 - Y) / (1 + Y)) ** 2:.6f}")quarter-wave coating: R = 0.000000000000, T = 1.000000000000
(HL)^3 mirror: R = 0.694285, closed form 0.694285
n_layers is (N,) for indices that do not change with wavelength,
or (N, L) for one value per layer and wavelength. transmittance
and reflectance return T or R alone; oblique incidence is set
with theta0_rad and pol.
import numpy as np
import fabtwin as ft
lam = np.linspace(0.40, 0.80, 161) # wavelengths, um
S = ft.dispersion_shape(lam, 0.550) # Si3N4 shape, 1 at 0.55 um
nsub = ft.SIO2_MALITSON1965.n(lam) # fused-silica substrate
w, c0 = ft.notch_weights(lam, 0.532, 0.015, 0.030)
rng = np.random.default_rng(0)
t = rng.uniform(0.020, 0.120, 20) # 20 thicknesses, um
n0 = rng.uniform(1.6, 2.4, 20) # 20 indices at 0.55 um
J, dJ_dt, dJ_dn = ft.merit_and_grad(lam, t, n0, S, w, c0, n_sub=nsub)
def merit(tt):
T = ft.transmittance(lam, tt, n0[:, None] * S, 1.0, nsub)
return float(ft.merit(T, w, c0))
h = 1e-7
e = np.zeros(20); e[4] = h
fd = (merit(t + e) - merit(t - e)) / (2 * h)
print(f"merit J = {J:.6f}")
print(f"dJ/dt_5: adjoint {dJ_dt[4]:.8f}, finite difference {fd:.8f}")merit J = 0.484024
dJ/dt_5: adjoint 1.39661451, finite difference 1.39661451
notch_weights(lam_um, center_um, half_um, guard_um) blocks
[center_um - half_um, center_um + half_um] and passes everything
outside a further guard_um on each side; wavelengths in the guard do
not count. One call to merit_and_grad gives all 40 derivatives; the
finite difference above needs two full calculations per derivative.
import numpy as np
import fabtwin as ft
lam = np.linspace(0.45, 0.65, 41) # um
S = ft.dispersion_shape(lam, 0.550)
nsub = ft.SIO2_MALITSON1965.n(lam)
w, c0 = ft.notch_weights(lam, 0.532, 0.015, 0.030) # block 532 nm
box = ft.DesignBox(8, 0.020, 0.120, 1.6, 2.4) # 8 layers
# 1. Best nominal design (as if fabrication were perfect).
t, n0, J, _ = ft.inverse_design(lam, S, w, c0, box, n_probe=60,
n_seed=2, n_iter=30, n_sub=nsub, seed=0)
print(f"nominal merit: {J:.4f}")
# 2. Historical runs of a tool (here: the built-in reference process).
rng = np.random.default_rng(1)
rt, rn, ftd, fnd = ft.PAPER_PROCESS.trace_dataset(*box.sample(rng, 40), 2, rng)
twin = ft.GaussianTwin(ft.errors_from_traces(rt, rn, ftd, fnd))
# 3. Make the worst 10 % of fabricated outcomes as good as possible.
t_r, n_r, _ = ft.robustify(lam, t, n0, S, w, c0, twin, box, alpha=0.1,
K=32, steps=40, lr=3e-3, seed=0, n_sub=nsub)
# 4. Score both designs with 500 fresh runs of the reference process.
def score(tt, nn):
r = np.random.default_rng(7)
return ft.evaluate_under_process(
lambda a, b, K: ft.PAPER_PROCESS.ensemble(a, b, K, r),
tt, nn, lam, S, w, c0, K=500, n_sub=nsub, alpha=0.1)
for name, (tt, nn) in [("nominal", (t, n0)), ("robust", (t_r, n_r))]:
s = score(tt, nn)
print(f"{name:8s} mean {s['mean']:.4f} CVaR(10 %) {s['CVaR']:.4f}")nominal merit: 0.7704
nominal mean 0.7335 CVaR(10 %) 0.6779
robust mean 0.7583 CVaR(10 %) 0.7218
In this run the robust design keeps its worst 10 % of fabricated devices better than the nominal design does (0.7218 against 0.6779), when both are scored by the reference process rather than by the twin that was used to robustify. That is one illustrative case, not a guarantee: the tests check only that robustification raises the CVaR under the twin it was given (see below).
inverse_design first scores n_probe random designs, then refines
the best n_seed of them by gradient ascent (Adam, a standard rule
for choosing the step sizes; each step is pulled back inside the
box). robustify draws K fresh samples of fabrication errors from
the twin at each of steps steps and follows the exact gradient of
their CVaR (or of mean minus beta times standard deviation with
mean_variance=True). Short runs are used here to keep the example
fast; the defaults are K=128, steps=150.
import os, tempfile
import numpy as np
import fabtwin as ft
rng = np.random.default_rng(0)
box = ft.DesignBox(4, 0.040, 0.120, 1.7, 2.1)
rt, rn, ftd, fnd = ft.PAPER_PROCESS.trace_dataset(*box.sample(rng, 3), 2, rng)
path = os.path.join(tempfile.mkdtemp(), "fab_traces.csv")
ft.save_traces_csv(path, rt, rn, ftd, fnd)
print(open(path).readline().strip()) # the header
rt, rn, ftd, fnd = ft.load_traces_csv(path)
ftd[2, 1] *= 1000.0 # a value typed in nm, not um
report = ft.validate_traces(rt, rn, ftd, fnd)
print(report["n_runs"], "runs,", report["n_layers"], "layers")
for run, layer, kind, value in report["flags"]:
print(f"flag: run index {run}, layer index {layer}, {kind} = {value:.1f}")run,layer,t_recipe_um,n_recipe,t_fab_um,n_fab
6 runs, 4 layers
flag: run index 2, layer index 1, rel_t = 1041.1
The file has one row per run and layer: thicknesses in micrometres,
indices at the reference wavelength, and layers numbered 1 to N
with no gaps in every run. validate_traces raises on structural
problems (wrong shapes, non-finite values, non-positive thicknesses)
but only flags implausible values (a relative thickness error or
an index error larger than 0.5 in size, by default; both limits are
keyword arguments). It
does not drop them: unusual runs are exactly what a twin should learn
from, so the decision is yours. Flag indices count from 0.
If your tool logs thickness but not index, write n_fab = n_recipe;
the fitted twin then has almost no index errors (the test asserts
below 1e-3), and DesignBox.thickness_only freezes the indices during
design.
To judge a twin on real data, fit it on part of the runs and score it
on the rest with twin_fidelity_report: it compares means,
covariances and correlations of the error vectors, and the mean,
lower percentile, CVaR and Wasserstein distance (W1) of the merit
distributions the two sets of errors produce on a set of designs,
through the exact optics. W1 measures how far apart two sets of
values are as a whole: roughly, the average distance each value must
be moved to turn one set into the other (0 when they are identical).
import numpy as np
import fabtwin as ft
lam = np.linspace(0.40, 0.80, 201)
S = ft.dispersion_shape(lam, 0.550)
nsub = ft.SIO2_MALITSON1965.n(lam)
recipe_t = np.random.default_rng(1).uniform(0.050, 0.120, 6) # um
n0 = np.array([1.46, 2.0, 1.46, 2.0, 1.46, 2.0])
# Pretend the tool made these relative thickness errors, and all we
# have is the measured transmittance, with 0.2 % measurement noise.
x_true = np.array([0.015, -0.022, 0.030, -0.011, 0.006, -0.028])
T, _, _ = ft.transmittance_and_grads(lam, recipe_t * (1 + x_true), n0, S,
n_sub=nsub)
T_meas = np.clip(T + np.random.default_rng(7).normal(0, 2e-3, lam.size), 0, 1)
rec = ft.errors_from_spectrum(lam, T_meas, recipe_t, n0, S, n_sub=nsub,
sigma_T=2e-3)
for i in range(6):
print(f"layer {i + 1}: true {x_true[i]:+.4f} "
f"recovered {rec.dt_over_t[i]:+.4f} +/- {rec.sigma[i]:.4f}")
print(f"chi2 {rec.chi2:.1f} for {rec.dof} degrees of freedom")
# Two touching layers of the same index: only their total thickness shows.
t3, n3 = np.array([0.080, 0.060, 0.100]), np.array([2.0, 2.0, 1.46])
T3, _, _ = ft.transmittance_and_grads(lam, t3, n3, S, n_sub=nsub)
try:
ft.errors_from_spectrum(lam, T3, t3, n3, S, n_sub=nsub)
except ValueError as err:
print("refused:", str(err).split(":")[0])layer 1: true +0.0150 recovered +0.0217 +/- 0.0027
layer 2: true -0.0220 recovered -0.0204 +/- 0.0012
layer 3: true +0.0300 recovered -0.0182 +/- 0.0206
layer 4: true -0.0110 recovered -0.0107 +/- 0.0004
layer 5: true +0.0060 recovered +0.0412 +/- 0.0155
layer 6: true -0.0280 recovered -0.0350 +/- 0.0036
chi2 146.0 for 195 degrees of freedom
refused: non-unique recovery
This is the same case as one of the tests. chi2 is the sum of the
squared misfits, each divided by the stated noise level; when the fit
is good and the noise level is right it is close to the number of
degrees of freedom (here, wavelengths minus layers). The uncertainties
matter:
layers 3 and 5 are poorly determined by this spectrum (their +/- is
large), and the recovered values there are far from the truth; the
well-determined layers are close. errors_from_spectrum fits
thickness errors only, holding the indices at the recipe values. It
tries n_starts starting points (8 by default) and refuses when
different error vectors fit the spectrum equally well, when some
combination of layers has no effect on the spectrum, or when there
are fewer wavelengths than layers.
import numpy as np
import fabtwin as ft
box = ft.DesignBox(6, 0.04, 0.16, 1.7, 2.1)
rec_t, rec_n = ft.design_recipes(box, n_recipes=4, seed=0)
print("first recipe, thicknesses (um):", np.round(rec_t[0], 3))
# A pilot of 64 runs of one recipe (here from the reference process).
rng = np.random.default_rng(5)
rt, rn, ftd, fnd = ft.PAPER_PROCESS.trace_dataset(
[np.full(6, 0.08)], [np.full(6, 1.9)], 64, rng)
pilot = ft.errors_from_traces(rt, rn, ftd, fnd)
M, se = ft.runs_for_twin_mean(0.004, pilot)
print(f"runs needed for a standard error of 0.004: {M} "
f"(largest predicted standard error {se.max():.5f})")first recipe, thicknesses (um): [0.087 0.087 0.084 0.1 0.125 0.093]
runs needed for a standard error of 0.004: 102 (largest predicted standard error 0.00399)
design_recipes spreads recipes over the box: it draws 512 candidate
recipes, starts from the one nearest their centre, and then
repeatedly adds the candidate farthest from all recipes chosen so far
(a greedy "maximin" rule; Johnson, Moore and Ylvisaker, J. Statist.
Plann. Inference 26, 131 (1990)). Distances are measured after
scaling each range to 0..1, and frozen parameters do not count. It is
a sensible rule of thumb, not a proof of the best possible choice.
runs_for_twin_mean uses the standard result that the standard error
of a mean after M independent runs is sqrt(variance / M), with the
variances estimated from a pilot of at least 8 runs, and returns the
smallest M that meets the target for every component.
import numpy as np
import fabtwin as ft
rng = np.random.default_rng(23)
t0, n0 = np.full(4, 0.08), np.full(4, 1.9)
# 60 held-out runs; the "prediction" is simply their mean error vector.
rt, rn, ftd, fnd = ft.PAPER_PROCESS.trace_dataset([t0], [n0], 60, rng)
x = ft.errors_from_traces(rt, rn, ftd, fnd)
pred = x.mean(axis=0)
scores = np.max(np.abs(x - pred), axis=1) # worst component per run
q = ft.conformal_quantile(scores, alpha=0.1) # 90 % level
lo, hi = ft.conformal_interval(pred, q)
print(f"q = {q:.4f}; guaranteed coverage "
f"{ft.conformal_coverage_exact(60, 0.1):.4f} (for continuous scores)")
# Check on 400 fresh runs.
rt, rn, ftd, fnd = ft.PAPER_PROCESS.trace_dataset([t0], [n0], 400, rng)
x_new = ft.errors_from_traces(rt, rn, ftd, fnd)
inside = np.all((x_new >= lo) & (x_new <= hi), axis=1)
print(f"fresh runs inside the interval: {inside.mean():.3f}")
try:
ft.conformal_quantile(scores[:5], alpha=0.1)
except ValueError as err:
print("refused:", err)q = 0.0872; guaranteed coverage 0.9016 (for continuous scores)
fresh runs inside the interval: 0.907
refused: 5 held-out scores cannot certify level 0.9: the required rank 6 exceeds n. Hold out at least 9 runs, or lower the confidence
conformal_quantile returns the ceil((n + 1)(1 - alpha))-th smallest
of the n held-out scores. The split-conformal theorem (Vovk,
Gammerman and Shafer (2005); Lei et al., J. Am. Stat. Assoc. 113,
1094 (2018); Angelopoulos and Bates, arXiv:2107.07511) says a new
exchangeable run then falls within q of its prediction with
probability at least 1 - alpha; for continuous scores the exact
probability is ceil((n + 1)(1 - alpha)) / (n + 1), which
conformal_coverage_exact returns. This holds on average over runs,
not for each recipe separately, and only while the process does not
change.
import numpy as np
import jax
import jax.numpy as jnp
import fabtwin as ft
from fabtwin import twin_jax as tj # needs fabtwin[twin]
lam = np.linspace(0.40, 0.80, 161)
S = ft.dispersion_shape(lam, 0.550)
nsub = ft.SIO2_MALITSON1965.n(lam)
w, c0 = ft.notch_weights(lam, 0.532, 0.015, 0.030)
rng = np.random.default_rng(1)
t, n0 = rng.uniform(0.020, 0.120, 12), rng.uniform(1.6, 2.4, 12)
g_jax = jax.grad(lambda tt, nn: tj.merit_jax(lam, tt, nn, S, w, c0,
n_sub=jnp.asarray(nsub)),
argnums=(0, 1))(jnp.asarray(t), jnp.asarray(n0))
_, g_t, g_n = ft.merit_and_grad(lam, t, n0, S, w, c0, n_sub=nsub)
diff = max(np.abs(np.asarray(g_jax[0]) - g_t).max(),
np.abs(np.asarray(g_jax[1]) - g_n).max())
print("JAX autodiff and hand adjoint agree to 1e-12:", diff < 1e-12)JAX autodiff and hand adjoint agree to 1e-12: True
Two separate derivations of the same gradient -- algebra by hand in
NumPy, automatic differentiation in JAX -- give the same numbers. The
learned twin itself is used as: build a FabTwinConfig, normalize the
recipes with norm_recipe, train with train_wgan on the error
vectors, draw fabricated stacks with sample_twin, and robustify with
robustify_gan. Training takes 4000 steps by default and is not
shown here.
import numpy as np
import fabtwin as ft
print(f"Si3N4 at 0.532 um: n = {ft.SI3N4_LUKE2015.n(0.532):.4f}")
print(f"fused silica at 0.532 um: n = {ft.SIO2_MALITSON1965.n(0.532):.4f}")
# Your own measured table (these numbers are illustrative only).
film = ft.TabulatedMaterial(
name="my SiNx film",
lam_um=[0.40, 0.50, 0.60, 0.70, 0.80],
n_table=[2.08, 2.04, 2.02, 2.01, 2.00],
reference="ellipsometry run 17, lab notebook p. 42 (illustrative)")
print(f"table at 0.55 um: n = {film.n(0.55):.4f}")
try:
film.n(0.90) # outside the table
except ValueError as err:
print("refused:", str(err).split(";")[0])Si3N4 at 0.532 um: n = 2.0559
fused silica at 0.532 um: n = 1.4607
table at 0.55 um: n = 2.0283
refused: my SiNx film: wavelength outside the tabulated range [0.4, 0.8] um (ellipsometry run 17, lab notebook p. 42 (illustrative))
A table is interpolated with PCHIP, a piecewise cubic that does not
overshoot between measured points. The reference field is required.
A table may include an extinction column k_table; then .n()
refuses (the design path is lossless) and .nk() gives n + i k for
the forward optics.
import numpy as np
import fabtwin as ft
lam = np.linspace(0.45, 0.75, 7)
t = np.array([0.080, 0.110, 0.012]) # the last layer: a thin metal-like film
n = np.array([2.10, 1.46, 0.50 + 3.0j]) # n + i k; k > 0 absorbs
g = ft.stack_rta_and_grads(lam, t, n, n_inc=1.0, n_sub=1.52,
theta0_rad=np.deg2rad(45), pol="u")
print("R + T + A = 1:", np.allclose(g["R"] + g["T"] + g["A"], 1.0))
print("A at 0.60 um:", round(float(g["A"][3]), 4))
# check dT/dt of the absorbing layer against a finite difference
h = 1e-6
Tp = ft.stack_rta_and_grads(lam, t + [0, 0, h], n, 1.0, 1.52, np.deg2rad(45), "u")["T"]
Tm = ft.stack_rta_and_grads(lam, t - [0, 0, h], n, 1.0, 1.52, np.deg2rad(45), "u")["T"]
print("exact:", round(float(g["dT_dt"][2, 3]), 6),
" finite difference:", round(float((Tp[3] - Tm[3]) / (2 * h)), 6))R + T + A = 1: True
A at 0.60 um: 0.1015
exact: -23.888131 finite difference: -23.888131
stack_rta_and_grads gives R, T, A = 1 - R - T and their
exact derivatives with respect to every thickness, the real part of
every index and every extinction coefficient k, for any angle and
for s, p or unpolarized ("u") light. It uses the same two-pass
adjoint idea as the original normal-incidence code; the tests check
it against finite differences, against the original code at normal
incidence, and against automatic differentiation of a separate JAX
implementation.
import numpy as np
import fabtwin as ft
lam = np.linspace(0.45, 0.70, 51)
shape = np.ones(lam.size) # dispersionless, for the example
stop = (lam >= 0.52) & (lam <= 0.545) # block this band ...
pas = (lam <= 0.50) | (lam >= 0.565) # ... and pass these
conds = ((0.0, "u"), (np.deg2rad(20), "u")) # straight on and at 20 degrees
box = ft.DesignBox(20, 0.02, 0.20, 1.45, 2.30)
# 1) a linear notch merit, averaged over both angles, as a starting point
w, c = ft.notch_weights(lam, 0.5325, 0.0125, 0.02)
start = ft.OpticalModel(ft.LinearMerit(np.tile(w / 2, (2, 1)), c, "T"), conds)
t, n, _, _ = ft.inverse_design(lam, shape, None, None, box, n_probe=150,
n_seed=3, n_iter=150, n_sub=1.52, model=start)
# 2) then the specification itself, as a smooth margin
model = ft.OpticalModel(ft.SpecMarginMerit(stop, pas, leak_max=0.10,
pass_min=0.85, sharpness=150.0),
conds)
t, n, J, _ = ft.adam_ascent(lam, t, n, shape, None, None, box, n_iter=150,
lr=2e-3, n_sub=1.52, model=model)
R, T, A = ft.model_spectra(model, lam, t, n, shape, 1.0, 1.52)
for (ang, pol), Tc in zip(model.conditions, T):
print(f"{np.rad2deg(ang):4.0f} deg: max T in stop band {Tc[stop].max():.3f}, "
f"mean T in pass bands {Tc[pas].mean():.3f}")
print(f"smooth margin J = {J:.4f}")
print("meets the spec at both angles:", bool(ft.pass_fail(T, stop, pas, 0.10, 0.85).all())) 0 deg: max T in stop band 0.042, mean T in pass bands 0.903
20 deg: max T in stop band 0.047, mean T in pass bands 0.910
smooth margin J = 0.0446
meets the spec at both angles: True
SpecMarginMerit scores the pass/fail specification of
pass_fail directly: the worst leakage in the stop band and the mean
transmission in the pass bands, at every angle in the model, combined
into one smooth number. It is built never to overstate the margin, so
J > 0 guarantees the specification is met (the tests check this on
3000 random spectra). TargetMerit (distance to a target spectrum),
LinearMerit on R, T or A, and FunctionMerit (your own
function and its derivatives) work the same way. Here a linear notch
merit gives the starting point and the specification margin finishes
the design.
import numpy as np
import fabtwin as ft
lam = np.linspace(0.45, 0.65, 61)
S = ft.dispersion_shape(lam, 0.55)
w, c = ft.notch_weights(lam, 0.532, 0.012, 0.02)
box = ft.DesignBox(10, 0.03, 0.15, 1.7, 2.3)
t, n, J, _ = ft.inverse_design(lam, S, w, c, box, n_probe=120, n_seed=3,
n_iter=40, n_sub=1.46)
rng = np.random.default_rng(3)
rec_t, rec_n = ft.design_recipes(box, 12)
x = ft.errors_from_traces(*ft.PAPER_PROCESS.trace_dataset(rec_t, rec_n, 10, rng))
def fab_cvar(tt, nn): # judged by 600 fresh runs of the reference process
s = lambda a, b, K: ft.PAPER_PROCESS.ensemble(a, b, K, np.random.default_rng(99))
return ft.evaluate_under_process(s, tt, nn, lam, S, w, c, 600, n_sub=1.46,
alpha=0.1)["CVaR"]
print(f"nominal design: CVaR10 = {fab_cvar(t, n):.4f}")
for label, twin, est in (("sorted worst draws", ft.GaussianTwin(x), "sort"),
("Rockafellar-Uryasev", ft.GaussianTwin(x), "ru"),
("RU + twin ensemble", ft.TwinEnsemble(x, n_members=16), "ru")):
tr, nr, _ = ft.robustify(lam, t, n, S, w, c, twin, box, alpha=0.1, K=64,
steps=80, n_sub=1.46, estimator=est)
print(f"{label:26s} CVaR10 = {fab_cvar(tr, nr):.4f}")nominal design: CVaR10 = 0.6165
sorted worst draws CVaR10 = 0.6801
Rockafellar-Uryasev CVaR10 = 0.6710
RU + twin ensemble CVaR10 = 0.6723
estimator="ru" ascends the Rockafellar-Uryasev form of CVaR,
tau - mean((tau - J)_+) / alpha, maximized over the threshold tau
together with the design. Its minibatch gradient is unbiased at any
K for that objective at the current tau, and that objective is
the CVaR when tau is the alpha quantile, which the ascent tracks.
The paper's estimator ("sort", the default) averages the worst draws
and is biased at small K (the tests show both facts). TwinEnsemble refits the
Gaussian twin on bootstrap resamples of the runs, so the design also
allows for what a limited number of runs cannot pin down about the
machine. In this example all three raise the CVaR of the fabricated
merit by a similar amount; the unbiased gradient is a guarantee about
the method, not a promise of a better design every time.
import dataclasses
import numpy as np
import fabtwin as ft
t = np.array([0.07, 0.09, 0.06, 0.11])
n = np.array([2.1, 1.6, 2.1, 1.6])
def runs(process, K, seed):
tf, nf = process.ensemble(t, n, K, np.random.default_rng(seed))
return ft.errors_from_traces(np.tile(t, (K, 1)), np.tile(n, (K, 1)), tf, nf)
old = runs(ft.PAPER_PROCESS, 40, 1)
same = runs(ft.PAPER_PROCESS, 40, 2)
drifted = runs(dataclasses.replace(ft.PAPER_PROCESS, beta_t=0.04), 40, 3)
print("same machine: p =", ft.drift_test(old, same, n_perm=499)["p_value"])
print("rate bias 2->4%: p =", ft.drift_test(old, drifted, n_perm=499)["p_value"])
# Is this one run unlike anything logged?
x = runs(ft.PAPER_PROCESS, 600, 4)
odd = x[:1].copy()
odd[0, 0] += 0.2 # a 20 % error on layer 1
print("ordinary run p =", ft.novelty_pvalues(x[:300], x[300:599], x[599:])[0].round(3),
"| unusual run p =", ft.novelty_pvalues(x[:300], x[300:599], odd)[0].round(4))
# Error bands that keep their long-run miss rate while the machine drifts
rng = np.random.default_rng(5)
aci = ft.AdaptiveConformal(np.abs(rng.normal(size=100)), alpha=0.1, gamma=0.01)
q_fixed = ft.conformal_quantile(np.abs(rng.normal(size=100)), 0.1)
miss_fixed = 0
for k in range(3000):
s = abs(rng.normal(0, 1.0 + 2.0 * (k > 1000) + 0.002 * k)) # a jump, then a ramp
miss_fixed += s > q_fixed
aci.update(s)
print(f"fixed band misses {miss_fixed / 3000:.3f} of runs; adaptive band "
f"{aci.miss_rate():.3f} (target 0.100, guaranteed within {aci.bound():.3f})")same machine: p = 0.534
rate bias 2->4%: p = 0.002
ordinary run p = 0.88 | unusual run p = 0.0033
fixed band misses 0.634 of runs; adaptive band 0.100 (target 0.100, guaranteed within 0.030)
drift_test compares the error vectors of old and new runs with a
permutation test of the energy distance; a small p-value says the
machine has changed and the twin needs refitting (a large one does
not prove nothing changed). novelty_pvalues gives each new run a
conformal p-value: for runs like the logged ones,
P(p <= u) <= u on average over runs and calibration sets (any one
calibration set can be somewhat off), and a run unlike anything
logged gets a small one.
AdaptiveConformal keeps adjusting its level so that the long-run
share of missed runs stays at alpha even when the process drifts
(Gibbs and Candes, NeurIPS 2021), with a bound that holds for any
sequence of runs. A band fixed at the start fails badly once the
process changes.
import numpy as np
import fabtwin as ft
lam = np.linspace(0.40, 0.90, 121)
S = ft.dispersion_shape(lam, 0.55)
t0 = np.array([0.070, 0.095, 0.060, 0.110, 0.080]) # recipe (um)
n0 = np.array([2.2, 1.5, 2.2, 1.5, 2.2])
xt = np.array([0.03, -0.02, 0.015, 0.01, -0.025]) # what the tool did (unknown)
dn = np.array([0.02, -0.015, 0.01, 0.0, -0.02])
rng = np.random.default_rng(1)
def measure(angle_deg, pol): # 0.2 % measurement noise
g = ft.stack_rta_and_grads(lam, t0 * (1 + xt), (n0 + dn)[:, None] * S, 1.0,
1.46, np.deg2rad(angle_deg), pol)
return ft.Measurement(np.deg2rad(angle_deg), pol, "T",
np.clip(g["T"] + rng.normal(0, 0.002, lam.size), 0, 1), 0.002)
try:
ft.errors_from_spectra(lam, [measure(0, "s")], t0, n0, S, fit_index=True,
n_sub=1.46, n_starts=4)
except ValueError as err:
print("one normal spectrum -> refused:", str(err)[:60], "...")
ms = [measure(0, "s")] + [measure(a, p) for a in (45, 60) for p in "sp"]
r = ft.errors_from_spectra(lam, ms, t0, n0, S, fit_index=True, n_sub=1.46, n_boot=30)
print("thickness error found:", np.round(r.dt_over_t, 4), "+/-", np.round(r.sigma_t, 4))
print("thickness error true: ", xt)
print("index error found: ", np.round(r.dn, 4), "+/-", np.round(r.sigma_n, 4))
print("index error true: ", dn)
print("bootstrap spread of the index errors:", np.round(r.boot_sigma_n, 4))one normal spectrum -> refused: non-unique recovery: distinct error vectors reproduce the me ...
thickness error found: [ 0.0243 -0.0179 0.0179 0.0088 -0.0224] +/- [0.0037 0.003 0.0043 0.0029 0.0024]
thickness error true: [ 0.03 -0.02 0.015 0.01 -0.025]
index error found: [ 0.0239 -0.012 0.0077 -0.0025 -0.0223] +/- [0.0025 0.0022 0.0018 0.0018 0.0021]
index error true: [ 0.02 -0.015 0.01 0. -0.02 ]
bootstrap spread of the index errors: [0.0022 0.002 0.0015 0.0015 0.0018]
From one normal-incidence spectrum, thickness and index errors
cannot be told apart (here two different error vectors fit equally
well, and the function refuses). Spectra at 45 and 60 degrees in both
polarizations add enough independent information: every recovered
error lies within a few error bars of the truth. The function keeps
all the refusals of errors_from_spectrum and decides from the data,
case by case, whether the indices are determined. n_boot repeats the
fit on synthetic noisy copies (a parametric bootstrap) as a check on
the linear error bars.
import numpy as np
import fabtwin as ft
lam = np.linspace(0.45, 0.65, 61)
S = ft.dispersion_shape(lam, 0.55)
w, c = ft.notch_weights(lam, 0.532, 0.012, 0.02)
box = ft.DesignBox(10, 0.03, 0.15, 1.7, 2.3)
t, n, J, _ = ft.inverse_design(lam, S, w, c, box, n_probe=120, n_seed=3,
n_iter=40, n_sub=1.46)
merit = lambda tt, nn: float(ft.merit(ft.transmittance(lam, tt, nn[:, None] * S,
1.0, 1.46), w, c))
rng = np.random.default_rng(7)
m = 5 # the 5 layers on the substrate side are deposited and measured
done, rest = slice(10 - m, 10), slice(0, 10 - m)
gains = []
for run in range(10):
tf, nf = ft.PAPER_PROCESS.corrupt(t, n, rng) # what the machine does
fix = ft.reoptimize_remaining(lam, t, n, S, w, c, m, tf[done], nf[done], box,
n_sub=1.46)
# the remaining layers suffer the same errors, corrected or not
t2, n2 = fix["t"].copy(), fix["n"].copy()
t2[rest] *= tf[rest] / t[rest]
n2[rest] += nf[rest] - n[rest]
gains.append(merit(t2, n2) - merit(tf, nf))
print("merit gained by correcting after 5 of 10 layers:", np.round(gains, 3))
print(f"mean gain {np.mean(gains):.3f}")merit gained by correcting after 5 of 10 layers: [ 0.016 0.028 0.021 0.008 0.015 0.014 0.016 0.017 0.004 -0.001]
mean gain 0.014
After the 5 layers next to the substrate (grown first) are deposited
and measured, reoptimize_remaining re-designs the other 5 to make
up for the errors already made. Each corrected run is compared with
the same run left alone, with the remaining layers given the same
errors: 9 of the 10 runs gain, one loses slightly, and the mean gain
is 0.014. Layer 0 of a fabtwin stack is the one next to the incidence
medium; first="incidence" treats layer 0 as grown first instead.
With twin= the remaining layers are instead made robust to what the
machine will still do, after conditioning the twin on the errors
already measured (GaussianTwin.conditional). This is a simple
re-optimization rule, not the paper's Stage 3.
import fabtwin as ft
box = ft.DesignBox(3, 0.03, 0.15, 1.7, 2.3)
greedy = ft.design_recipes(box, 6)
refined = ft.design_recipes(box, 6, refine=True)
print(f"smallest distance between recipes: greedy {ft.maximin_distance(box, *greedy):.3f}, "
f"refined {ft.maximin_distance(box, *refined):.3f}")smallest distance between recipes: greedy 1.019, refined 1.283
refine=True improves the greedy recipe choice by swapping recipes
for candidates while that increases the smallest distance between any
two recipes. It is never worse than the greedy choice, and here it is
26 % better, but it is still a local search, not a proven optimum.
Every name below is exported from fabtwin unless marked
fabtwin.twin_jax. Each docstring (help(fabtwin.robustify), for
example) gives the inputs, units and conventions.
Materials (fabtwin.materials)
SellmeierMaterial(name, terms, lam_min, lam_max, reference)-- a Sellmeier formula with its valid range and a source;sellmeieris the bare formula without a range check.SI3N4_LUKE2015-- silicon nitride, K. Luke et al., Opt. Lett. 40, 4823 (2015), valid 0.310-5.504 um;SIO2_MALITSON1965-- fused silica, I. H. Malitson, J. Opt. Soc. Am. 55, 1205 (1965), valid 0.21-6.7 um. Both via the refractiveindex.info database (CC0).TabulatedMaterial-- your measured table (example 9).dispersion_shape(lam, lam0, material)--S(lam) = n(lam)/n(lam0);layer_index(n_at_lam0, lam, lam0, material)--n0 * S(lam). This "one material, variable index" layer model follows the SiNx platform of Yesilyurt et al., Nanophotonics 12, 993 (2023).
Optics (fabtwin.tmm)
stack_rt,transmittance,reflectance-- exact transfer-matrixRandTfor any stack, normal or oblique incidence, s or p polarization, absorbing layers allowed.notch_weights,bandpass_weights-- the two filter merits of the paper as weights(w, const);merit(T, w, const)evaluatesw . T + const.weights_from_reflectance(w_R, const)-- turns a merit written in terms ofR(for a mirror) into one in terms ofT, usingR = 1 - T, which holds only for non-absorbing stacks.
Gradients (fabtwin.adjoint)
transmittance_and_grads(lam, t, n0, shape, ...)--Tand its derivatives with respect to every thickness and index.merit_and_grad--Jand its derivatives.shapeis one shared dispersion shape(L,)or one per layer(N, L)(for stacks of two or more materials). Normal incidence and non-absorbing layers only.
Gradients at any angle, with absorption, and other merits
(fabtwin.gradients, fabtwin.merits; new in 0.7.0)
stack_rta_and_grads(lam, t, n_layers, n_inc, n_sub, theta0_rad, pol)--R,T,Aand their derivatives with respect to every thickness, index real part and extinctionk;polis"s","p"or"u"(unpolarized).layer_indices(n0, shape, kext)buildsn0 * S + i k.- Merits:
LinearMerit(weights, const, quantity)(w . X + constforX=R,TorA),TargetMerit(closeness to a target spectrum),SpecMarginMerit(a smooth pass/fail margin;J > 0guarantees the specification),FunctionMerit(your own). OpticalModel(merit, conditions, kext)-- the merit, the measuring conditions (a list of(angle, polarization)) and the fixed absorption of the layers.inverse_design,adam_ascent,random_search,robustify,cvar_objective_and_grad,evaluate_under_process,induced_merits,twin_fidelity_reportandreoptimize_remainingtake it asmodel=(withw, constpassed asNone).model_spectraandmodel_merit_and_gradgive the spectra and the merit with its gradient.
Fabrication process and twins (fabtwin.process, fabtwin.twins)
DepositionProcess-- a simulated deposition machine with six error mechanisms: a systematic thickness bias, an index drift through the stack, intermixing with the previous layer, correlated thickness noise that grows with layer thickness, skewed index noise, and rare particle defects that thicken one layer. Each size is a field you can change or set to zero.PAPER_PROCESShas the paper's values (fieldreference: Mahim et al., IEEE Sensors J. (2026), Sec. V-D; magnitudes per Yesilyurt 2023 / Wilbrandt 2008). Methods:corrupt(one run),ensemble(many runs of one recipe),trace_dataset(traces for many recipes),deterministic_map(the result with all randomness off).errors_from_traces(rt, rn, ft, fn)-> error vectors;apply_errors(t, n, x)does the reverse.GaussianTwin(x, diagonal=False)-- a normal distribution fitted to error vectors;sample_errors,sample, andtransform(z)(mu + L z: turns standard normal random numberszinto error vectors, wheremuis the mean andL L^Tthe covariance; this letsrobustifyhold the random numbers fixed while it takes derivatives);conditional(observed, values)(new in 0.7.0) -- the twin given some error components already measured, from the exact Gaussian conditional distribution.
Risk and yield (fabtwin.risk)
cvar(values, alpha)-- mean of theceil(alpha K)smallest values (Rockafellar and Uryasev, J. Risk 2, 21 (2000), lower tail).tail_statistics-- mean, standard deviation,100*alphapercentile and CVaR.pass_fail(T, stop_mask, pass_mask, leak_max=0.09, pass_min=0.90)-- passes when the largestTin the stop band is at mostleak_maxand the meanTin the pass band is at leastpass_min;yield_fraction-- the fraction that pass.evaluate_under_process(sampler, ...)-- scores a design withKfresh fabricated samples from any process or twin.
Design and robust design (fabtwin.design, fabtwin.robust)
DesignBox(n_layers, t_lo, t_hi, n_lo, n_hi)-- each bound a number or one value per layer; equal bounds freeze a parameter;DesignBox.thickness_only(n_layers, t_lo, t_hi, n_fixed);sample,clip.inverse_design-- random probes then gradient ascent (query budgetn_probe + 2 * n_seed * n_iter);adam_ascent-- the ascent from one start;random_search-- the equal-budget random baseline.robustify-- CVaR (or mean minusbetastandard deviations) ascent through aGaussianTwin(or aTwinEnsemble) and the exact optics;estimator="sort"(default, the paper's) or"ru"(Rockafellar-Uryasev: unbiased gradient of an objective whose maximum over the threshold is the CVaR; new in 0.7.0);cvar_objective_and_gradandru_objective_and_grad-- the two objectives and their exact gradients for a fixed batch of random draws;TwinEnsemble(x, n_members)-- Gaussian twins refitted on bootstrap resamples of the runs. Fabricated thicknesses are clipped to[0.5 t_lo, 2 t_hi]and indices to[n_lo - 0.2, n_hi + 0.2]inside the objective.
Your data and how good the twin is (fabtwin.data,
fabtwin.fidelity)
save_traces_csv,load_traces_csv,validate_traces(example 4).moment_errors-- largest difference of the means, and the size (Frobenius norm: square root of the sum of squared entries) of the difference of the covariance matrices and of the correlation matrices, between two sets of error vectors.distribution_distances-- differences in mean, lower percentile and CVaR between two sets of merit values, and the 1-D Wasserstein distance (from SciPy).induced_merits-- the merit of one design under each error vector.twin_fidelity_report--moment_errorson the error vectors, plus thedistribution_distancesof the induced merits, averaged over a set of designs.- New in 0.7.0:
energy_distance;drift_test(x_old, x_new)-- a permutation test of "same machine";novelty_pvalues(x_train, x_calib, x_new)-- conformal p-values for "this run is like the logged ones" (example 13).
Reverse engineering (fabtwin.reverse)
errors_from_spectrum->SpectrumRecovery(fieldsdt_over_t,t_um,sigma,chi2,dof,condition_number,n_converged) (example 5).errors_from_spectra(lam, measurements, ...)->JointRecovery(new in 0.7.0): severalMeasurement(angle, pol, "T" or "R", values, sigma)at once, thickness errors and, withfit_index, index errors, optionally a parametric bootstrap (example 14).
Planning (fabtwin.lab)
design_recipes(withrefine=True, new in 0.7.0),maximin_distance,runs_for_twin_mean(examples 6 and 16).
Correcting a run (fabtwin.correct, new in 0.7.0)
reoptimize_remaining-- re-design the layers not yet deposited, given the measured ones;first="substrate"(default) or"incidence"says which end was grown first (example 15).
Conformal prediction (fabtwin.conformal)
conformal_quantile,conformal_interval,conformal_coverage_exact(example 7); new in 0.7.0:mondrian_quantiles(one guarantee per group, for example per recipe) andAdaptiveConformal(for a drifting process; example 13).
Learned twin (fabtwin.twin_jax, needs [twin])
FabTwinConfig-- layer count, box (plain numbers, strictly increasing), latent size (the number of random inputs the network turns into one error vector), hidden-layer sizes, andout_scale, the largest error the generator can output (0.30 by default).train_wgan-- trains the conditional WGAN-GP (Gulrajani et al., NeurIPS 2017) with an extra penalty matching means and covariances; the optionaltailargument adds the paper's physics-in-the-loop calibration of the worst merit losses. The module docstring reports the paper's finding that this calibration helps only when traces are scarce.sample_twin,robustify_gan(CVaR ascent through the generator and the JAX optics together; the paper's Stage 2).transmittance_jax,merit_jax-- the optics in JAX, normal incidence;norm_recipe,init_models,gen_forward-- building blocks;HAVE_JAX-- whether JAX could be imported.
fabtwin raises an error instead of guessing when:
- a wavelength is outside a Sellmeier formula's valid range or a table's measured range;
- a measured table has no
reference, is not strictly increasing, has non-positivenor negativek, or is asked for.n()while it has non-zerok; - an index has a negative imaginary part (gain, not absorption);
- the gradient functions (and so
inverse_design,robustify,errors_from_spectrum) get a complex thickness, index, dispersion shape, substrate index or incidence index -- they cover non-absorbing stacks only (the shape, substrate and incidence cases are new in 0.6.1); - array shapes do not match (layer indices vs thicknesses and
wavelengths, a per-layer shape that is not
(N, L), error vectors not of length2N); - a notch or bandpass layout leaves no wavelengths in a band;
- a design box is inverted, has a non-positive thickness bound, or is fully frozen (nothing to design);
- a
FabTwinConfiggets frozen or per-layer bounds (its recipe normalization divides byhi - lo); - process parameters are out of range (
rhooutside [0, 1), negative noise,p_flakenot a probability); - a trace file has the wrong header, no rows, a wrong number of fields, a duplicate or missing layer, or runs with different layer counts; or trace arrays have different shapes, non-finite values or non-positive thicknesses;
errors_from_spectrumhas fewer wavelengths than layers, a measuredToutside [0, 1], no converged fit, a combination of layers the spectrum cannot see, or two different answers that fit equally well;runs_for_twin_meangets fewer than 8 pilot runs, a pilot with zero variance in every component, or a target that is not positive;design_recipesgets fewer candidates than recipes;- a conformal level cannot be certified with the number of held-out scores (the message names the minimum), or scores are negative;
cvargets no values oralphaoutside (0, 1];pass_failgets an empty stop or pass mask;GaussianTwingets fewer than 2 error vectors;twin_fidelity_reportgets no designs;errors_from_spectrumgets asigma_Tthat is not positive and finite;weights_from_reflectancegets weights that are not 1-D;polis not"s"or"p";- the
twin_jaxfunctions are called without JAX installed (ImportErrornaming the extra); - new in 0.7.0:
stack_rta_and_gradsgets an absorbing incidence medium, an angle outside [0, 90) degrees, a polarization other than"s","p","u", or a layer exactly at its critical angle (no derivative there); anOpticalModelgets a bad condition; a merit gets an empty band, negative weights or a non-positive sharpness; a design or robust function gets neither weights normodel=;robustifygets an unknown estimator or"ru"with mean-variance;TwinEnsemblegets fewer than 4 runs or 1 member;errors_from_spectragets no measurements, a quantity other thanTorR, fewer values than unknowns, an ill-conditioned or non-unique fit;drift_testgets fewer than 2 runs per group or fewer than 19 permutations;mondrian_quantileshas a group too small for the level;reoptimize_remaininggets no deposited or no remaining layers;GaussianTwin.conditionalgets repeated or invalid indices.
117 automated tests run on every change, on Python 3.10, 3.11, 3.12,
3.13 and 3.14 with the [test,twin] extras, and once more on Python
3.10 with the oldest versions pyproject.toml allows (NumPy 1.26.0,
SciPy 1.11.0, JAX 0.4.30, optax 0.2.0; tmm 0.1.8 for the reference
check). Without JAX, 9 of the tests do not run (pytest reports the 6
in test_twin_jax.py as one skipped module, plus 3 others). Numerical
checks compare against a closed-form answer, an independent code, or
a second calculation; none compares against a number stored from an
earlier run. Checks of random processes use fixed seeds and
statistical tolerances. The main checks:
Optics
- A bare interface (substrate with no layers) gives exactly the
textbook Fresnel transmittance
4 n_sub / (1 + n_sub)^2; a half-wave layer is invisible and a quarter-wave anti-reflection layer givesT = 1, both to 1e-14. R + T = 1to 1e-12 on random 20-layer non-absorbing stacks; absorbing layers absorb (R + T < 1); gain is refused.- At 40 degrees, s and p reflectance match the Fresnel formulas to 1e-14; at the Brewster angle (the angle at which p-polarized light is not reflected at all) p reflectance is below 1e-30; at normal incidence s and p transmittance are identical.
- Agreement with the independent
tmmpackage (S. J. Byrnes, arXiv:1603.02720) to 1e-12 on random 8-layer stacks. (HL)^pquarter-wave mirrors match the textbook closed formR = ((1 - Y)/(1 + Y))^2,Y = (nH/nL)^(2p) n_sub, to 1e-12 for p = 1, 3, 6, also throughweights_from_reflectance.- For absorbing stacks,
weights_from_reflectanceis off by exactly-w_R . A(to 1e-12), whereAis the absorbed fraction. - The Sellmeier engine equals the formula evaluated directly (no difference); a table sampled from the Malitson formula reproduces its nodes to 1e-15 and the formula between nodes to 1e-6.
Gradients
- The adjoint's
Tequals the optics module's to 1e-13. - Against central finite differences on a random 20-layer stack: median relative error below 1e-8, largest below 1e-6 (the paper reports a median near 1e-10 for its JAX version). With per-layer dispersion on a two-material stack: median below 1e-7, largest below 1e-5. With a tabulated material and a user-defined Sellmeier material, each derivative checked has a relative error below 1e-6.
- At the quarter-wave anti-reflection optimum both derivatives are below 1e-12.
- JAX automatic differentiation equals the hand adjoint to 1e-12, and the JAX optics equal the NumPy optics to 1e-12.
- Complex dispersion shapes, substrate or incidence indices are refused (new in 0.6.1).
Process and twins
- With the random parts off, the reference process equals its closed form exactly.
- Over 4000 runs, the thickness noise has standard deviation within
0.002 of
sig_tand neighbour correlation within 0.05 ofrho; over 20000 runs, the mean index is within 2e-3 of the noise-free value and the index noise is right-skewed, and the particle-defect rate is within 0.006 ofp_flake. - Errors extracted from traces and applied back reproduce the traces to 1e-15.
- A
GaussianTwinfitted to 60000 samples recovers the true mean to 3e-4 and covariance to 5e-6.
Design and robust design
- On a one-layer anti-reflection problem,
inverse_designfinds the closed-form optimum: merit within 1e-10 of 1, thickness and index within 1e-6. - On an 8-layer notch it beats random search that is given one merit
evaluation for every probe and every gradient step of
inverse_design(n_probe + n_seed * n_iterevaluations). - The exact CVaR gradient of
cvar_objective_and_grad(fixed random draws) matches finite differences to a relative 1e-6, as does the mean-minus-deviation version. robustifyraises the CVaR measured under the twin it used; on the two-material platform and with frozen indices, it does not lower it (tolerance 1e-6).- With frozen indices,
inverse_designandrobustifyreturn the frozen values unchanged (exact equality).
Data, fidelity, reverse engineering, planning, conformal
- The CSV trace file round-trips exactly; five kinds of malformed file are refused; a unit mix-up is flagged, not dropped.
- Every fidelity distance is exactly zero between a sample and itself; W1 equals SciPy's value exactly; a mean-shifted twin scores worse than a fitted one.
- Spectrum recovery without noise returns the true errors to 1e-6, and a perfect deposition to 1e-8. With seeded 0.2 % noise (example 5), every recovered error is within 4 reported standard deviations of the truth and chi2 is between 0.5 and 1.7 times the degrees of freedom (one seeded case).
- Every
design_recipespick is re-derived independently (to 1e-12); the frozen index does not change the chosen thicknesses. runs_for_twin_meanreturns the smallest sufficientM(checked on both sides of the target); over 60 simulated calibrations of 40 runs, the scatter of the estimated mean matches the predicted standard error within 35 % for the largest component.conformal_quantileequals the rank formula exactly. In 4000 simulated trials (n = 39, alpha = 0.1) the coverage is within 4 standard errors of the exact value, and on the reference process 400 fresh runs reach at least1 - alphaminus 4 standard errors.
Gradients at any angle, with absorption; other merits (0.7.0)
- The forward optics agree with the independent
tmmpackage to 1e-12 on 200 random stacks: absorbing and lossless layers, s and p, angles up to 86 degrees, incidence media of index 1, 1.52 and 2, lossless and absorbing substrates, including cases beyond the critical angle (0.6.1 failed some of these; see Corrections). stack_rta_and_gradsequals the original adjoint at normal incidence to 1e-12 (relative); all its derivatives (thickness,n,k; forR,TandA) match central finite differences of the forward optics on 40 random stacks with absorption, oblique angles, s, p and unpolarized light, to a relative 1e-7 (plus an absolute 1e-8 for round-off); for thekof a non-absorbing layer the difference can only be one-sided (k cannot go negative) and the tolerance is 1e-4. The derivatives ofTfor one absorbing three-layer stack at 40 degrees (s) equal automatic differentiation of a separate JAX implementation to a relative 1e-10.- The gradients of every merit (
LinearMeritonRand onA,TargetMerit,SpecMarginMerit,FunctionMerit) through a three-angle absorbing model match finite differences to a relative 1e-6;LinearMeritonTat normal incidence equals the original path: the merit to 1e-13, its gradient to a relative 1e-11. - On 3000 random spectra
SpecMarginMeritnever exceeds the true margins, and every spectrum withJ > 0passespass_fail. - A 45-degree, unpolarized, absorbing bandpass design by
inverse_design(model=)is not beaten by random search with the same number of merit evaluations.
Robust design (0.7.0)
ru_objective_and_gradequals its formula (to 1e-14) and its gradient matches finite differences (relative 1e-5); whenalpha Kis a whole number, its maximum over the threshold is the CVaR of the batch (1e-13); otherwise the two differ slightly, becausecvaraverages theceil(alpha K)worst values.- Averaging 20000 minibatch gradients with
K = 20from one pool of 6000 draws: the Rockafellar-Uryasev gradient agrees with the pool's own CVaR gradient (every component within 4 standard errors), while the sorting estimator is off by more than 6 standard errors (the bias the paper describes). robustify(estimator="ru"), with a Gaussian twin and with aTwinEnsemble, raises the CVaR measured on 400 fresh runs of the reference process; the default path and the model path give the same objective and gradient (1e-10).
Drift, novelty, conformal (0.7.0)
energy_distanceequals a direct pairwise computation (1e-10).drift_testat the 5 % level rejects at most 15 % of 40 same-machine comparisons, and detects a thickness-bias change from 2 % to 4 % with 30 runs each (p <= 0.01).novelty_pvalueson 300 runs of an unchanged process, with one seeded calibration set, satisfyP(p <= u) <= uwithin 3 binomial standard errors for u = 0.05, 0.1, 0.2 (the guarantee is on average over calibration sets; one set adds its own spread), and runs with an unseen 20 % layer error get the smallest possible p-value.mondrian_quantilescovers each of two groups at the exact rate (1500 trials, within 4 standard errors);AdaptiveConformalmeets its long-run bound at every one of 3000 steps of a drifting sequence and ends within 0.02 of the target, where a fixed band misses more than 30 % of runs.
Joint recovery, correction, calibration design (0.7.0)
errors_from_spectrarecovers thickness and index errors from noise-free spectra at 0, 45 and 60 degrees to 1e-6; it refuses one normal-incidence spectrum, and also normal-incidenceTplusR(which carry the same information for a lossless stack); with 0.2 % noise every error is within 4 error bars, chi2/dof is between 0.8 and 1.25, and the bootstrap spread is within a factor 2 of the linear error bars. With thicknesses only it reproduceserrors_from_spectrum(1e-8).GaussianTwin.conditionalequals the closed-form conditional (1e-12); its measured components stay fixed; a regression over 200000 joint draws recovers the same coefficients (within 0.02).reoptimize_remainingwithout a twin never lowers the nominal merit given the measured layers; with or without one it never changes the measured layers, in both deposition orders; on the reference process the corrected runs end up better on average than the same runs uncorrected, with and without a twin. The conditioned twin reproduces the measured layers to 1e-15.design_recipes(refine=True)is never worse than the greedy design and is more than 20 % better in the tested three-layer case; the default is unchanged.
Learned twin ([twin] extra): the moment penalty is exactly zero
for identical batches; short training runs finish with finite losses;
samples stay within out_scale; the CVaR gradient through the
generator matches finite differences to a relative 1e-5;
robustify_gan stays in the box. These tests check that the machinery
works, not how good a trained twin is.
0.7.0 (this release) fixes a wrong reflectance beyond the critical
angle. Past the critical angle of a lossless medium (light arriving
from glass at a steep angle, for example) the wave in that medium
decays, and of the two square roots for n cos(theta) the solver must
take the decaying one. Versions up to 0.6.1 took the other one. This
matters only for a lossless substrate beyond its critical angle (for
a layer inside the stack either root gives the same result, and the
incidence medium never is beyond it). There T = 0 either way; with
all layers lossless R = 1 either way too, but when the stack
absorbs, R came out wrong: for glass (1.52) -> 100 nm of index 1.38 -> 80 nm of
2.1 + 0.05i -> air at 0.8 rad (s), 0.6.1 gave R = 0.938, the
independent tmm package and 0.7.0 give 0.888. Below the critical
angle of every medium the results are unchanged, bit for bit (checked
on 286 random cases). The tests compared with tmm only at normal
incidence before; they now cover oblique, absorbing and
beyond-critical cases.
0.6.1 closed a gap in the lossless check of the
gradient functions. transmittance_and_grads and merit_and_grad
refused complex thicknesses and indices, but a complex dispersion
shape, a complex substrate index array, or a NumPy complex scalar
substrate or incidence index was silently converted to its real part
(NumPy printed only a ComplexWarning), so the returned T and
gradients belonged to a different, non-absorbing
stack than the one given. For example, for three layers of index 2.0
and thickness 0.06 um on an absorbing substrate of index 1.5 + 0.2i
(given as an array), 0.6.0 returned T = 0.807 at 0.45 um where the
forward optics give T = 0.787. The same applied to inverse_design,
robustify and errors_from_spectrum, which call these functions.
This affected only calls with such complex inputs, which are outside
the documented scope; they are now refused with a clear message. As
was already the case for thicknesses and indices, a complex-typed
array is refused even when its imaginary part is zero (for example
.nk() of a table without k); pass the real values (.n())
instead. If
you passed TabulatedMaterial.nk() values or another complex
substrate to these functions, re-run with the forward optics
(fabtwin.tmm) instead.
The earlier README and CHANGELOG used "exact", "exactly" or "machine precision" for several checks that the tests perform with a small tolerance; the list above gives the tolerances the tests actually use. The full history is in CHANGELOG.md.
What 0.7.0 changed about the limits of 0.6.1, and what is left:
- Gradients. The whole gradient path (design, robust design,
spectrum recovery, correction) now works at any angle, for s, p and
unpolarized light, with absorbing layers, through
model=. The learned JAX twin'srobustify_ganstill uses its own normal-incidence, non-absorbing solver and linear merit; use aGaussianTwinorTwinEnsemblewithrobustify(model=...)for the other cases. An absorbing incidence medium is refused. - Merits. Any differentiable function of
R,TandAat one or several angles works (OpticalModel).SpecMarginMeritis a smooth stand-in for the pass/fail specification that never overstates it. The design is still found by local gradient ascent from random probes; nothing guarantees the global best. - CVaR estimate.
estimator="ru"has an unbiased gradient at anyK(for the Rockafellar-Uryasev objective, whose maximum over the threshold is the CVaR); the default stays the paper's estimator.TwinEnsemblecovers the uncertainty of a twin fitted to few runs. Neither can make a twin a fair model of a machine it has not observed. - Twin fidelity and drift.
drift_testdetects a change in the machine,novelty_pvaluesflags runs unlike the logged ones, andreoptimize_remainingcorrects a run in progress. They cannot predict error patterns the machine has never shown, and a detected drift still means refitting the twin. The paper's yield gains were obtained with its simulated process, as were this package's examples. - Spectrum recovery. Thickness and index errors can now be fitted
together from several measurements (angles, polarizations,
RandT); whether they are determined is decided case by case by the refusals, and one normal-incidence spectrum is still not enough. Error bars are still local (linear or bootstrap around the best fit); the dispersion shape of each layer is held at the recipe. - Conformal guarantees.
mondrian_quantilesgives a guarantee per group, but only for groups with their own held-out runs; a guarantee for every recipe at once, without such runs, is impossible for any method of this kind (Barber et al., Information and Inference 10, 455 (2021)).AdaptiveConformalkeeps a long-run miss rate under drift; it says nothing about a single run. - Calibration design.
refine=Trueimproves the greedy design by local exchanges; it is still not a proven optimum. - Not included, by choice: specification-conditioned correction
policies (the paper's Stage 3, which the paper treats as
exploratory;
reoptimize_remainingis a simpler, well-defined rule); neural forward surrogates (the paper's protocol study found the exact differentiable solver better in this setting: faster, exact, with exact gradients); and the paper's benchmark data (300 designs / 48,300 samples / 400 traces), which stays with the companion repository and its Zenodo archive --DepositionProcessgenerates equivalent data instead. - The programming interface may change before version 1.0.
The package is the reusable library form of the FabGAN-ID framework
of the associated paper. The paper's companion repository reproduces
the paper itself (a fixed 20-layer SiNx notch platform, with JAX
throughout); fabtwin generalizes it to any layer count, design box,
wavelength grid, cited material and linear merit, and replaces JAX
autodiff in the core with the hand-derived adjoint so the core needs
only NumPy and SciPy. The paper is a simulation study and names a twin
trained on real in-situ monitoring data as the essential next step;
the trace-file interface is meant for that.
T. M. Mahim, M. N. Islam, M. M. Rahman, A. S. M. Mohsin, "FabGAN-ID: Learning the Fabrication Process for Yield-Aware Inverse Design of Multilayer Photonic Sensor Filters", IEEE Sensors Journal (2026). Companion repository: Learned-generative-process-twins... (benchmark archived on Zenodo, doi:10.5281/zenodo.21315793).
If fabtwin helps your work, please cite it together with the
associated paper above. Every release is archived on Zenodo under the
concept DOI
10.5281/zenodo.22697049,
which always resolves to the latest version.
CITATION.cff has the details.
The package is written and maintained by Tanvir Mahmud Mahim (Department of Electrical and Electronic Engineering, BRAC University), who reviews every change and takes the final decision on scope and releases. There is no separate governance body; design questions are discussed in the open in issues and pull requests, and the standing rule of CONTRIBUTING.md binds the maintainer exactly as it binds contributors: a change that touches physics arrives with a test, and a constant arrives with its source. Questions and bug reports are welcome in the issue tracker.
Licensed under Apache-2.0.