Pushing the Limits of Inverse Lithography with Generative Reinforcement Learning
This addresses the critical but computationally challenging problem of mask optimization for semiconductor manufacturers, representing a hybrid advance over existing methods.
The paper tackles the problem of inverse lithography mask synthesis in semiconductor manufacturing by reformulating it as conditional sampling with a generator that proposes multiple mask candidates, then refining them with fast batched ILT. The result is a 20% EPE improvement on ICCAD13 cases with 3x speedup over SOTA numerical ILT, and doubled throughput with reduced EPE violations on LithoBench.
Inverse lithography (ILT) is critical for modern semiconductor manufacturing but suffers from highly non-convex objectives that often trap optimization in poor local minima. Generative AI has been explored to warm-start ILT, yet most approaches train deterministic image-to-image translators to mimic sub-optimal datasets, providing limited guidance for escaping non-convex traps during refinement. We reformulate mask synthesis as conditional sampling: a generator learns a distribution over masks conditioned on the design and proposes multiple candidates. The generator is first pretrained with WGAN plus a reconstruction loss, then fine-tuned using Group Relative Policy Optimization (GRPO) with an ILT-guided imitation loss. At inference, we sample a small batch of masks, run fast batched ILT refinement, evaluate lithography metrics (e.g., EPE, process window), and select the best candidate. On \texttt{LithoBench} dataset, the proposed hybrid framework reduces EPE violations under a 3\,nm tolerance and roughly doubles throughput versus a strong numerical ILT baseline, while improving final mask quality. We also present over 20\% EPE improvement on \texttt{ICCAD13} contest cases with 3$\times$ speedup over the SOTA numerical ILT solver. By learning to propose ILT-friendly initializations, our approach mitigates non-convexity and advances beyond what traditional solvers or GenAI can achieve.