PRISM: Position-encoded Regressive Inverse Spectral Model for Multilayer Thin-Film Design
Runtian Wang ⋅ Renhao Xue ⋅ Baige Chen ⋅ Hao Wu
Abstract
The inverse problem of multilayer thin-film optical coatings design represents a complex combinatorial-continuous optimization challenge. We present $\textbf{PRISM}$ (Position-encoded Regressive Inverse Spectral Model), a unified decoder-only autoregressive transformer that streamlines this process by jointly predicting discrete material selection and continuous thickness regression within a single backbone. PRISM introduces two primary architectural innovations: (1) $\textit{spectrum prefix conditioning}$, which utilizes standard prefix tokens for in-context target injection, and (2) $\textit{cumulative-depth Rotary Position Embeddings}$, which encode continuous thickness directly into the positional representation to preserve the physical spatial relationships of the stack. Our benchmarks demonstrate that a PRISM-13M model reduces MAE by over 50\% compared to other transformer baselines while utilizing only one-fifth of the parameters. Furthermore, a 44M-parameter variant achieves state-of-the-art performance (MAE = 0.010) on our in-distribution validation benchmark and operates significantly faster than simulated annealing, offering a highly efficient alternative to classical optimization methods.
Chat is not available.
Successful Page Load