English

Surrogate Neural Architecture Codesign Package (SNAC-Pack)

Machine Learning 2025-12-19 v1 Artificial Intelligence High Energy Physics - Experiment

Abstract

Neural Architecture Search is a powerful approach for automating model design, but existing methods struggle to accurately optimize for real hardware performance, often relying on proxy metrics such as bit operations. We present Surrogate Neural Architecture Codesign Package (SNAC-Pack), an integrated framework that automates the discovery and optimization of neural networks focusing on FPGA deployment. SNAC-Pack combines Neural Architecture Codesign's multi-stage search capabilities with the Resource Utilization and Latency Estimator, enabling multi-objective optimization across accuracy, FPGA resource utilization, and latency without requiring time-intensive synthesis for each candidate model. We demonstrate SNAC-Pack on a high energy physics jet classification task, achieving 63.84% accuracy with resource estimation. When synthesized on a Xilinx Virtex UltraScale+ VU13P FPGA, the SNAC-Pack model matches baseline accuracy while maintaining comparable resource utilization to models optimized using traditional BOPs metrics. This work demonstrates the potential of hardware-aware neural architecture search for resource-constrained deployments and provides an open-source framework for automating the design of efficient FPGA-accelerated models.

Keywords

Cite

@article{arxiv.2512.15998,
  title  = {Surrogate Neural Architecture Codesign Package (SNAC-Pack)},
  author = {Jason Weitz and Dmitri Demler and Benjamin Hawks and Nhan Tran and Javier Duarte},
  journal= {arXiv preprint arXiv:2512.15998},
  year   = {2025}
}

Comments

NeurIPS 2025 Machine Learning and the Physical Sciences Workshop, 8 pages, 4 figures, 3 tables

R2 v1 2026-07-01T08:30:18.550Z