Search papers, labs, and topics across Lattice.
CapBench, a new multi-PDK dataset, is introduced for machine learning-based post-layout capacitance extraction, derived from open-source designs across three technology nodes. The dataset contains 61,855 3D windows with high-fidelity capacitance labels generated using RWCap and validated against Raphael. Benchmarking 10 ML architectures reveals that CNNs achieve the lowest errors (1.75%), while GNNs offer a significant speedup (up to 41.4x) at the cost of higher errors (10.2%), highlighting an accuracy-speed trade-off.
You can now train your capacitance extraction models on a diverse, multi-PDK dataset of open-source designs, but be ready to trade accuracy for speed when choosing between CNNs and GNNs.
We present CapBench, a fully reproducible, multi-PDK dataset for capacitance extraction. The dataset is derived from open-source designs, including single-core CPUs, systems-on-chip, and media accelerators. All designs are fully placed and routed using 14 independent OpenROAD flow runs spanning three technology nodes: ASAP7, NanGate45, and Sky130HD. From these layouts, we extract 61,855 3D windows across three size tiers to enable transfer learning and scalability studies. High-fidelity capacitance labels are generated using RWCap, a state-of-the-art random-walk solver, and validated against the industry-standard Raphael, achieving a mean absolute error of 0.64% for total capacitance. Each window is pre-processed into density maps, graph representations, and point clouds. We evaluate 10 machine learning architectures that illustrate dataset usage and serve as baselines, including convolutional neural networks (CNNs), point cloud transformers, and graph neural networks (GNNs). CNNs demonstrate the lowest errors (1.75%), while GNNs are up to 41.4x faster but exhibit larger errors (10.2%), illustrating a clear accuracy-speed trade-off. Code and dataset are available at https://github.com/THU-numbda/CapBench.